mai2: add aquadx to tachi json script

1 files changed, 119 insertions, 0 deletions

diff --git a/mai2/aquadx/mai_aquadx_to_tachi.py b/mai2/aquadx/mai_aquadx_to_tachi.py
new file mode 100644
index 0000000..475d837
--- /dev/null
+++ b/mai2/aquadx/mai_aquadx_to_tachi.py
@@ -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)