From e8af14132f86f086699403a86c35bee4e7f0ac03 Mon Sep 17 00:00:00 2001 From: Pinapelz Date: Tue, 1 Apr 2025 16:55:40 -0700 Subject: aquadx_ongeki: add pulling ongeki playdata using token from AquaNet --- ongeki/aquadx/ongeki_aquadx_to_tachi.py | 55 +++++++++++++++++++++++---------- 1 file changed, 39 insertions(+), 16 deletions(-) (limited to 'ongeki') diff --git a/ongeki/aquadx/ongeki_aquadx_to_tachi.py b/ongeki/aquadx/ongeki_aquadx_to_tachi.py index 70d7c19..a56d16a 100644 --- a/ongeki/aquadx/ongeki_aquadx_to_tachi.py +++ b/ongeki/aquadx/ongeki_aquadx_to_tachi.py @@ -1,6 +1,7 @@ import argparse import json import urllib.request +import os ONGEKI_AQUADX_JSON = "https://aquadx.net/d/ongeki/00/all-music.json" @@ -16,17 +17,7 @@ headers = { "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/119.0.0.0 Safari/537.36" } -def convert_from_aquadx_json_to_tachi_json(input_json: str, output_file: str, service: str, music_json: str): - if music_json == "online": - req = urllib.request.Request(ONGEKI_AQUADX_JSON, headers=headers) - with urllib.request.urlopen(req) as response: - music_json = json.load(response) - else: - with open(music_json, "r", encoding="utf-8") as file: - music_json = json.load(file) - - with open(input_json, "r", encoding="utf-8") as f: - raw_data = json.load(f) +def convert_from_aquadx_json_to_tachi_json(raw_data: str, output_file: str, service: str, music_json: str): batch_manual = { "meta": {"game": "ongeki", "playtype": "Single", "service": service}, @@ -112,16 +103,48 @@ if __name__ == "__main__": description="Converts AquaDX API JSON for O.N.G.E.K.I to Tachi compatible JSON", epilog="damage, fast, slow, unavailable on the webui" ) - parser.add_argument("input_file", help="Path to the input JSON file exported from AquaDX") + parser.add_argument("-f", "--file", help="Manual. Specify path to the input exported score JSON file exported from AquaDX", required=False) parser.add_argument( "-s", "--service", help="Service description to be shown on Tachi (Note for where this score came from)", - default="AquaDX O.N.G.E.K.I Import (API JSON)", + default="AquaDX Chuni Import", ) + parser.add_argument("-t", "--token", help="Use AquaNet Token to directly grab data from API. Get it from Network tab in your browser and check the API request it makes ?token=???", required=False) + parser.add_argument("-u", "--url", help="AquaNet API endpoint. No need to use this unless you self-host AquaDX", default="https://aquadx.net/aqua") parser.add_argument( - "-o", "--output", help="Output filename", default="aquadx_ongeki_tachi.json" + "-o", "--output", help="Output filename", default="aquadx_mai2_tachi.json" ) - parser.add_argument("--music", "--music-file", help="JSON file containing the mappings of song names to IDs. Check README for moe info", default="online") + parser.add_argument("-m", "--music", help="all-music.json from AquaNet that maps song id to name (required for Tachi). It will automatically pull from main AquaDX if not specified", default="online") + args = parser.parse_args() + # Some checks to make sure input is valid + if args.token is None and args.file is None: + print("ERROR: No valid input method specified. You must specify either --file or --token") + exit(1) + aquadx_url = args.url + if not aquadx_url.startswith("https://") and not aquadx_url.startswith("http://"): + aquadx_url = "https://" + aquadx_url + + + if args.file is not None: + print("An input file has been specified, using local file as input") + if not os.path.exists(args.file): + print(f"ERROR: The file {args.file} does not exist.") + exit(1) + with open(args.file, "r", encoding="utf-8") as f: + raw_data = json.load(f) + else: + print("Pulling mu3/ONGEKI playdata from remote AquaDX at: " + aquadx_url) + req = urllib.request.Request(aquadx_url+"/api/v2/game/ongeki/user-summary?username=pinapelz&token="+args.token, headers=headers) + with urllib.request.urlopen(req) as response: + raw_data = json.load(response) + + if args.music == "online": + req = urllib.request.Request(ONGEKI_AQUADX_JSON, headers=headers) + with urllib.request.urlopen(req) as response: + music_json = json.load(response) + else: + with open(args.music, "r", encoding="utf-8") as file: + music_json = json.load(file) args = parser.parse_args() - convert_from_aquadx_json_to_tachi_json(args.input_file, args.output, args.service, args.music) + convert_from_aquadx_json_to_tachi_json(raw_data, args.output, args.service, music_json) -- cgit v1.2.3 From f9c6fd12b218613ccf7877720a05677301a715e9 Mon Sep 17 00:00:00 2001 From: Pinapelz Date: Tue, 1 Apr 2025 17:00:25 -0700 Subject: aquadx_wacca: add pulling wacca playdata using token from AquaNet --- ongeki/aquadx/ongeki_aquadx_to_tachi.py | 2 +- wacca/aquadx/wacca_aquadx_to_tachi.py | 54 +++++++++++++++++++++++---------- 2 files changed, 39 insertions(+), 17 deletions(-) (limited to 'ongeki') diff --git a/ongeki/aquadx/ongeki_aquadx_to_tachi.py b/ongeki/aquadx/ongeki_aquadx_to_tachi.py index a56d16a..f2250fd 100644 --- a/ongeki/aquadx/ongeki_aquadx_to_tachi.py +++ b/ongeki/aquadx/ongeki_aquadx_to_tachi.py @@ -113,7 +113,7 @@ if __name__ == "__main__": parser.add_argument("-t", "--token", help="Use AquaNet Token to directly grab data from API. Get it from Network tab in your browser and check the API request it makes ?token=???", required=False) parser.add_argument("-u", "--url", help="AquaNet API endpoint. No need to use this unless you self-host AquaDX", default="https://aquadx.net/aqua") parser.add_argument( - "-o", "--output", help="Output filename", default="aquadx_mai2_tachi.json" + "-o", "--output", help="Output filename", default="aquadx_ongeki_tachi.json" ) parser.add_argument("-m", "--music", help="all-music.json from AquaNet that maps song id to name (required for Tachi). It will automatically pull from main AquaDX if not specified", default="online") args = parser.parse_args() diff --git a/wacca/aquadx/wacca_aquadx_to_tachi.py b/wacca/aquadx/wacca_aquadx_to_tachi.py index 6a2c763..e1af59c 100644 --- a/wacca/aquadx/wacca_aquadx_to_tachi.py +++ b/wacca/aquadx/wacca_aquadx_to_tachi.py @@ -2,9 +2,10 @@ import argparse import json import urllib.request import pytz +import os from datetime import datetime -ONGEKI_AQUADX_JSON = "https://aquadx.net/d/wacca/00/all-music.json" +WACCA_AQUADX_JSON = "https://aquadx.net/d/wacca/00/all-music.json" DIFFICULTY_MAPPING = { 0: "NORMAL", @@ -18,17 +19,6 @@ headers = { } def convert_to_aquadx_json_to_tachi_json(input_json: str, output_file: str, service: str, music_json: str): - if music_json == "online": - req = urllib.request.Request(ONGEKI_AQUADX_JSON, headers=headers) - with urllib.request.urlopen(req) as response: - music_json = json.load(response) - else: - with open(music_json, "r", encoding="utf-8") as file: - music_json = json.load(file) - - with open(input_json, "r", encoding="utf-8") as f: - raw_data = json.load(f) - batch_manual = { "meta": {"game": "wacca", "playtype": "Single", "service": service}, "scores": [], @@ -107,16 +97,48 @@ if __name__ == "__main__": description="Converts AquaDX API JSON for WACCA to Tachi compatible JSON", epilog="damage, fast, slow, unavailable on the webui" ) - parser.add_argument("input_file", help="Path to the input JSON file exported from AquaDX") + parser.add_argument("-f", "--file", help="Manual. Specify path to the input exported score JSON file exported from AquaDX", required=False) parser.add_argument( "-s", "--service", help="Service description to be shown on Tachi (Note for where this score came from)", - default="AquaDX WACCA Import (API JSON)", + default="AquaDX Chuni Import", ) + parser.add_argument("-t", "--token", help="Use AquaNet Token to directly grab data from API. Get it from Network tab in your browser and check the API request it makes ?token=???", required=False) + parser.add_argument("-u", "--url", help="AquaNet API endpoint. No need to use this unless you self-host AquaDX", default="https://aquadx.net/aqua") parser.add_argument( "-o", "--output", help="Output filename", default="aquadx_wacca_tachi.json" ) - parser.add_argument("--music", "--music-file", help="JSON file containing the mappings of song names to IDs. Check README for moe info", default="online") + parser.add_argument("-m", "--music", help="all-music.json from AquaNet that maps song id to name (required for Tachi). It will automatically pull from main AquaDX if not specified", default="online") + args = parser.parse_args() + # Some checks to make sure input is valid + if args.token is None and args.file is None: + print("ERROR: No valid input method specified. You must specify either --file or --token") + exit(1) + aquadx_url = args.url + if not aquadx_url.startswith("https://") and not aquadx_url.startswith("http://"): + aquadx_url = "https://" + aquadx_url + + + if args.file is not None: + print("An input file has been specified, using local file as input") + if not os.path.exists(args.file): + print(f"ERROR: The file {args.file} does not exist.") + exit(1) + with open(args.file, "r", encoding="utf-8") as f: + raw_data = json.load(f) + else: + print("Pulling WACCA playdata from remote AquaDX at: " + aquadx_url) + req = urllib.request.Request(aquadx_url+"/api/v2/game/wacca/user-summary?username=pinapelz&token="+args.token, headers=headers) + with urllib.request.urlopen(req) as response: + raw_data = json.load(response) + + if args.music == "online": + req = urllib.request.Request(WACCA_AQUADX_JSON, headers=headers) + with urllib.request.urlopen(req) as response: + music_json = json.load(response) + else: + with open(args.music, "r", encoding="utf-8") as file: + music_json = json.load(file) args = parser.parse_args() - convert_to_aquadx_json_to_tachi_json(args.input_file, args.output, args.service, args.music) + convert_to_aquadx_json_to_tachi_json(raw_data, args.output, args.service, music_json) -- cgit v1.2.3 From 5baf70249d765e12172d1971a898dc105c0963de Mon Sep 17 00:00:00 2001 From: Pinapelz Date: Tue, 1 Apr 2025 17:02:35 -0700 Subject: aquadx_ongeki: use userPlayDate as timestamp --- ongeki/aquadx/ongeki_aquadx_to_tachi.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'ongeki') diff --git a/ongeki/aquadx/ongeki_aquadx_to_tachi.py b/ongeki/aquadx/ongeki_aquadx_to_tachi.py index f2250fd..3280c3f 100644 --- a/ongeki/aquadx/ongeki_aquadx_to_tachi.py +++ b/ongeki/aquadx/ongeki_aquadx_to_tachi.py @@ -2,6 +2,8 @@ import argparse import json import urllib.request import os +import pytz +from datetime import datetime ONGEKI_AQUADX_JSON = "https://aquadx.net/d/ongeki/00/all-music.json" @@ -55,7 +57,8 @@ def convert_from_aquadx_json_to_tachi_json(raw_data: str, output_file: str, serv bell_lamp = "FULL BELL" if is_full_bell else "NONE" - timestamp = entry.get("sortNumber", None) + dt = datetime.fromisoformat(entry.get("userPlayDate").rstrip("Z")).replace(tzinfo=pytz.UTC) + timestamp = int(dt.timestamp()) combo = entry.get("maxCombo",0) bell_count = entry.get("bellCount",0) total_bell_count = entry.get("totalBellCount", 0) -- cgit v1.2.3