Skip to content

Commit

Permalink
Merge pull request #1 from noxxusnx/add-shebang
Browse files Browse the repository at this point in the history 
add shebang to upload.py
  • Loading branch information
@edge20200
edge20200 authored May 1, 2024
2 parents 27b64fd + 7748947 commit cbb6bc7
Showing 1 changed file with 26 additions and 24 deletions.
50 changes: 26 additions & 24 deletions upload.py
100644 → 100755
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
#!/usr/bin/env python3

import requests
from src.args import Args
from src.clients import Clients
Expand Down Expand Up @@ -71,8 +73,8 @@
with open(f"{base_dir}/data/config.py", 'w') as f:
f.write(f"config = {json.dumps(json_config, indent=4)}")
f.close()
cli_ui.info(cli_ui.green, "Successfully updated config from .json to .py")
cli_ui.info(cli_ui.green, "It is now safe for you to delete", cli_ui.yellow, "data/config.json", "if you wish")
cli_ui.info(cli_ui.green, "Successfully updated config from .json to .py")
cli_ui.info(cli_ui.green, "It is now safe for you to delete", cli_ui.yellow, "data/config.json", "if you wish")
from data.config import config
else:
raise NotImplementedError
Expand Down Expand Up @@ -124,7 +126,7 @@ async def do_the_thing(base_dir):
console.print("\n\n")
else:
console.print(f"[red]Path: [bold red]{path}[/bold red] does not exist")

elif os.path.exists(os.path.dirname(path)) and len(paths) != 1:
queue = paths
md_text = "\n - ".join(queue)
Expand All @@ -151,7 +153,7 @@ async def do_the_thing(base_dir):
console.print("\n[bold green]Queuing these files:[/bold green]", end='')
console.print(Markdown(f"- {md_text.rstrip()}\n\n", style=Style(color='cyan')))
console.print("\n\n")

else:
# Add Search Here
console.print(f"[red]There was an issue with your input. If you think this was not an issue, please make a report that includes the full command used.")
Expand All @@ -168,8 +170,8 @@ async def do_the_thing(base_dir):
saved_meta = json.load(f)
for key, value in saved_meta.items():
overwrite_list = [
'trackers', 'dupe', 'debug', 'anon', 'category', 'type', 'screens', 'nohash', 'manual_edition', 'imdb', 'tmdb_manual', 'mal', 'manual',
'hdb', 'ptp', 'blu', 'no_season', 'no_aka', 'no_year', 'no_dub', 'no_tag', 'no_seed', 'client', 'desclink', 'descfile', 'desc', 'draft', 'region', 'freeleech',
'trackers', 'dupe', 'debug', 'anon', 'category', 'type', 'screens', 'nohash', 'manual_edition', 'imdb', 'tmdb_manual', 'mal', 'manual',
'hdb', 'ptp', 'blu', 'no_season', 'no_aka', 'no_year', 'no_dub', 'no_tag', 'no_seed', 'client', 'desclink', 'descfile', 'desc', 'draft', 'region', 'freeleech',
'personalrelease', 'unattended', 'season', 'episode', 'torrent_creation', 'qbit_tag', 'qbit_cat', 'skip_imghost_upload', 'imghost', 'manual_source', 'webdv', 'hardcoded-subs'
]
if meta.get(key, None) != value and key in overwrite_list:
Expand All @@ -187,7 +189,7 @@ async def do_the_thing(base_dir):
meta['unattended'] = True
console.print("[yellow]Running in Auto Mode")
prep = Prep(screens=meta['screens'], img_host=meta['imghost'], config=config)
meta = await prep.gather_prep(meta=meta, mode='cli')
meta = await prep.gather_prep(meta=meta, mode='cli')
meta['name_notag'], meta['name'], meta['clean_name'], meta['potential_missing'] = await prep.get_name(meta)

if meta.get('image_list', False) in (False, []) and meta.get('skip_imghost_upload', False) == False:
Expand All @@ -213,7 +215,7 @@ async def do_the_thing(base_dir):
prep.create_torrent(meta, Path(meta['path']), "BASE", meta.get('piece_size_max', 0))
if int(meta.get('randomized', 0)) >= 1:
prep.create_random_torrents(meta['base_dir'], meta['uuid'], meta['randomized'], meta['path'])

