StuffAnThings/qbit_manage
1
1
Fork 0
mirror of https://github.com/StuffAnThings/qbit_manage.git synced 2025-10-10 05:46:55 +08:00
Code Issues Projects Releases Packages Wiki Activity

Bug Fix: Create Category if missing in qbittorrent

Browse source
This commit is contained in:
bobokun 2022-01-06 12:13:59 -05:00
parent 2e8cfa4798
commit 1fb198f613
No known key found for this signature in database
GPG key ID: 9665BA6CF5DC2671
1 changed files with 9 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
modules/qbittorrent.py
View file

@ -1,5 +1,5 @@
import logging, os import logging, os
from qbittorrentapi import Client, LoginFailed, APIConnectionError, NotFound404Error from qbittorrentapi import Client, LoginFailed, APIConnectionError, NotFound404Error, Conflict409Error
from modules import util from modules import util
from modules.util import Failed, print_line, print_multiline, separator from modules.util import Failed, print_line, print_multiline, separator
from datetime import timedelta from datetime import timedelta
@ -132,7 +132,14 @@ class Qbt:
if torrent.category == '': if torrent.category == '':
new_cat = self.config.get_category(torrent.save_path) new_cat = self.config.get_category(torrent.save_path)
tracker = self.config.get_tags([x.url for x in torrent.trackers if x.url.startswith('http')]) tracker = self.config.get_tags([x.url for x in torrent.trackers if x.url.startswith('http')])
if not dry_run: torrent.set_category(category=new_cat) if not dry_run:
try:
torrent.set_category(category=new_cat)
except Conflict409Error:
e = print_line(f'Existing category "{new_cat}" not found for save path {torrent.save_path}, category will be created.', loglevel)
self.config.notify(e, 'Update Category', False)
self.client.torrent_categories.create_category(name=new_cat, save_path=torrent.save_path)
torrent.set_category(category=new_cat)
body = [] body = []
body += print_line(util.insert_space(f'Torrent Name: {torrent.name}', 3), loglevel) body += print_line(util.insert_space(f'Torrent Name: {torrent.name}', 3), loglevel)
body += print_line(util.insert_space(f'New Category: {new_cat}', 3), loglevel) body += print_line(util.insert_space(f'New Category: {new_cat}', 3), loglevel)