mirror of
https://github.com/ManiMatter/decluttarr.git
synced 2026-04-21 16:25:35 +02:00
41 lines
2.5 KiB
Python
41 lines
2.5 KiB
Python
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__)
|
|
from src.utils.rest import rest_get
|
|
|
|
async def remove_unmonitored(settings_dict, BASE_URL, API_KEY, NAME, deleted_downloads, defective_tracker, protectedDownloadIDs, privateDowloadIDs, arr_type):
|
|
# Removes downloads belonging to movies/tv shows that are not monitored. Does not add to blocklist
|
|
try:
|
|
failType = 'unmonitored'
|
|
queue = await get_queue(BASE_URL, API_KEY)
|
|
logger.debug('remove_unmonitored/queue IN: %s', formattedQueueInfo(queue))
|
|
if not queue: return 0
|
|
# Find items affected
|
|
monitoredDownloadIDs = []
|
|
for queueItem in queue['records']:
|
|
if arr_type == 'sonarr':
|
|
isMonitored = (await rest_get(f'{BASE_URL}/episode/{str(queueItem["episodeId"])}', API_KEY))['monitored']
|
|
elif arr_type == 'radarr':
|
|
isMonitored = (await rest_get(f'{BASE_URL}/movie/{str(queueItem["movieId"])}', API_KEY))['monitored']
|
|
elif arr_type == 'lidarr':
|
|
isMonitored = (await rest_get(f'{BASE_URL}/album/{str(queueItem["albumId"])}', API_KEY))['monitored']
|
|
elif arr_type == 'readarr':
|
|
isMonitored = (await rest_get(f'{BASE_URL}/book/{str(queueItem["bookId"])}', API_KEY))['monitored']
|
|
if isMonitored:
|
|
monitoredDownloadIDs.append(queueItem['downloadId'])
|
|
|
|
affectedItems = []
|
|
for queueItem in queue['records']:
|
|
if queueItem['downloadId'] not in monitoredDownloadIDs:
|
|
affectedItems.append(queueItem) # One downloadID may be shared by multiple queueItems. Only removes it if ALL queueitems are unmonitored
|
|
|
|
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)
|
|
return len(affectedItems)
|
|
except Exception as error:
|
|
errorDetails(NAME, error)
|
|
return 0 |