diff options
Diffstat (limited to 'jubeat/asphyxia/czeave_to_tachi.py')
| -rw-r--r-- | jubeat/asphyxia/czeave_to_tachi.py | 7 |
1 files changed, 4 insertions, 3 deletions
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) |