jubeat_asphyxia: add profile id check

2 files changed, 6 insertions, 3 deletions

diff --git a/jubeat/asphyxia/README.md b/jubeat/asphyxia/README.md
index e889746..8b8db54 100644
--- a/jubeat/asphyxia/README.md
+++ b/jubeat/asphyxia/README.md
@@ -6,3 +6,5 @@
 The official Asphyxia Plugin (up to Festo) also stores data in the same format so it will work.
 
 Pass in your `jubeat@asphyxia.db` as `-f` or `--file`, this can be found in the `savedata` folder of wherever your Asphyxia executable is.
+
+Your Profile ID can be found by going to  `Asphyxia WebUI -> Profile` and copy the ID there for the appropriate profile
diff --git a/jubeat/asphyxia/czeave_to_tachi.py b/jubeat/asphyxia/czeave_to_tachi.py
index da58707..e06b396 100644
--- a/jubeat/asphyxia/czeave_to_tachi.py
+++ b/jubeat/asphyxia/czeave_to_tachi.py
@@ -13,13 +13,13 @@ SEQUENCE_MAP = {
     2: "EXT"
 }
 
-def convert_from_czeave_json_to_tachi_json(file: str, output_path: str, service: str):
+def convert_from_czeave_json_to_tachi_json(file: str, output_path: str, service: str, profile_id: str):
     with open(file, "r", encoding="utf-8") as infile:
         batch_manual = {
             "meta": {"game": "jubeat", "playtype": "Single", "service": service},
         }
         scores = []
-        data = [json.loads(line) for line in infile if '"collection":"score"' in line]
+        data = [json.loads(line) for line in infile if '"collection":"score"' in line and profile_id in line]
         for score in data:
             difficulty = SEQUENCE_MAP[score["seq"]]
             if score["isHardMode"]:
@@ -65,6 +65,7 @@ if __name__ == "__main__":
     parser.add_argument(
         "-o", "--output", help="Output filename", default="czeave_asphyxia_batch_manual.json"
     )
+    parser.add_argument("-p", "--profile", help="Asphyxia Profile ID to export for")
     args = parser.parse_args()
     if args.file is None:
         print("ERROR: Please specify Asphyxia DB file (from savedata folder)")
@@ -73,4 +74,4 @@ if __name__ == "__main__":
         print(f"ERROR: The file {args.file} does not exist.")
         exit(1)
 
-    convert_from_czeave_json_to_tachi_json(args.file, args.output, args.service)
+    convert_from_czeave_json_to_tachi_json(args.file, args.output, args.service, args.profile)