Major refactor & code-cleanup

Improved Slow Downloads (more accurate determining of progress)
Added possiblity to exclude private Trackers from cleaning
This commit is contained in:
Benjamin Harder
2024-02-18 18:59:32 +01:00
parent 5ef85d1ac5
commit 34842a4d0c
16 changed files with 502 additions and 336 deletions

34
src/remove_orphans.py Normal file
View File

@@ -0,0 +1,34 @@
from src.utils.shared import (errorDetails, formattedQueueInfo, get_queue, privateTrackerCheck, protectedDownloadCheck, execute_checks, permittedAttemptsCheck, remove_download)
import sys, os, traceback
import logging, verboselogs
logger = verboselogs.VerboseLogger(__name__)
async def remove_orphans(settings_dict, BASE_URL, API_KEY, NAME, deleted_downloads, defective_tracker, protectedDownloadIDs, privateDowloadIDs, full_queue_param):
# Removes downloads belonging to movies/tv shows that have been deleted in the meantime. Does not add to blocklist
try:
failType = 'orphan'
full_queue = await get_queue(BASE_URL, API_KEY, params = {full_queue_param: True})
if not full_queue: return 0 # By now the queue may be empty
queue = await get_queue(BASE_URL, API_KEY)
logger.debug('remove_orphans/full queue IN: %s', str(full_queue))
logger.debug('remove_orphans/queue IN: %s', str(queue))
# Find items affected
# 1. create a list of the "known" queue items
queueIDs = [queueItem['id'] for queueItem in queue['records']] if queue else []
affectedItems = []
# 2. compare all queue items against the known ones, and those that are not found are the "unknown" or "orphan" ones
for queueItem in full_queue['records']:
if queueItem['id'] not in queueIDs:
affectedItems.append(queueItem)
affectedItems = await execute_checks(settings_dict, affectedItems, failType, BASE_URL, API_KEY, NAME, deleted_downloads, defective_tracker, privateDowloadIDs, protectedDownloadIDs,
addToBlocklist = False,
doPrivateTrackerCheck = True,
doProtectedDownloadCheck = True,
doPermittedAttemptsCheck = False)
logger.debug('remove_orphans/full queue OUT: %s', str(await get_queue(BASE_URL, API_KEY, params = {full_queue_param: True})))
return len(affectedItems)
except Exception as error:
errorDetails(NAME, error)
return 0