[+] maimai DX Tickets

tools/maimai_convert.py

@@ -3,6 +3,7 @@ from pathlib import Path
 from typing import NamedTuple
 
 import pandas as pd
+import sqlglot
 import xmltodict
 from hypy_utils import write
 
@@ -56,20 +57,55 @@ def parse_event(d: dict) -> Event:
     )
 
 
+def add_migration(f_name: str, mysql: str):
+    (migration_path / 'mysql' / f_name).write_text(mysql)
+    (migration_path / 'mariadb' / f_name).write_text(mysql)
+
+    # Translate to sqlite
+    sqlite = sqlglot.transpile(mysql, read='mysql', write='sqlite', pretty=True)
+    (migration_path / 'sqlite' / f_name).write_text(';\n'.join(sqlite) + ';\n')
+
+
 if __name__ == '__main__':
     agupa = argparse.ArgumentParser(description='Convert maimai data to csv')
     agupa.add_argument('path', type=Path, help='Path to A000 data folder')
     args = agupa.parse_args()
     path = Path(args.path)
+    src = Path(__file__).parent.parent
 
     tickets = read_list('ticket', '*/Ticket.xml', parse_ticket)
 
     events = read_list('event', '*/Event.xml', parse_event)
 
     # Write incremental sql
-    ids = [int(v.split(",")[0]) for v in (Path(__file__).parent / 'maimai2_game_event.csv').read_text().splitlines()]
-    new_events = [e for e in events if e.id not in ids]
-    sql = "INSERT INTO `maimai2_game_event` (`id`, `end_date`, `start_date`, `type`, `enable`) VALUES \n" + \
-        ",\n".join([f"({e.id}, '2029-01-01 00:00:00.000000', '2019-01-01 00:00:00.000000', {e.type}, '1')" for e in new_events])
+    # ids = [int(v.split(",")[0]) for v in (Path(__file__).parent / 'maimai2_game_event.csv').read_text().splitlines()]
+    # new_events = [e for e in events if e.id not in ids]
+    # sql = "INSERT INTO `maimai2_game_event` (`id`, `end_date`, `start_date`, `type`, `enable`) VALUES \n" + \
+    #     ",\n".join([f"({e.id}, '2029-01-01 00:00:00.000000', '2019-01-01 00:00:00.000000', {e.type}, '1')" for e in new_events])
+    # sql += ";\n"
+    # write('sql/maimai2_game_event.sql', sql)
+
+    # Find the highest V{}__*.sql file in src/main/resources/db/migration/sqlite
+    migration_path = src / 'src/main/resources/db/migration'
+    last_sql_version = max([int(v.name[1:].split("__")[0]) for v in (migration_path / 'sqlite').glob('V*__*.sql')])
+    last_sql_version = 248
+    print(f"Last sql version: {last_sql_version}")
+
+    # Write ticket sql
+    sql = """
+CREATE TABLE `maimai2_game_ticket` (
+    `id` bigint(20) NOT NULL,
+    `name` varchar(255) NOT NULL,
+    `credits` int(8) NOT NULL,
+    `kind` varchar(255) NOT NULL,
+    `max` int(16) NOT NULL,
+    `detail` varchar(255) NOT NULL,
+    `event_id` bigint(20) NOT NULL,
+    `event_name` varchar(255) NOT NULL
+);\n\n"""
+    sql += "INSERT INTO `maimai2_game_ticket` (`id`, `name`, `credits`, `kind`, `max`, `detail`, `event_id`, `event_name`) VALUES \n" + \
+        ",\n".join([f"({t.id}, '{t.name}', {t.credits}, '{t.kind}', {t.max}, '{t.detail}', {t.eventId}, '{t.eventName}')" for t in tickets])
     sql += ";\n"
-    write('sql/maimai2_game_event.sql', sql)
+    add_migration(f"V{last_sql_version + 1}__maimai2_tickets.sql", sql)
+
+