if meta.get('trackers', None) != None:
trackers = meta['trackers']
else:
Expand All @@ -223,7 +225,7 @@ async def do_the_thing(base_dir):
with open (f"{meta['base_dir']}/tmp/{meta['uuid']}/meta.json", 'w') as f:
json.dump(meta, f, indent=4)
f.close()
confirm = get_confirmation(meta)
confirm = get_confirmation(meta)
while confirm == False:
# help.print_help()
editargs = cli_ui.ask_string("Input args that need correction e.g.(--tag NTb --category tv --tmdb 12345)")
Expand All @@ -233,16 +235,16 @@ async def do_the_thing(base_dir):
meta, help, before_args = parser.parse(editargs, meta)
# meta = await prep.tmdb_other_meta(meta)
meta['edit'] = True
meta = await prep.gather_prep(meta=meta, mode='cli')
meta = await prep.gather_prep(meta=meta, mode='cli')
meta['name_notag'], meta['name'], meta['clean_name'], meta['potential_missing'] = await prep.get_name(meta)
confirm = get_confirmation(meta)

if isinstance(trackers, list) == False:
trackers = [trackers]
trackers = [s.strip().upper() for s in trackers]
if meta.get('manual', False):
trackers.insert(0, "MANUAL")



####################################
Expand All @@ -252,7 +254,7 @@ async def do_the_thing(base_dir):
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,
'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, 'FNP': FNP, 'CBR': CBR, 'UTP': UTP}

Expand All @@ -264,7 +266,7 @@ async def do_the_thing(base_dir):
debug = "(DEBUG)"
else:
debug = ""

if tracker in api_trackers:
tracker_class = tracker_class_map[tracker](config=config)
if meta['unattended']:
Expand All @@ -286,7 +288,7 @@ async def do_the_thing(base_dir):
if tracker == 'SN':
await asyncio.sleep(16)
await client.add_to_client(meta, tracker_class.tracker)

if tracker in http_trackers:
tracker_class = tracker_class_map[tracker](config=config)
if meta['unattended']:
Expand All @@ -306,7 +308,7 @@ async def do_the_thing(base_dir):
await client.add_to_client(meta, tracker_class.tracker)

if tracker == "MANUAL":
if meta['unattended']:
if meta['unattended']:
do_manual = True
else:
do_manual = cli_ui.ask_yes_no(f"Get files for manual upload?", default=True)
Expand All @@ -324,7 +326,7 @@ async def do_the_thing(base_dir):
console.print(f"[yellow]Unable to upload prep files, they can be found at `tmp/{meta['uuid']}")
else:
console.print(f"[green]{meta['name']}")
console.print(f"[green]Files can be found at: [yellow]{url}[/yellow]")
console.print(f"[green]Files can be found at: [yellow]{url}[/yellow]")

if tracker == "BHD":
bhd = BHD(config=config)
Expand All @@ -347,7 +349,7 @@ async def do_the_thing(base_dir):
if meta['upload'] == True:
await bhd.upload(meta)
await client.add_to_client(meta, "BHD")

if tracker == "THR":
if meta['unattended']:
upload_to_thr = True
Expand Down Expand Up @@ -425,7 +427,7 @@ async def do_the_thing(base_dir):
if check_banned_group(tracker_class.tracker, tracker_class.banned_groups, meta):
continue
await tracker_class.upload(meta)
await client.add_to_client(meta, tracker_class.tracker)
await client.add_to_client(meta, tracker_class.tracker)


def get_confirmation(meta):
Expand Down Expand Up @@ -466,7 +468,7 @@ def get_confirmation(meta):
if meta.get('unattended', False) == False:
get_missing(meta)
ring_the_bell = "\a" if config['DEFAULT'].get("sfx_on_prompt", True) == True else "" # \a rings the bell
cli_ui.info_section(cli_ui.yellow, f"Is this correct?{ring_the_bell}")
cli_ui.info_section(cli_ui.yellow, f"Is this correct?{ring_the_bell}")
cli_ui.info(f"Name: {meta['name']}")
confirm = cli_ui.ask_yes_no("Correct?", default=False)
else:
Expand All @@ -477,10 +479,10 @@ def get_confirmation(meta):
def dupe_check(dupes, meta):
if not dupes:
console.print("[green]No dupes found")
meta['upload'] = True
meta['upload'] = True
return meta
else:
console.print()
console.print()
dupe_text = "\n".join(dupes)
console.print()
cli_ui.info_section(cli_ui.bold, "Are these dupes?")
Expand Down Expand Up @@ -547,7 +549,7 @@ def get_missing(meta):
for each in meta['potential_missing']:
if str(meta.get(each, '')).replace(' ', '') in ["", "None", "0"]:
if each == "imdb_id":
each = 'imdb'
each = 'imdb'
missing.append(f"--{each} | {info_notes.get(each)}")
if missing != []:
cli_ui.info_section(cli_ui.yellow, "Potentially missing information:")
Expand All @@ -572,4 +574,4 @@ def get_missing(meta):
loop.run_until_complete(do_the_thing(base_dir))
else:
asyncio.run(do_the_thing(base_dir))

0 comments on commit cbb6bc7

Please sign in to comment.