StuffAnThings/qbit_manage
1
1
Fork 0
mirror of https://github.com/StuffAnThings/qbit_manage.git synced 2025-01-09 09:27:40 +08:00
Code Issues Projects Releases Packages Wiki Activity

Changes

Browse source 
- Added custom dry-run logger
- Fixed double logging.
- Fixed description to have proper text on --log-file
This commit is contained in:
Visorask 2021-02-24 12:28:07 -06:00
parent 41725e3702
commit c00f1d44b0
2 changed files with 71 additions and 57 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
config.yml
View file

@ -14,9 +14,15 @@ cat:
# Tag Parameters
tags:
# <Tracker URL Keyword>: <Tag Name>
awesome-hd: AHD
beyond-hd: Beyond-HD
privatehd: PrivateHD
animebytes.tv: AnimeBytes
avistaz: Avistaz
landof.tv: BroadcasTheNet
passthepopcorn: PassThePopcorn
torrentleech: TorrentLeech
tleechreload: TorrentLeech
gazellegames: GGn
blutopia: Blutopia
hdts: HDTorrents
tv-vault: TV-Vault

120
qbit_manage.py
View file

@ -1,12 +1,14 @@
#!/usr/bin/python3
from qbittorrentapi import Client
import sys
import yaml
import argparse
import logging
from logging.handlers import RotatingFileHandler
import sys
from qbittorrentapi import Client
import urllib3
# import apprise
parser = argparse.ArgumentParser("qBittorrent Manager.",
@ -15,12 +17,12 @@ parser.add_argument('-c', '--config-file',
dest='config',
action='store',
default='config.yml',
help='This is used if you want to use different names for your config.yml. Example: tv.yml')
help='This is used if you want to use a different name for your config.yml. Example: tv.yml')
parser.add_argument('-l', '--log-file',
dest='logfile',
action='store',
default='activity.log',
help='This is used if you want to use different names for your config.yml. Example: tv.yml')
help='This is used if you want to use a different name for your log file. Example: tv.log')
parser.add_argument('-m', '--manage',
dest='manage',
action='store_const',
@ -55,76 +57,78 @@ parser.add_argument('--log',
help='Change your log level. ')
args = parser.parse_args()
with open(args.config, "r") as cfg_file:
cfg = yaml.load(cfg_file, Loader=yaml.FullLoader)
# Logging
log_lev = getattr(logging, args.loglevel.upper())
file_handler = logging.FileHandler(filename=args.logfile)
stdout_handler = logging.StreamHandler(sys.stdout)
rotate_handler = RotatingFileHandler(filename=args.logfile,
maxBytes=1024 * 1024 * 2,
backupCount=5)
stdout_handler = logging.StreamHandler(sys.stderr)
rotate_handler = RotatingFileHandler(filename=args.logfile, maxBytes=1024 * 1024 * 2, backupCount=5)
handlers = [file_handler, stdout_handler, rotate_handler]
# noinspection PyArgumentList
logging.basicConfig(level=log_lev,
format='%(asctime)s - %(levelname)s - %(message)s',
handlers=handlers)
logging.basicConfig(level=log_lev, format='%(asctime)s - %(levelname)s: %(message)s', handlers=handlers)
logger = logging.getLogger('qBit Manage')
# Add dry-run to logging.
logging.DRYRUN = 25
logging.addLevelName(logging.DRYRUN, 'DRY-RUN')
setattr(logger, 'dryrun', lambda message, *args: logger._log(logging.DRYRUN, message, args))
# Actual API call to connect to qbt.
client = Client(host=cfg["qbt"]["host"],
username=cfg["qbt"]["user"],
password=cfg["qbt"]["pass"])
torrent_list = client.torrents.info()
client = Client(host=cfg["qbt"]["host"], username=cfg["qbt"]["user"], password=cfg["qbt"]["pass"])
urllib3.disable_warnings()
def get_category(path):
cat_path = cfg["cat"]
for p, c in cat_path.items():
if p in path:
category = c
for i, f in cat_path.items():
if i in path:
category = f
return category
else:
category = ''
logger.warning('No categories matched. Check your config.yml file. - Setting tag to NULL')
return category
def get_tags(url):
tag_path = cfg["tags"]
for t, n in tag_path.items():
if t in url:
tag = n
for i, f in tag_path.items():
if i in url:
tag = f
return tag
else:
tag = ''
return tag
logger.warning('No tags matched. Check your config.yml file. Setting category to NULL')
return tag
def update_category():
if args.manage == 'manage' or args.cat_update == 'cat_update':
num_cat = 0
torrent_list = client.torrents.info()
for torrent in torrent_list:
if torrent.category == '':
new_cat = get_category(torrent.save_path)
if args.dry_run == 'dry_run':
logger.info('DRY-RUN: Torrent Name: %s', torrent.name)
logger.info('DRY-RUN: - New Category: %s', new_cat)
num_cat += 1
else:
logger.info('Torrent Name: %s', torrent.name)
logger.info(' - New Category: %s', new_cat)
torrent.set_category(category=new_cat)
num_cat += 1
for x in torrent.trackers:
if x.url.startswith('http'):
new_cat = get_category(torrent.save_path)
if args.dry_run == 'dry_run':
logger.dryrun('\n - Torrent Name: %s \n - New Category: %s \n - Tracker: %s',
torrent.name, new_cat, x.url)
num_cat += 1
else:
logger.info('\n - Torrent Name: %s \n - New Category: %s \n - Tracker: %s',
torrent.name, new_cat, x.url)
torrent.set_category(category=new_cat)
num_cat += 1
if args.dry_run == 'dry_run':
if num_cat >= 1:
logger.info('DRY-RUN: Did not update %s new categories.', num_cat)
logger.dryrun('Did not update %s new categories.', num_cat)
else:
logger.info('DRY-RUN: No new torrents to categorize.')
logger.dryrun('No new torrents to categorize.')
else:
if num_cat >= 1:
logger.info('Updated %s new categories.', num_cat)
@ -135,25 +139,26 @@ def update_category():
def update_tags():
if args.manage == 'manage' or args.tag_update == 'tag_update':
num_tags = 0
torrent_list = client.torrents.info()
for torrent in torrent_list:
if torrent.tags == '':
for x in torrent.trackers:
if x.url.startswith('http'):
new_tag = get_tags(x.url)
if args.dry_run == 'dry_run':
logger.info('DRY-RUN: Torrent Name: %s', torrent.name)
logger.info('DRY-RUN: - New Tag: %s', new_tag)
logger.dryrun('\n - Torrent Name: %s \n - New Tag: %s \n - Tracker: %s',
torrent.name, new_tag, x.url)
num_tags += 1
else:
logger.info('Torrent Name: %s', torrent.name)
logger.info(' - New Tag: %s', new_tag)
logger.info('\n - Torrent Name: %s \n - New Tag: %s \n - Tracker: %s',
torrent.name, new_tag, x.url)
torrent.add_tags(tags=new_tag)
num_tags += 1
if args.dry_run == 'dry_run':
if num_tags >= 1:
logger.info('DRY-RUN: Did not update %s new tags.', num_tags)
logger.dryrun('Did not update %s new tags.', num_tags)
else:
logger.info('DRY-RUN: No new torrents to tag.')
logger.dryrun('No new torrents to tag.')
else:
if num_tags >= 1:
logger.info('Updated %s new tags.', num_tags)
@ -164,24 +169,27 @@ def update_tags():
def rem_unregistered():
if args.manage == 'manage' or args.rem_unregistered == 'rem_unregistered':
rem_unr = 0
torrent_list = client.torrents.info()
for torrent in torrent_list:
for status in torrent.trackers:
if 'Unregistered torrent' in status.msg:
if args.dry_run == 'dry_run':
logger.info('DRY-RUN: %s -> %s', torrent.name, status.msg)
logger.info('DRY-RUN: - NOT Deleted')
rem_unr += 1
else:
logger.info('%s -> %s', torrent.name, status.msg)
logger.info(' - Deleted')
torrent.delete(hash=torrent.hash,
delete_files=True)
rem_unr += 1
for x in torrent.trackers:
if x.url.startswith('http'):
if 'Unregistered torrent' in status.msg:
if args.dry_run == 'dry_run':
logger.dryrun('\n - Torrent Name: %s \n - Status: %s \n - Tracker: %s \n '
'- File NOT Deleted', torrent.name, status.msg, x.url)
rem_unr += 1
else:
logger.info('\n - Torrent Name: %s \n - Status: %s \n - Tracker: %s \n '
'- Deleted', torrent.name, status.msg, x.url)
torrent.delete(hash=torrent.hash,
delete_files=True)
rem_unr += 1
if args.dry_run == 'dry_run':
if rem_unr >= 1:
logger.info('DRY-RUN: Did not delete %s torrents.', rem_unr)
logger.dryrun('Did not delete %s torrents.', rem_unr)
else:
logger.info('DRY-RUN: No unregistered torrents found.')
logger.dryrun('No unregistered torrents found.')
else:
if rem_unr >= 1:
logger.info('Deleted %s torrents.', rem_unr)