Leahnaya/s3s
1
0
Fork 0
mirror of https://github.com/frozenpandaman/s3s.git synced 2026-09-30 13:16:55 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

add argument to control whether download from specific battle type

Browse Source
This commit is contained in:
bhiaibogf
2022-09-12 20:18:19 +08:00
parent c4e4ac4cad
commit 16a1c34218
1 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
s3s.py
View File

@@ -222,7 +222,7 @@ def gen_new_tokens(reason, force=False):
print(f"Wrote tokens for {acc_name} to config.txt.")
def fetch_json(which, separate=False, exportall=False):
def fetch_json(which, separate=False, exportall=False, specific=False):
'''Returns results JSON from SplatNet 3, including a combined dict for ink battles + SR jobs if requested.'''
if exportall and not separate:
@@ -236,10 +236,12 @@ def fetch_json(which, separate=False, exportall=False):
sha_list = []
if which == "both" or which == "ink":
sha_list.append(translate_rid["LatestBattleHistoriesQuery"])
sha_list.append(translate_rid["RegularBattleHistoriesQuery"])
sha_list.append(translate_rid["BankaraBattleHistoriesQuery"])
sha_list.append(translate_rid["PrivateBattleHistoriesQuery"])
if specific:
sha_list.append(translate_rid["RegularBattleHistoriesQuery"])
sha_list.append(translate_rid["BankaraBattleHistoriesQuery"])
sha_list.append(translate_rid["PrivateBattleHistoriesQuery"])
else:
sha_list.append(translate_rid["LatestBattleHistoriesQuery"])
else:
sha_list.append(None)
if which == "both" or which == "salmon":
@@ -817,7 +819,7 @@ def main():
prefetch_checks()
print("\nFetching your JSON files to export locally... this might take a while.")
try:
parents, results, coop_results = fetch_json("both", True, True) # calls prefetch_checks() to gen or check tokens
parents, results, coop_results = fetch_json("both", True, True, True) # calls prefetch_checks() to gen or check tokens
except:
print("Please run the script again.")
sys.exit(1)