mai2: add aquadx to tachi json script

This commit is contained in:
Pinapelz
2025-02-28 02:18:02 -08:00
parent 5d26d7a3f0
commit b6aa2b4a9d
5 changed files with 70625 additions and 0 deletions
+1
View File
@@ -0,0 +1 @@
`chuni_aquadx_to_tachi.py` is for converting the user data exported JSON from AquaNet. It does not require a connection to the DB
+3
View File
@@ -0,0 +1,3 @@
This is for maimai DX
Requries AquaDX music file, it should grab the latest one from AquaNet but if it does not there is a backup included as `all-music.json`
File diff suppressed because it is too large Load Diff
+119
View File
@@ -0,0 +1,119 @@
import argparse
import json
import urllib.request
MAI2_AQUADX_JSON = "https://aquadx.net/d/mai2/00/all-music.json"
DIFFICULTY_MAPPING = {
0: "DX Basic",
1: "DX Advanced",
2: "DX Expert",
3: "DX Master",
4: "DX Re:Master"
}
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_chuni_aquadx_json_to_tachi_json(input_json: str, output_file: str, service: str, music_json: str):
if music_json == "online":
req = urllib.request.Request(MAI2_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": "maimaidx", "playtype": "Single", "service": service},
"scores": [],
}
processed_count = 0
skipped_count = 0
if "userPlaylogList" in raw_data:
for entry in raw_data["userPlaylogList"]:
level = entry.get("level", 0)
processed_count += 1
song_title = music_json[str(entry["musicId"])]["name"]
raw_score_value = str(entry.get("achievement", 0))
score_value = float(f"{raw_score_value[:2]}.{raw_score_value[2]}")
is_clear = entry.get("isClear", False)
is_full_combo = entry.get("isFullCombo", False)
is_all_perfect = entry.get("isAllPerfect", False)
lamp = "FAILED"
if is_all_perfect:
lamp = "ALL PERFECT"
elif is_full_combo:
lamp = "FULL COMBO"
elif is_clear:
lamp = "CLEAR"
timestamp = entry.get("loginDate", None)
combo = entry.get("maxCombo",0)
fast = entry.get("fastCount", 0)
slow = entry.get("lateCount", 0)
pcrit = entry.get("tapCriticalPerfect", 0) + entry.get("holdCriticalPerfect", 0) + entry.get("slideCriticalPerfect", 0) + entry.get("touchCriticalPerfect", 0) + entry.get("breakCriticalPerfect", 0)
perfect = entry.get("tapPerfect", 0) + entry.get("holdPerfect", 0) + entry.get("slidePerfect", 0) + entry.get("touchPerfect", 0) + entry.get("breakPerfect", 0)
great = entry.get("tapGreat", 0) + entry.get("holdGreat", 0) + entry.get("slideGreat", 0) + entry.get("touchGreat", 0) + entry.get("breakGreat", 0)
good = entry.get("tapGood", 0) + entry.get("holdGood", 0) + entry.get("slideGood", 0) + entry.get("touchGood", 0) + entry.get("breakGood", 0)
miss = entry.get("tapMiss", 0) + entry.get("holdMiss", 0) + entry.get("slideMiss", 0) + entry.get("touchMiss", 0) + entry.get("breakMiss", 0)
score_entry = {
"percent": score_value,
"lamp": lamp,
"matchType": "songTitle",
"identifier": str(song_title),
"difficulty": DIFFICULTY_MAPPING[level],
"timeAchieved": timestamp * 1000 if timestamp else None,
"judgements": {
"pcrit": pcrit,
"perfect": perfect,
"great": great,
"good": good,
"miss": miss
},
"optional": {
"maxCombo": combo,
"fast": fast,
"slow": slow
},
}
batch_manual["scores"].append(score_entry)
with open(output_file, "w", encoding="utf-8") as f:
print("--- Processing Summary ---")
print(f"Total scores processed: {processed_count}")
print(f"Scores skipped (level 5 or invalid): {skipped_count}")
print(f"Output saved to {output_file}")
json.dump(batch_manual, f, indent=4, ensure_ascii=False)
if __name__ == "__main__":
parser = argparse.ArgumentParser(
prog="mai_aquadx_to_tachi.py",
description="Converts AquaDX score data for maimai (DX) to Tachi compatible JSON",
epilog="Dan and matchingClass not implemented. Please just use other methods for that kind of static info"
)
parser.add_argument("input_file", help="Path to the input JSON file exported from AquaDX")
parser.add_argument(
"-s",
"--service",
help="Service description to be shown on Tachi (Note for where this score came from)",
default="AquaDX maimaidx Import",
)
parser.add_argument(
"-o", "--output", help="Output filename", default="aquadx_mai_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")
args = parser.parse_args()
convert_chuni_aquadx_json_to_tachi_json(args.input_file, args.output, args.service, args.music)