94 lines
2.8 KiB
Python
94 lines
2.8 KiB
Python
|
"""
|
|||
|
|
数据库初始化脚本
|
|||
|
|
在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()
|