commit

db/init_databases.py

@@ -0,0 +1,100 @@
+#!/usr/bin/env python3
+"""
+SQLite数据库初始化脚本
+自动创建开发环境(ai_mip_dev.db)和生产环境(ai_mip_prod.db)数据库
+"""
+
+import sqlite3
+import os
+from pathlib import Path
+
+# 数据库文件路径
+DB_DIR = Path(__file__).parent
+DEV_DB = DB_DIR / "ai_mip_dev.db"
+PROD_DB = DB_DIR / "ai_mip_prod.db"
+
+# SQL脚本路径
+INIT_SQL = DB_DIR / "init_sqlite.sql"
+SEED_DEV_SQL = DB_DIR / "seed_dev.sql"
+
+
+def execute_sql_file(conn, sql_file):
+    """执行SQL文件"""
+    with open(sql_file, 'r', encoding='utf-8') as f:
+        sql_script = f.read()
+    
+    # SQLite需要逐条执行语句
+    conn.executescript(sql_script)
+    conn.commit()
+    print(f"✓ 已执行: {sql_file.name}")
+
+
+def init_database(db_path, with_seed=False):
+    """初始化数据库"""
+    # 如果数据库已存在，询问是否覆盖
+    if db_path.exists():
+        response = input(f"\n数据库 {db_path.name} 已存在，是否覆盖? (y/n): ").strip().lower()
+        if response != 'y':
+            print(f"跳过 {db_path.name}")
+            return
+        os.remove(db_path)
+        print(f"已删除旧数据库: {db_path.name}")
+    
+    print(f"\n创建数据库: {db_path.name}")
+    
+    # 连接数据库（自动创建）
+    conn = sqlite3.connect(db_path)
+    
+    try:
+        # 执行初始化SQL
+        execute_sql_file(conn, INIT_SQL)
+        
+        # 如果需要，执行种子数据
+        if with_seed:
+            execute_sql_file(conn, SEED_DEV_SQL)
+        
+        print(f"✓ 数据库 {db_path.name} 创建成功")
+        
+        # 验证表是否创建成功
+        cursor = conn.cursor()
+        cursor.execute("SELECT name FROM sqlite_master WHERE type='table'")
+        tables = cursor.fetchall()
+        print(f"  创建的表: {', '.join([t[0] for t in tables])}")
+        
+    except Exception as e:
+        print(f"✗ 创建数据库失败: {str(e)}")
+        raise
+    finally:
+        conn.close()
+
+
+def main():
+    print("=" * 60)
+    print("SQLite数据库初始化工具")
+    print("=" * 60)
+    
+    # 检查SQL文件是否存在
+    if not INIT_SQL.exists():
+        print(f"错误: 找不到初始化脚本 {INIT_SQL}")
+        return
+    
+    # 初始化开发数据库（带测试数据）
+    print("\n[1] 初始化开发环境数据库")
+    init_database(DEV_DB, with_seed=True)
+    
+    # 初始化生产数据库（不带测试数据）
+    print("\n[2] 初始化生产环境数据库")
+    init_database(PROD_DB, with_seed=False)
+    
+    print("\n" + "=" * 60)
+    print("数据库初始化完成")
+    print("=" * 60)
+    print(f"开发数据库: {DEV_DB}")
+    print(f"生产数据库: {PROD_DB}")
+    print("\n使用方法:")
+    print("  开发环境: 在 .env.development 中设置 DATABASE_PATH=db/ai_mip_dev.db")
+    print("  生产环境: 在 .env.production 中设置 DATABASE_PATH=db/ai_mip_prod.db")
+
+
+if __name__ == "__main__":
+    main()