StuffAnThings/qbit_manage
1
1
Fork 0
mirror of https://github.com/StuffAnThings/qbit_manage.git synced 2024-09-20 15:26:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

New FR: Add Share/Upload Limits to Tags

Browse source
This commit is contained in:
Jon 2021-11-28 13:55:55 -05:00
parent d7e232b60c
commit b04beca5e4
No known key found for this signature in database
GPG key ID: 9665BA6CF5DC2671
2 changed files with 130 additions and 28 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
config/config.yml.sample
View file

@ -20,17 +20,25 @@ cat:
movies: "/data/torrents/Movies"
tv: "TV"
# Tag Parameters
# Tag Parameters (Use either Format 1 or Format 2)
tags:
# Format 1
# <Tracker URL Keyword>: <Tag Name>
# Format 2
# <Tracker URL Keyword>:
# tag: <Tag Name>
# <MANDATORY> Will set the torrent Maximum share ratio until torrent is stopped from seeding/uploading. -2 means the global limit should be used, -1 means no limit.
# max_ratio: 5.0
# <OPTIONAL> Will set the torrent Maximum seeding time (min) until torrent is stopped from seeding. -2 means the global limit should be used, -1 means no limit.
# max_seeding_time: 129600
# <OPTIONAL> Will limit the upload speed KiB/s (KiloBytes/second) (-1 sets the limit to infinity)
# limit_upload_speed: 150
#Format 1 Examples
animebytes.tv: AnimeBytes
avistaz: Avistaz
beyond-hd: Beyond-HD
blutopia: Blutopia
cartoonchaos: CartoonChaos
digitalcore: DigitalCore
gazellegames: GGn
hdts: HDTorrents
landof.tv: BroadcasTheNet
myanonamouse: MaM
passthepopcorn: PassThePopcorn
@ -39,6 +47,22 @@ tags:
torrentdb: TorrentDB
torrentleech: TorrentLeech
tv-vault: TV-Vault
#Format 2 Examples
avistaz:
tag: Avistaz
max_ratio: 5.0
max_seeding_time: 129600
limit_upload_speed: 150
digitalcore
tag: DigitalCore
max_ratio: 5.0
gazellegames:
tag: GGn
limit_upload_speed: 150
hdts:
tag: HDTorrents
max_seeding_time: 129600
#Tag Movies/Series that are not hard linked
nohardlinks:

124
qbit_manage.py
View file

