mirror of
https://github.com/ManiMatter/decluttarr.git
synced 2026-04-17 21:53:58 +02:00
2e6973bea4
Fixed typos and various linting issues such as PEP violations. Added ruff and fixed common issues and linting issues.
105 lines
3.8 KiB
Python
105 lines
3.8 KiB
Python
# Cleans the download queue
|
|
from src.jobs.remove_bad_files import RemoveBadFiles
|
|
from src.jobs.remove_failed_downloads import RemoveFailedDownloads
|
|
from src.jobs.remove_failed_imports import RemoveFailedImports
|
|
from src.jobs.remove_metadata_missing import RemoveMetadataMissing
|
|
from src.jobs.remove_missing_files import RemoveMissingFiles
|
|
from src.jobs.remove_orphans import RemoveOrphans
|
|
from src.jobs.remove_slow import RemoveSlow
|
|
from src.jobs.remove_stalled import RemoveStalled
|
|
from src.jobs.remove_unmonitored import RemoveUnmonitored
|
|
from src.jobs.search_handler import SearchHandler
|
|
from src.utils.log_setup import logger
|
|
from src.utils.queue_manager import QueueManager
|
|
|
|
|
|
class JobManager:
|
|
arr = None
|
|
|
|
def __init__(self, settings):
|
|
self.settings = settings
|
|
|
|
async def run_jobs(self, arr):
|
|
self.arr = arr
|
|
await self.removal_jobs()
|
|
await self.search_jobs()
|
|
|
|
async def removal_jobs(self):
|
|
logger.verbose("")
|
|
logger.verbose(f"Cleaning queue on {self.arr.name}:")
|
|
if not await self._queue_has_items():
|
|
return
|
|
|
|
if not await self._qbit_connected():
|
|
return
|
|
|
|
# Refresh trackers
|
|
await self.arr.tracker.refresh_private_and_protected(self.settings)
|
|
|
|
# Execute Cleaning
|
|
removal_jobs = self._get_removal_jobs()
|
|
items_detected = 0
|
|
for removal_job in removal_jobs:
|
|
items_detected += await removal_job.run()
|
|
|
|
if items_detected == 0:
|
|
logger.verbose(">>> Queue is clean.")
|
|
|
|
async def search_jobs(self):
|
|
if (
|
|
self.arr.arr_type == "whisparr"
|
|
): # Whisparr does not support this endpoint (yet?)
|
|
return
|
|
if self.settings.jobs.search_missing_content.enabled:
|
|
await SearchHandler(self.arr, self.settings).handle_search("missing")
|
|
if self.settings.jobs.search_unmet_cutoff_content.enabled:
|
|
await SearchHandler(self.arr, self.settings).handle_search("cutoff")
|
|
|
|
async def _queue_has_items(self):
|
|
queue_manager = QueueManager(self.arr, self.settings)
|
|
full_queue = await queue_manager.get_queue_items("full")
|
|
if full_queue:
|
|
logger.debug(
|
|
"job_runner/full_queue at start: %s",
|
|
queue_manager.format_queue(full_queue),
|
|
)
|
|
return True
|
|
logger.verbose(">>> Queue is empty.")
|
|
return False
|
|
|
|
async def _qbit_connected(self):
|
|
for qbit in self.settings.download_clients.qbittorrent:
|
|
# Check if any client is disconnected
|
|
if not await qbit.check_qbit_connected():
|
|
logger.warning(
|
|
f">>> qBittorrent is disconnected. Skipping queue cleaning on {self.arr.name}.",
|
|
)
|
|
return False
|
|
return True
|
|
|
|
def _get_removal_jobs(self):
|
|
"""
|
|
Return a list of enabled removal job instances based on the provided settings.
|
|
|
|
Each job is included if the corresponding attribute exists and is truthy in settings.jobs.
|
|
"""
|
|
removal_job_classes = {
|
|
"remove_bad_files": RemoveBadFiles,
|
|
"remove_failed_imports": RemoveFailedImports,
|
|
"remove_failed_downloads": RemoveFailedDownloads,
|
|
"remove_metadata_missing": RemoveMetadataMissing,
|
|
"remove_missing_files": RemoveMissingFiles,
|
|
"remove_orphans": RemoveOrphans,
|
|
"remove_slow": RemoveSlow,
|
|
"remove_stalled": RemoveStalled,
|
|
"remove_unmonitored": RemoveUnmonitored,
|
|
}
|
|
|
|
jobs = []
|
|
for removal_job_name, removal_job_class in removal_job_classes.items():
|
|
if getattr(self.settings.jobs, removal_job_name, False):
|
|
jobs.append(
|
|
removal_job_class(self.arr, self.settings, removal_job_name),
|
|
)
|
|
return jobs
|