wehub-resource-sync
57 lines
1.4 KiB
Python
57 lines
1.4 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
数据库初始化脚本
|
|
"""
|
|
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
# 添加项目根目录到路径
|
|
current_dir = Path(__file__).parent
|
|
sys.path.insert(0, str(current_dir))
|
|
sys.path.insert(0, str(current_dir / "backend"))
|
|
|
|
# 设置工作目录
|
|
import os
|
|
os.chdir(current_dir)
|
|
|
|
def init_database():
|
|
"""初始化数据库"""
|
|
print("🚀 开始初始化数据库...")
|
|
|
|
try:
|
|
# 导入所有模型确保表被创建
|
|
from backend.models import Base, BilibiliAccount, UploadRecord
|
|
from backend.core.database import init_database, create_tables
|
|
|
|
print("✅ 所有模型导入成功")
|
|
|
|
# 初始化数据库
|
|
if init_database():
|
|
print("✅ 数据库初始化成功")
|
|
else:
|
|
print("❌ 数据库初始化失败")
|
|
return False
|
|
|
|
# 创建表
|
|
create_tables()
|
|
print("✅ 数据库表创建成功")
|
|
|
|
return True
|
|
|
|
except Exception as e:
|
|
print(f"❌ 数据库初始化失败: {e}")
|
|
return False
|
|
|
|
if __name__ == "__main__":
|
|
success = init_database()
|
|
if success:
|
|
print("\n🎉 数据库初始化完成！")
|
|
print("现在可以启动系统了：")
|
|
print("1. ./start_autoclip_with_upload.sh")
|
|
print("2. 或者手动启动各个服务")
|
|
else:
|
|
print("\n❌ 数据库初始化失败，请检查错误信息")
|
|
sys.exit(1)
|
|
|