127 lines
3.8 KiB
Python
127 lines
3.8 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
简化的数据同步执行脚本
|
|
直接使用项目配置执行数据同步
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
import subprocess
|
|
from urllib.parse import urlparse
|
|
from app.config import settings
|
|
|
|
def parse_database_url():
|
|
"""解析数据库 URL"""
|
|
url = urlparse(settings.DATABASE_URL)
|
|
return {
|
|
'host': url.hostname or 'localhost',
|
|
'port': url.port or 3306,
|
|
'user': url.username or 'root',
|
|
'password': url.password or '',
|
|
'database': url.path.lstrip('/') if url.path else 'osu_api'
|
|
}
|
|
|
|
def run_sql_script(script_path: str):
|
|
"""使用 mysql 命令行执行 SQL 脚本"""
|
|
if not os.path.exists(script_path):
|
|
print(f"错误: SQL 脚本不存在 - {script_path}")
|
|
return False
|
|
|
|
# 解析数据库配置
|
|
db_config = parse_database_url()
|
|
|
|
# 构建 mysql 命令
|
|
cmd = [
|
|
'mysql',
|
|
f'--host={db_config["host"]}',
|
|
f'--port={db_config["port"]}',
|
|
f'--user={db_config["user"]}',
|
|
db_config['database']
|
|
]
|
|
|
|
# 添加密码（如果有的话）
|
|
if db_config['password']:
|
|
cmd.insert(-1, f'--password={db_config["password"]}')
|
|
|
|
try:
|
|
print(f"执行 SQL 脚本: {script_path}")
|
|
with open(script_path, 'r', encoding='utf-8') as f:
|
|
result = subprocess.run(
|
|
cmd,
|
|
stdin=f,
|
|
capture_output=True,
|
|
text=True,
|
|
check=True
|
|
)
|
|
|
|
if result.stdout:
|
|
print("执行结果:")
|
|
print(result.stdout)
|
|
|
|
print(f"✓ 成功执行: {script_path}")
|
|
return True
|
|
|
|
except subprocess.CalledProcessError as e:
|
|
print(f"✗ 执行失败: {script_path}")
|
|
print(f"错误信息: {e.stderr}")
|
|
return False
|
|
except FileNotFoundError:
|
|
print("错误: 未找到 mysql 命令行工具")
|
|
print("请确保 MySQL 客户端已安装并添加到 PATH 环境变量中")
|
|
return False
|
|
|
|
def main():
|
|
"""主函数"""
|
|
print("Lazer API 快速数据同步")
|
|
print("=" * 40)
|
|
|
|
db_config = parse_database_url()
|
|
print(f"数据库: {db_config['host']}:{db_config['port']}/{db_config['database']}")
|
|
print()
|
|
|
|
# 确认是否继续
|
|
print("这将执行以下操作:")
|
|
print("1. 创建 lazer 专用表结构")
|
|
print("2. 同步现有用户数据到新表")
|
|
print("3. 不会修改现有的原始表数据")
|
|
print()
|
|
|
|
confirm = input("是否继续? (y/N): ").strip().lower()
|
|
if confirm != 'y':
|
|
print("操作已取消")
|
|
return
|
|
|
|
# 获取脚本路径
|
|
script_dir = os.path.dirname(__file__)
|
|
migrations_dir = os.path.join(script_dir, 'migrations')
|
|
|
|
# 第一步: 创建表结构
|
|
print("\n步骤 1: 创建 lazer 专用表结构...")
|
|
add_fields_script = os.path.join(migrations_dir, 'add_missing_fields.sql')
|
|
if not run_sql_script(add_fields_script):
|
|
print("表结构创建失败，停止执行")
|
|
return
|
|
|
|
# 第二步: 同步数据
|
|
print("\n步骤 2: 同步历史数据...")
|
|
sync_script = os.path.join(migrations_dir, 'sync_legacy_data.sql')
|
|
if not run_sql_script(sync_script):
|
|
print("数据同步失败")
|
|
return
|
|
|
|
# 第三步: 添加缺失的字段
|
|
print("\n步骤 3: 添加缺失的字段...")
|
|
add_rank_fields_script = os.path.join(migrations_dir, 'add_lazer_rank_fields.sql')
|
|
if not run_sql_script(add_rank_fields_script):
|
|
print("添加字段失败")
|
|
return
|
|
|
|
print("\n🎉 数据同步完成!")
|
|
print("\n现在您可以:")
|
|
print("1. 启动 Lazer API 服务器")
|
|
print("2. 使用现有用户账号登录")
|
|
print("3. 查看同步后的用户数据")
|
|
|
|
if __name__ == "__main__":
|
|
main()
|