dl/commands/tinfoil.py

82 lines
1.7 KiB
Python
Raw Normal View History

2023-09-10 15:38:49 +00:00
#!/usr/bin/env python3
import argparse
2023-09-11 07:02:57 +00:00
import os
2023-09-10 15:38:49 +00:00
import re
import subprocess
import sys
import requests
parser = argparse.ArgumentParser()
2023-09-10 15:44:39 +00:00
parser.add_argument("-r", "--remote", type=str, required=True)
2023-09-10 15:38:49 +00:00
args = parser.parse_args()
remote = args.remote
req_titles = requests.get(
"https://raw.githubusercontent.com/blawar/titledb/master/FR.fr.json"
)
json_titles = req_titles.json()
titles = []
for json_title in json_titles:
titles.append(json_titles[json_title]["id"])
print("Titles OK")
2023-09-11 07:02:57 +00:00
proc_bases = subprocess.run(
["rclone", "lsf", f"{remote}:nsz/base"],
2023-09-10 15:38:49 +00:00
check=True,
capture_output=True,
)
2023-09-11 07:02:57 +00:00
bases = proc_bases.stdout.decode().split("\n")
print("Base OK")
2023-09-10 15:38:49 +00:00
proc_dlcs = subprocess.run(
["rclone", "lsf", f"{remote}:nsz/dlc"],
check=True,
capture_output=True,
)
dlcs = proc_dlcs.stdout.decode().split("\n")
print("DLC OK")
proc_updates = subprocess.run(
["rclone", "lsf", f"{remote}:nsz/updates"],
check=True,
capture_output=True,
)
updates = proc_updates.stdout.decode().split("\n")
print("Updates OK")
2023-09-11 07:02:57 +00:00
def download(path, file):
if os.path.isfile(os.path.join(path, file)):
return
subprocess.run(
["rclone", "copy", "-P", "-v", f"{remote}:{path}/{base}", f"{path}/"],
check=True,
stdout=sys.stdout,
)
for base in bases:
re_id = re.search(r"\[([0-9A-F]*)\]", base)
2023-09-10 15:38:49 +00:00
if not re_id:
continue
id = re_id.group(1)
if id not in titles:
continue
2023-09-10 16:36:56 +00:00
id = id[0:-4]
2023-09-10 15:38:49 +00:00
2023-09-11 07:02:57 +00:00
download("nsz/base", base)
2023-09-10 15:38:49 +00:00
for dlc in dlcs:
if id in dlc:
2023-09-11 07:02:57 +00:00
download("nsz/dlc", dlc)
2023-09-10 15:38:49 +00:00
for update in updates:
if id in update:
2023-09-11 07:02:57 +00:00
download("nsz/updates", update)