260710_0

init_db.py

@@ -0,0 +1,51 @@
+"""DB 마이그레이션 실행 스크립트."""
+
+import asyncio
+import sys
+from pathlib import Path
+
+import aiomysql
+
+from config.config_db import DB_HOST, DB_NAME, DB_PASSWORD, DB_PORT, DB_USER
+
+
+async def run_migration():
+    """SQL 마이그레이션 파일 실행."""
+    connection = await aiomysql.connect(
+        host=DB_HOST,
+        port=DB_PORT,
+        user=DB_USER,
+        password=DB_PASSWORD,
+        db=DB_NAME,
+        charset="utf8mb4",
+    )
+
+    try:
+        async with connection.cursor() as cursor:
+            migration_file = Path(__file__).parent / "migrations" / "001_create_upload_tables.sql"
+            sql_content = migration_file.read_text(encoding="utf-8")
+
+            # SQL 파일의 각 명령문 실행 (';'으로 분리)
+            statements = [stmt.strip() for stmt in sql_content.split(";") if stmt.strip()]
+
+            for i, stmt in enumerate(statements, 1):
+                print(f"[{i}/{len(statements)}] executing...")
+                await cursor.execute(stmt)
+                print(f"  OK")
+
+        await connection.commit()
+        print("\nMigration complete!")
+        return True
+
+    except Exception as e:
+        print(f"\nMigration failed: {e}")
+        await connection.rollback()
+        return False
+
+    finally:
+        connection.close()
+
+
+if __name__ == "__main__":
+    success = asyncio.run(run_migration())
+    sys.exit(0 if success else 1)