first commit

backend/init_db.py

@@ -0,0 +1,51 @@
+#!/usr/bin/env python3
+"""
+初始化数据库，创建默认管理员账号
+"""
+
+from sqlalchemy.orm import Session
+from app.models.database import engine, Base, SessionLocal
+from app.models.models import User
+from app.services.user import UserService
+
+
+def init_db():
+    """初始化数据库"""
+    # 创建所有表
+    print("创建数据库表...")
+    Base.metadata.create_all(bind=engine)
+    print("数据库表创建完成")
+    
+    # 创建默认管理员账号
+    print("\n创建默认管理员账号...")
+    db = SessionLocal()
+    
+    try:
+        # 检查是否已存在管理员账号
+        admin_user = db.query(User).filter(User.username == "admin").first()
+        
+        if not admin_user:
+            # 创建默认管理员账号
+            admin_user = User(
+                id="user-admin",
+                username="admin",
+                email="admin@example.com",
+                password_hash=UserService.get_password_hash("admin123"),
+                role="admin",
+                status="active"
+            )
+            db.add(admin_user)
+            db.commit()
+            db.refresh(admin_user)
+            print("✅ 默认管理员账号创建成功")
+            print(f"用户名: admin")
+            print(f"密码: admin123")
+        else:
+            print("⚠️  管理员账号已存在")
+            
+    finally:
+        db.close()
+
+
+if __name__ == "__main__":
+    init_db()