75 lines
2.0 KiB
Python
75 lines
2.0 KiB
Python
|
"""删库重建脚本"""
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
import pymysql
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
SQL_DIR = Path(__file__).parent
|
|||
|
|
SERVER_DIR = SQL_DIR.parent
|
|||
|
|
|
|||
|
|
# 加载 .env 文件
|
|||
|
|
env_file = SERVER_DIR / '.env'
|
|||
|
|
if env_file.exists():
|
|||
|
|
with open(env_file, 'r', encoding='utf-8') as f:
|
|||
|
|
for line in f:
|
|||
|
|
line = line.strip()
|
|||
|
|
if line and not line.startswith('#') and '=' in line:
|
|||
|
|
key, value = line.split('=', 1)
|
|||
|
|
os.environ.setdefault(key.strip(), value.strip())
|
|||
|
|
|
|||
|
|
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):
|
|||
|
|
with open(SQL_DIR / filename, 'r', encoding='utf-8') as f:
|
|||
|
|
return f.read()
|
|||
|
|
|
|||
|
|
def execute_sql(cursor, sql, desc):
|
|||
|
|
print(f'{desc}...')
|
|||
|
|
for stmt in [s.strip() for s in sql.split(';') if s.strip()]:
|
|||
|
|
try:
|
|||
|
|
cursor.execute(stmt)
|
|||
|
|
except pymysql.Error as e:
|
|||
|
|
if e.args[0] not in [1007, 1050]:
|
|||
|
|
print(f' 警告: {e.args[1]}')
|
|||
|
|
print(f' {desc}完成！')
|
|||
|
|
|
|||
|
|
def rebuild():
|
|||
|
|
print('=' * 50)
|
|||
|
|
print('星域故事汇 - 删库重建')
|
|||
|
|
print('=' * 50)
|
|||
|
|
|
|||
|
|
conn = pymysql.connect(**DB_CONFIG)
|
|||
|
|
cur = conn.cursor()
|
|||
|
|
|
|||
|
|
# 删库
|
|||
|
|
print('删除旧数据库...')
|
|||
|
|
cur.execute('DROP DATABASE IF EXISTS stardom_story')
|
|||
|
|
conn.commit()
|
|||
|
|
print(' 删除完成！')
|
|||
|
|
|
|||
|
|
# 重建
|
|||
|
|
schema_sql = read_sql_file('schema.sql')
|
|||
|
|
execute_sql(cur, schema_sql, '创建数据库表结构')
|
|||
|
|
conn.commit()
|
|||
|
|
|
|||
|
|
seed1 = read_sql_file('seed_stories_part1.sql')
|
|||
|
|
execute_sql(cur, seed1, '导入种子数据（第1部分）')
|
|||
|
|
conn.commit()
|
|||
|
|
|
|||
|
|
seed2 = read_sql_file('seed_stories_part2.sql')
|
|||
|
|
execute_sql(cur, seed2, '导入种子数据（第2部分）')
|
|||
|
|
conn.commit()
|
|||
|
|
|
|||
|
|
print('\n数据库重建完成！')
|
|||
|
|
cur.close()
|
|||
|
|
conn.close()
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
rebuild()
|