仓库模块完了

backend/init_db.py

@@ -1,6 +1,6 @@
 #!/usr/bin/env python3
 """
-初始化数据库，创建默认管理员账号
+初始化数据库，创建默认管理员账号和角色
 """
 
 from sqlalchemy.orm import Session
@@ -16,11 +16,25 @@ def init_db():
     Base.metadata.create_all(bind=engine)
     print("数据库表创建完成")
     
-    # 创建默认管理员账号
-    print("\n创建默认管理员账号...")
+    # 初始化默认角色
+    print("\n初始化默认角色...")
     db = SessionLocal()
     
     try:
+        # 初始化默认角色
+        UserService.init_default_roles(db)
+        print("✅ 默认角色初始化完成")
+        
+        # 获取默认角色
+        admin_role = UserService.get_role_by_name(db, "admin")
+        user_role = UserService.get_role_by_name(db, "user")
+        
+        if not admin_role or not user_role:
+            print("❌ 默认角色创建失败")
+            return
+        
+        # 创建默认管理员账号
+        print("\n创建默认管理员账号...")
         # 检查是否已存在管理员账号
         admin_user = db.query(User).filter(User.username == "admin").first()
         
@@ -31,7 +45,7 @@ def init_db():
                 username="admin",
                 email="admin@example.com",
                 password_hash=UserService.get_password_hash("admin123"),
-                role="admin",
+                role_id=admin_role.id,
                 status="active"
             )
             db.add(admin_user)