refactor: database initialize checker

core/database/init_checker.py

@@ -0,0 +1,30 @@
+from enum import Flag, auto
+from pathlib import Path
+
+from arcaea_offline.database import Database
+
+from .utils import create_engine, db_path_to_sqlite_url
+
+
+class DatabaseInitCheckResult(Flag):
+    NONE = 0
+    FILE_EXISTS = auto()
+    INITIALIZED = auto()
+
+    OK = FILE_EXISTS | INITIALIZED
+
+
+def check_db_init(file: Path) -> DatabaseInitCheckResult:
+    flags = DatabaseInitCheckResult.NONE
+
+    if not file.exists():
+        return flags
+
+    flags |= DatabaseInitCheckResult.FILE_EXISTS
+
+    db_url = db_path_to_sqlite_url(file)
+    db = Database(create_engine(db_url))
+    if db.check_init():
+        flags |= DatabaseInitCheckResult.INITIALIZED
+
+    return flags