Skip to content

Commit

Permalink
Update upload.py
Browse files Browse the repository at this point in the history
  • Loading branch information
edge20200 authored Apr 30, 2024
1 parent e04a58a commit 526eb49
Showing 1 changed file with 7 additions and 2 deletions.
9 changes: 7 additions & 2 deletions upload.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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?'):
Expand All @@ -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.
Expand Down

0 comments on commit 526eb49

Please sign in to comment.