mirror of
https://github.com/StuffAnThings/qbit_manage.git
synced 2025-09-17 18:45:34 +08:00
131 lines
7 KiB
Python
131 lines
7 KiB
Python
from datetime import timedelta
|
|
|
|
from modules import util
|
|
|
|
logger = util.logger
|
|
|
|
|
|
class ReCheck:
|
|
def __init__(self, qbit_manager):
|
|
self.qbt = qbit_manager
|
|
self.config = qbit_manager.config
|
|
self.client = qbit_manager.client
|
|
self.stats_resumed = 0
|
|
self.stats_rechecked = 0
|
|
|
|
self.torrents_updated_recheck = [] # List of torrents updated
|
|
self.notify_attr_recheck = [] # List of single torrent attributes to send to notifiarr
|
|
self.torrents_updated_resume = [] # List of torrents updated
|
|
self.notify_attr_resume = [] # List of single torrent attributes to send to notifiarr
|
|
|
|
self.recheck()
|
|
self.config.webhooks_factory.notify(self.torrents_updated_resume, self.notify_attr_resume, group_by="tag")
|
|
self.config.webhooks_factory.notify(self.torrents_updated_recheck, self.notify_attr_recheck, group_by="tag")
|
|
|
|
def recheck(self):
|
|
"""Function used to recheck paused torrents sorted by size and resume torrents that are completed"""
|
|
if self.config.commands["recheck"]:
|
|
logger.separator("Rechecking Paused Torrents", space=False, border=False)
|
|
# sort by size and paused
|
|
torrent_list = self.qbt.get_torrents({"status_filter": "paused", "sort": "size"})
|
|
if torrent_list:
|
|
for torrent in torrent_list:
|
|
tracker = self.qbt.get_tags(torrent.trackers)
|
|
t_name = torrent.name
|
|
t_category = torrent.category
|
|
# Resume torrent if completed
|
|
if torrent.progress == 1:
|
|
if torrent.max_ratio < 0 and torrent.max_seeding_time < 0:
|
|
self.stats_resumed += 1
|
|
body = logger.print_line(
|
|
f"{'Not Resuming' if self.config.dry_run else 'Resuming'} [{tracker['tag']}] - {t_name}",
|
|
self.config.loglevel,
|
|
)
|
|
attr = {
|
|
"function": "recheck",
|
|
"title": "Resuming Torrent",
|
|
"body": body,
|
|
"torrents": [t_name],
|
|
"torrent_tag": tracker["tag"],
|
|
"torrent_category": t_category,
|
|
"torrent_tracker": tracker["url"],
|
|
"notifiarr_indexer": tracker["notifiarr"],
|
|
}
|
|
self.torrents_updated_resume.append(t_name)
|
|
self.notify_attr_resume.append(attr)
|
|
if not self.config.dry_run:
|
|
torrent.resume()
|
|
else:
|
|
# Check to see if torrent meets AutoTorrentManagement criteria
|
|
logger.debug("DEBUG: Torrent to see if torrent meets AutoTorrentManagement Criteria")
|
|
logger.debug(logger.insert_space(f"- Torrent Name: {t_name}", 2))
|
|
logger.debug(
|
|
logger.insert_space(f"-- Ratio vs Max Ratio: {torrent.ratio:.2f} < {torrent.max_ratio:.2f}", 4)
|
|
)
|
|
logger.debug(
|
|
logger.insert_space(
|
|
(
|
|
f"-- Seeding Time vs Max Seed Time: {timedelta(seconds=torrent.seeding_time)} < "
|
|
f"{timedelta(minutes=torrent.max_seeding_time)}"
|
|
),
|
|
4,
|
|
)
|
|
)
|
|
if (
|
|
(torrent.max_ratio >= 0 and torrent.ratio < torrent.max_ratio and torrent.max_seeding_time < 0)
|
|
or (
|
|
torrent.max_seeding_time >= 0
|
|
and (torrent.seeding_time < (torrent.max_seeding_time * 60))
|
|
and torrent.max_ratio < 0
|
|
)
|
|
or (
|
|
torrent.max_ratio >= 0
|
|
and torrent.max_seeding_time >= 0
|
|
and torrent.ratio < torrent.max_ratio
|
|
and (torrent.seeding_time < (torrent.max_seeding_time * 60))
|
|
)
|
|
):
|
|
self.stats_resumed += 1
|
|
body = logger.print_line(
|
|
f"{'Not Resuming' if self.config.dry_run else 'Resuming'} [{tracker['tag']}] - {t_name}",
|
|
self.config.loglevel,
|
|
)
|
|
attr = {
|
|
"function": "recheck",
|
|
"title": "Resuming Torrent",
|
|
"body": body,
|
|
"torrents": [t_name],
|
|
"torrent_tag": tracker["tag"],
|
|
"torrent_category": t_category,
|
|
"torrent_tracker": tracker["url"],
|
|
"notifiarr_indexer": tracker["notifiarr"],
|
|
}
|
|
self.torrents_updated_resume.append(t_name)
|
|
self.notify_attr_resume.append(attr)
|
|
if not self.config.dry_run:
|
|
torrent.resume()
|
|
# Recheck
|
|
elif (
|
|
torrent.progress == 0
|
|
and self.qbt.torrentinfo[t_name]["is_complete"]
|
|
and not torrent.state_enum.is_checking
|
|
):
|
|
self.stats_rechecked += 1
|
|
body = logger.print_line(
|
|
f"{'Not Rechecking' if self.config.dry_run else 'Rechecking'} [{tracker['tag']}] - {t_name}",
|
|
self.config.loglevel,
|
|
)
|
|
attr = {
|
|
"function": "recheck",
|
|
"title": "Rechecking Torrent",
|
|
"body": body,
|
|
"torrents": [t_name],
|
|
"torrent_tag": tracker["tag"],
|
|
"torrent_category": t_category,
|
|
"torrent_tracker": tracker["url"],
|
|
"notifiarr_indexer": tracker["notifiarr"],
|
|
}
|
|
self.torrents_updated_recheck.append(t_name)
|
|
self.notify_attr_recheck.append(attr)
|
|
if not self.config.dry_run:
|
|
torrent.recheck()
|