refactor: 后端从Node.js重写为Python FastAPI

server/sql/init_db.py

@@ -0,0 +1,93 @@
+"""
+数据库初始化脚本
+在IDEA中直接运行此文件即可建库
+"""
+import os
+import pymysql
+from pathlib import Path
+
+# 获取当前脚本所在目录
+SQL_DIR = Path(__file__).parent
+
+# 数据库配置（从环境变量或默认值）
+DB_CONFIG = {
+    'host': os.getenv('DB_HOST', 'localhost'),
+    'port': int(os.getenv('DB_PORT', 3306)),
+    'user': os.getenv('DB_USER', 'root'),
+    'password': os.getenv('DB_PASSWORD', '123456'),
+    'charset': 'utf8mb4'
+}
+
+
+def read_sql_file(filename: str) -> str:
+    """读取SQL文件"""
+    filepath = SQL_DIR / filename
+    with open(filepath, 'r', encoding='utf-8') as f:
+        return f.read()
+
+
+def execute_sql(cursor, sql: str, description: str):
+    """执行SQL语句（支持多语句）"""
+    print(f'{description}...')
+    # 按分号分割SQL语句
+    statements = [s.strip() for s in sql.split(';') if s.strip()]
+    for stmt in statements:
+        if stmt:
+            try:
+                cursor.execute(stmt)
+            except pymysql.Error as e:
+                # 忽略某些错误（如数据库已存在）
+                if e.args[0] not in [1007, 1050]:  # 数据库已存在、表已存在
+                    print(f'  警告: {e.args[1]}')
+    print(f'  {description}完成！')
+
+
+def init_database():
+    """初始化数据库"""
+    print('=' * 50)
+    print('星域故事汇 - 数据库初始化')
+    print('=' * 50)
+    print(f'连接信息: {DB_CONFIG["user"]}@{DB_CONFIG["host"]}:{DB_CONFIG["port"]}')
+    print()
+
+    # 连接MySQL
+    connection = pymysql.connect(**DB_CONFIG)
+    cursor = connection.cursor()
+
+    try:
+        # 1. 执行schema.sql（创建数据库和表）
+        schema_sql = read_sql_file('schema.sql')
+        execute_sql(cursor, schema_sql, '创建数据库表结构')
+        connection.commit()
+
+        # 2. 执行种子数据第1部分
+        seed1_sql = read_sql_file('seed_stories_part1.sql')
+        execute_sql(cursor, seed1_sql, '导入种子数据（第1部分）')
+        connection.commit()
+
+        # 3. 执行种子数据第2部分
+        seed2_sql = read_sql_file('seed_stories_part2.sql')
+        execute_sql(cursor, seed2_sql, '导入种子数据（第2部分）')
+        connection.commit()
+
+        print()
+        print('=' * 50)
+        print('数据库初始化完成！')
+        print('共创建10个种子故事，包含66个剧情节点和多个结局分支。')
+        print('=' * 50)
+        print()
+        print('现在可以启动后端服务了：')
+        print('  cd "D:\\idea project\\ai_game\\server"')
+        print('  python -m app.main')
+
+    except Exception as e:
+        print(f'初始化失败: {e}')
+        connection.rollback()
+        raise
+    finally:
+        cursor.close()
+        connection.close()
+
+
+if __name__ == '__main__':
+    init_database()