diff --git a/upload.py b/upload.py index b1b4d00..dec3aa2 100644 --- a/upload.py +++ b/upload.py @@ -33,6 +33,9 @@ from src.trackers.BHDTV import BHDTV from src.trackers.RTF import RTF from src.trackers.OTW import OTW +from src.trackers.FNP import FNP +from src.trackers.CBR import CBR +from src.trackers.UTP import UTP import json from pathlib import Path import asyncio @@ -246,12 +249,12 @@ async def do_the_thing(base_dir): ####### Upload to Trackers ####### #################################### common = COMMON(config=config) - api_trackers = ['BLU', 'AITHER', 'STC', 'R4E', 'STT', 'RF', 'ACM','LCD','LST','HUNO', 'SN', 'LT', 'NBL', 'ANT', 'JPTV', 'OE', 'BHDTV', 'RTF', 'OTW'] + api_trackers = ['BLU', 'AITHER', 'STC', 'R4E', 'STT', 'RF', 'ACM','LCD','LST','HUNO', 'SN', 'LT', 'NBL', 'ANT', 'JPTV', 'OE', 'BHDTV', 'RTF', 'OTW', 'FNP', 'CBR', 'UTP'] http_trackers = ['HDB', 'TTG', 'FL', 'PTER', 'HDT', 'MTV'] tracker_class_map = { 'BLU' : BLU, 'BHD': BHD, 'AITHER' : AITHER, 'STC' : STC, 'R4E' : R4E, 'THR' : THR, 'STT' : STT, 'HP' : HP, 'PTP' : PTP, 'RF' : RF, 'SN' : SN, 'ACM' : ACM, 'HDB' : HDB, 'LCD': LCD, 'TTG' : TTG, 'LST' : LST, 'HUNO': HUNO, 'FL' : FL, 'LT' : LT, 'NBL' : NBL, 'ANT' : ANT, 'PTER': PTER, 'JPTV' : JPTV, - 'TL' : TL, 'HDT' : HDT, 'MTV': MTV, 'OE': OE, 'BHDTV': BHDTV, 'RTF': RTF, 'OTW': OTW} + 'TL' : TL, 'HDT' : HDT, 'MTV': MTV, 'OE': OE, 'BHDTV': BHDTV, 'RTF': RTF, 'OTW': OTW, 'FNP': FNP, 'CBR': CBR, 'UTP': UTP} for tracker in trackers: if meta['name'].endswith('DUPE?'): @@ -272,6 +275,8 @@ async def do_the_thing(base_dir): console.print(f"Uploading to {tracker_class.tracker}") if check_banned_group(tracker_class.tracker, tracker_class.banned_groups, meta): continue + if tracker == "RTF": + await tracker_class.api_test(meta) dupes = await tracker_class.search_existing(meta) dupes = await common.filter_dupes(dupes, meta) # note BHDTV does not have search implemented.