mirror of
https://github.com/StuffAnThings/qbit_manage.git
synced 2025-10-10 05:46:55 +08:00
8478409027
* 4.2.1-develop1 * Adds #747 * Fixes #764 * Replaced pre-commit hooks flake8, black, pyupgrade, etc. with ruff for linting and formatting. * Bump ruff from 0.9.10 to 0.10.0 (#767) Bumps [ruff](https://github.com/astral-sh/ruff) from 0.9.10 to 0.10.0. - [Release notes](https://github.com/astral-sh/ruff/releases) - [Changelog](https://github.com/astral-sh/ruff/blob/main/CHANGELOG.md) - [Commits](https://github.com/astral-sh/ruff/compare/0.9.10...0.10.0) --- updated-dependencies: - dependency-name: ruff dependency-type: direct:development update-type: version-update:semver-minor ... Signed-off-by: dependabot[bot] <support@github.com> Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com> * 4.2.1 --------- Signed-off-by: dependabot[bot] <support@github.com> Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
130 lines
7 KiB
Python
130 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
|
|
# List of single torrent attributes to send to notifiarr
|
|
self.notify_attr_recheck = []
|
|
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(self.qbt.get_tracker_urls(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": ", ".join(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} vs {torrent.max_ratio:.2f}", 4)
|
|
)
|
|
logger.debug(
|
|
logger.insert_space(
|
|
f"-- Seeding Time vs Max Seed Time: {str(timedelta(seconds=torrent.seeding_time))} vs "
|
|
f"{str(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": ", ".join(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": ", ".join(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()
|