@ -210,19 +210,33 @@ def get_category(path):
#Get tags from config file based on keyword
def get_tags(urls):
new_tag = ''
max_ratio = ''
max_seeding_time = ''
limit_upload_speed = ''
url = trunc_val(urls[0], '/')
if 'tags' in cfg and cfg["tags"] != None and urls:
tag_path = cfg['tags']
for i, f in tag_path.items():
tag_values = cfg['tags']
for tag_url, tag_details in tag_values.items():
# If using Format 1
if(type(tag_details) == str):
new_tag = tag_details
# Using Format 2
else:
if 'tag' in tag_details:
new_tag = tag_details['tag']
else:
logger.warning(f'No tags defined for {tag_url}. Please check your config.yml file.')
if 'max_ratio' in tag_details: max_ratio = tag_details['max_ratio']
if 'max_seeding_time' in tag_details: max_seeding_time = tag_details['max_seeding_time']
if 'limit_upload_speed' in tag_details: limit_upload_speed = tag_details['limit_upload_speed']
for url in urls:
if i in url:
tag = f
if tag: return tag,trunc_val(url, '/')
if tag_url in url:
return (new_tag,trunc_val(url, '/'),max_ratio,max_seeding_time,limit_upload_speed)
else:
tag = ('','')
return tag
tag = ('','')
logger.warning(f'No tags matched for {urls}. Check your config.yml file. Setting tag to NULL')
return tag
return (new_tag,url,max_ratio,max_seeding_time,limit_upload_speed)
logger.warning(f'No tags matched for {url}. Please check your config.yml file. Setting tag to NULL')
return (new_tag,url,max_ratio,max_seeding_time,limit_upload_speed)
#Move files from source to destination, mod variable is to change the date modified of the file being moved
@ -300,8 +314,7 @@ def set_recheck():
torrent_sorted_list = client.torrents.info(status_filter='paused',sort='size')
if torrent_sorted_list:
for torrent in torrent_sorted_list:
new_tag,t_url = get_tags([x.url for x in torrent.trackers if x.url.startswith('http')])
if torrent.tags == '' or ('cross-seed' in torrent.tags and len([e for e in torrent.tags.split(",") if not 'noHL' in e]) == 1): torrent.add_tags(tags=new_tag)
new_tag = get_tags([x.url for x in torrent.trackers if x.url.startswith('http')])[0]
#Resume torrent if completed
if torrent.progress == 1:
#Check to see if torrent meets AutoTorrentManagement criteria
@ -309,18 +322,18 @@ def set_recheck():
logger.debug(util.insert_space(f'- Torrent Name: {torrent.name}',2))
logger.debug(util.insert_space(f'-- Ratio vs Max Ratio: {torrent.ratio} < {torrent.max_ratio}',4))
logger.debug(util.insert_space(f'-- Seeding Time vs Max Seed Time: {timedelta(seconds=torrent.seeding_time)} < {timedelta(minutes=torrent.max_seeding_time)}',4))
if torrent.ratio < torrent.max_ratio and (torrent.seeding_time < (torrent.max_seeding_time * 60)):
if torrent.ratio < torrent.max_ratio and (torrent.seeding_time < (torrent.max_seeding_time * 60)) or (torrent.max_ratio < 0 or torrent.seeding_time < 0):
if dry_run:
logger.dryrun(f'Not Resuming {new_tag} - {torrent.name}')
logger.dryrun(f'Not Resuming [{new_tag}] - {torrent.name}')
else:
logger.info(f'Resuming {new_tag} - {torrent.name}')
logger.info(f'Resuming [{new_tag}] - {torrent.name}')
torrent.resume()
#Recheck
elif torrent.progress == 0 and torrentdict[torrent.name]['is_complete'] and not torrent.state_enum.is_checking:
if dry_run:
logger.dryrun(f'Not Rechecking {new_tag} - {torrent.name}')
logger.dryrun(f'Not Rechecking [{new_tag}] - {torrent.name}')
else:
logger.info(f'Rechecking {new_tag} - {torrent.name}')
logger.info(f'Rechecking [{new_tag}] - {torrent.name}')
torrent.recheck()
# Function used to move any torrents from the cross seed directory to the correct save directory
@ -447,19 +460,84 @@ def set_tags():
num_tags = 0
for torrent in torrent_list:
if torrent.tags == '' or ('cross-seed' in torrent.tags and len([e for e in torrent.tags.split(",") if not 'noHL' in e]) == 1):
new_tag,t_url = get_tags([x.url for x in torrent.trackers if x.url.startswith('http')])
new_tag,url,max_ratio,max_seeding_time,limit_upload_speed = get_tags([x.url for x in torrent.trackers if x.url.startswith('http')])
if new_tag:
if dry_run:
num_tags += 1
logger.dryrun(util.insert_space(f'Torrent Name: {torrent.name}',3))
logger.dryrun(util.insert_space(f'New Tag: {new_tag}',8))
logger.dryrun(util.insert_space(f'Tracker: {t_url}',8))
num_tags += 1
logger.dryrun(util.insert_space(f'Tracker: {url}',8))
if limit_upload_speed:
if limit_upload_speed == -1:
logger.dryrun(util.insert_space(f'Limit UL Speed: Infinity',1))
else:
logger.dryrun(util.insert_space(f'Limit UL Speed: {limit_upload_speed} kB/s',1))
if max_ratio:
if max_ratio == -2:
logger.dryrun(util.insert_space(f'Share Limit: Use Global Share Limit',4))
continue
elif max_ratio == -1:
logger.dryrun(util.insert_space(f'Share Limit: Set No Share Limit',4))
continue
else:
max_ratio = torrent.max_ratio
if max_seeding_time:
if max_seeding_time == -2:
logger.dryrun(util.insert_space(f'Share Limit: Use Global Share Limit',4))
continue
elif max_seeding_time == -1:
logger.dryrun(util.insert_space(f'Share Limit: Set No Share Limit',4))
continue
else:
max_seeding_time = torrent.max_seeding_time
if max_ratio != torrent.max_ratio and max_seeding_time != torrent.max_seeding_time:
logger.dryrun(util.insert_space(f'Share Limit: Max Ratio = {max_ratio}, Max Seed Time = {max_seeding_time} min',4))
elif max_ratio != torrent.max_ratio:
logger.dryrun(util.insert_space(f'Share Limit: Max Ratio = {max_ratio}',4))
elif max_seeding_time != torrent.max_seeding_time:
logger.dryrun(util.insert_space(f'Share Limit: Max Seed Time = {max_seeding_time} min',4))
else:
logger.info(util.insert_space(f'Torrent Name: {torrent.name}',3))
logger.info(util.insert_space(f'New Tag: {new_tag}',8))
logger.info(util.insert_space(f'Tracker: {t_url}',8))
torrent.add_tags(tags=new_tag)
num_tags += 1
logger.info(util.insert_space(f'Torrent Name: {torrent.name}',3))
logger.info(util.insert_space(f'New Tag: {new_tag}',8))
logger.info(util.insert_space(f'Tracker: {url}',8))
if limit_upload_speed:
if limit_upload_speed == -1:
logger.info(util.insert_space(f'Limit UL Speed: Infinity',1))
else:
logger.info(util.insert_space(f'Limit UL Speed: {limit_upload_speed} kB/s',1))
torrent.set_upload_limit(limit_upload_speed*1024)
if max_ratio:
if max_ratio == -2:
logger.info(util.insert_space(f'Share Limit: Use Global Share Limit',4))
torrent.set_share_limits(-2,-2)
continue
elif max_ratio == -1:
logger.info(util.insert_space(f'Share Limit: Set No Share Limit',4))
torrent.set_share_limits(-1,-1)
continue
else:
max_ratio = torrent.max_ratio
if max_seeding_time:
if max_seeding_time == -2:
logger.info(util.insert_space(f'Share Limit: Use Global Share Limit',4))
torrent.set_share_limits(-2,-2)
continue
elif max_seeding_time == -1:
logger.info(util.insert_space(f'Share Limit: Set No Share Limit',4))
torrent.set_share_limits(-1,-1)
continue
else:
max_seeding_time = torrent.max_seeding_time
if max_ratio != torrent.max_ratio and max_seeding_time != torrent.max_seeding_time:
logger.info(util.insert_space(f'Share Limit: Max Ratio = {max_ratio}, Max Seed Time = {max_seeding_time} min',4))
elif max_ratio != torrent.max_ratio:
logger.info(util.insert_space(f'Share Limit: Max Ratio = {max_ratio}',4))
elif max_seeding_time != torrent.max_seeding_time:
logger.info(util.insert_space(f'Share Limit: Max Seed Time = {max_seeding_time} min',4))
torrent.set_share_limits(max_ratio,max_seeding_time)
if dry_run:
if num_tags >= 1:
logger.dryrun(f'Did not update {num_tags} new tags.')