refactor

backend/app/db/migrate_multi_user.py

@@ -11,6 +11,7 @@ from app.auth.tokens import hash_token
 from app.character.card import DEFAULT_CARD, normalize_card
 from app.config import get_settings
 from app.db.base import engine
+from app.db.models import CharacterCard, User
 
 logger = logging.getLogger(__name__)
 
@@ -55,39 +56,11 @@ def _add_column_if_missing(table: str, column: str, ddl: str) -> None:
 
 
 def _ensure_users_table() -> None:
-    if _table_exists("users"):
-        return
-    with engine.begin() as conn:
-        conn.execute(
-            text(
-                "CREATE TABLE users ("
-                "id INTEGER PRIMARY KEY AUTOINCREMENT, "
-                "username VARCHAR(64) NOT NULL UNIQUE, "
-                "display_name VARCHAR(255) DEFAULT '', "
-                "api_token_hash VARCHAR(64) NOT NULL, "
-                "is_active BOOLEAN DEFAULT 1, "
-                "created_at DATETIME DEFAULT CURRENT_TIMESTAMP"
-                ")"
-            )
-        )
-        conn.execute(text("CREATE INDEX IF NOT EXISTS ix_users_api_token_hash ON users (api_token_hash)"))
+    User.__table__.create(engine, checkfirst=True)
 
 
 def _ensure_character_cards_table() -> None:
-    if _table_exists("character_cards"):
-        return
-    with engine.begin() as conn:
-        conn.execute(
-            text(
-                "CREATE TABLE character_cards ("
-                "id INTEGER PRIMARY KEY AUTOINCREMENT, "
-                "user_id INTEGER NOT NULL UNIQUE REFERENCES users(id) ON DELETE CASCADE, "
-                "card_json TEXT DEFAULT '{}', "
-                "updated_at DATETIME DEFAULT CURRENT_TIMESTAMP"
-                ")"
-            )
-        )
-        conn.execute(text("CREATE INDEX IF NOT EXISTS ix_character_cards_user_id ON character_cards (user_id)"))
+    CharacterCard.__table__.create(engine, checkfirst=True)
 
 
 def _add_user_id_columns() -> None: