40 lines
865 B
Python
40 lines
865 B
Python
#!/usr/bin/env python3
|
|
"""
|
|
数据库初始化脚本
|
|
"""
|
|
import sys
|
|
import os
|
|
|
|
# 添加项目路径
|
|
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
|
|
|
from app.database import init_db, SessionLocal
|
|
from app.init_data import init_default_data
|
|
|
|
|
|
def main():
|
|
print("🚀 开始初始化数据库...")
|
|
print("=" * 50)
|
|
|
|
# 1. 创建表结构
|
|
print("\n📦 创建数据库表...")
|
|
init_db()
|
|
print("✅ 数据库表创建完成")
|
|
|
|
# 2. 初始化默认数据
|
|
print("\n📝 初始化默认数据...")
|
|
db = SessionLocal()
|
|
try:
|
|
init_default_data(db)
|
|
finally:
|
|
db.close()
|
|
|
|
print("\n" + "=" * 50)
|
|
print("🎉 数据库初始化完成！")
|
|
print("\n你可以使用以下命令启动后端服务：")
|
|
print(" uvicorn app.main:app --reload")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|