注册服务

2026-02-09 23:59:25 +08:00
parent f145df4fa6
commit 3c03777b97
11 changed files with 865 additions and 456 deletions
--- a/backend/app/routes/services.py
+++ b/backend/app/routes/services.py
@@ -6,6 +6,7 @@ from pydantic import BaseModel
 import uuid
 import os

+from app.config.settings import settings
 from app.models.models import AlgorithmService, ServiceGroup, AlgorithmRepository
 from app.models.database import SessionLocal
 from app.routes.user import get_current_active_user
@@ -42,7 +43,7 @@ class ServiceResponse(BaseModel):
    api_url: str
    status: str
    created_at: str
-    updated_at: str
+    updated_at: Optional[str]


 class ServiceListResponse(BaseModel):
@@ -127,7 +128,7 @@ class BatchOperationResponse(BaseModel):
 # 初始化服务组件
 project_analyzer = ProjectAnalyzer()
 service_generator = ServiceGenerator()
-service_orchestrator = ServiceOrchestrator()
+service_orchestrator = ServiceOrchestrator(deployment_mode=settings.DEPLOYMENT_MODE)


@router.post("/register", status_code=status.HTTP_201_CREATED)
@@ -137,7 +138,9 @@ async def register_service(
 ):
    """注册新服务"""
    # 检查用户权限
-    if current_user.role_name != "admin":
+    print(f"用户角色: {current_user.role_name}")
+    print(f"用户角色对象: {current_user.role}")
+    if not hasattr(current_user, 'role_name') or current_user.role_name != "admin":
        raise HTTPException(status_code=403, detail="Insufficient permissions")
    
    # 创建数据库会话
@@ -236,7 +239,7 @@ def main(data):
                "api_url": new_service.api_url,
                "status": new_service.status,
                "created_at": new_service.created_at.isoformat(),
-                "updated_at": new_service.updated_at.isoformat()
+                "updated_at": new_service.updated_at.isoformat() if new_service.updated_at else None
            }
        }
    finally:
@@ -272,7 +275,7 @@ async def list_services(
                api_url=service.api_url,
                status=service.status,
                created_at=service.created_at.isoformat(),
-                updated_at=service.updated_at.isoformat()
+                updated_at=service.updated_at.isoformat() if service.updated_at else None
            ))
        
        return ServiceListResponse(
@@ -316,7 +319,7 @@ async def get_service(
                api_url=service.api_url,
                status=service.status,
                created_at=service.created_at.isoformat(),
-                updated_at=service.updated_at.isoformat()
+                updated_at=service.updated_at.isoformat() if service.updated_at else None
            )
        )
    finally:
--- a/backend/app/routes/user.py
+++ b/backend/app/routes/user.py
@@ -39,8 +39,14 @@ async def get_current_active_user(db: Session = Depends(get_db), token: str = De
            )
        
        # 使用UserService获取用户信息，避免直接使用User模型
+        print(f"尝试通过用户名获取用户: {username}")
        user = UserService.get_user_by_username(db, username)
+        print(f"获取用户结果: {user.id if user else 'None'}")
        if not user:
+            # 尝试直接查询数据库
+            from app.models.models import User as UserModel
+            direct_user = db.query(UserModel).filter(UserModel.username == username).first()
+            print(f"直接查询数据库结果: {direct_user.id if direct_user else 'None'}")
            raise HTTPException(
                status_code=status.HTTP_401_UNAUTHORIZED,
                detail="Could not validate credentials",
@@ -51,19 +57,38 @@ async def get_current_active_user(db: Session = Depends(get_db), token: str = De
        if user.status != "active":
            raise HTTPException(status_code=400, detail="Inactive user")
        
-        # 使用UserService获取角色信息
-        role = UserService.get_role_by_id(db, user.role_id)
-        
        # 构建角色响应
        role_response = None
-        if role:
-            role_response = RoleResponse(
-                id=role.id,
-                name=role.name,
-                description=role.description,
-                created_at=role.created_at,
-                updated_at=role.updated_at
-            )
+        role_name = None
+        
+        # 尝试获取角色信息
+        try:
+            # 先尝试使用预加载的角色
+            if hasattr(user, 'role') and user.role:
+                role = user.role
+                role_response = RoleResponse(
+                    id=role.id,
+                    name=role.name,
+                    description=role.description,
+                    created_at=role.created_at,
+                    updated_at=role.updated_at
+                )
+                role_name = role.name
+            else:
+                # 如果没有预加载角色，尝试通过role_id获取
+                role = UserService.get_role_by_id(db, user.role_id)
+                if role:
+                    role_response = RoleResponse(
+                        id=role.id,
+                        name=role.name,
+                        description=role.description,
+                        created_at=role.created_at,
+                        updated_at=role.updated_at
+                    )
+                    role_name = role.name
+        except Exception as e:
+            # 角色获取失败不影响用户认证
+            print(f"获取角色信息失败: {e}")
        
        # 构建用户响应
        user_response = UserResponse(
@@ -75,7 +100,7 @@ async def get_current_active_user(db: Session = Depends(get_db), token: str = De
            created_at=user.created_at,
            updated_at=user.updated_at,
            role=role_response,
-            role_name=role.name if role else None
+            role_name=role_name
        )
        
        return user_response
@@ -85,6 +110,13 @@ async def get_current_active_user(db: Session = Depends(get_db), token: str = De
            detail="Could not validate credentials",
            headers={"WWW-Authenticate": "Bearer"},
        )
+    except Exception as e:
+        print(f"获取当前用户失败: {e}")
+        raise HTTPException(
+            status_code=status.HTTP_401_UNAUTHORIZED,
+            detail="Could not validate credentials",
+            headers={"WWW-Authenticate": "Bearer"},
+        )


 from app.schemas.user import LoginRequest
@@ -151,6 +183,61 @@ async def get_users(
    return {"users": users, "total": len(users)}


+# 角色管理API
+@router.post("/roles", response_model=RoleResponse)
+async def create_role(
+    role: RoleCreate,
+    current_user: UserResponse = Depends(get_current_active_user),
+    db: Session = Depends(get_db)
+):
+    """创建角色"""
+    # 只有管理员可以创建角色
+    if current_user.role_name != "admin":
+        raise HTTPException(status_code=403, detail="Not enough permissions")
+    
+    # 检查角色名称是否已存在
+    if UserService.get_role_by_name(db, role.name):
+        raise HTTPException(status_code=400, detail="Role name already exists")
+    
+    # 创建角色
+    db_role = UserService.create_role(db, role)
+    
+    return db_role
+
+
+@router.get("/roles", response_model=List[RoleResponse])
+async def get_roles(
+    current_user: UserResponse = Depends(get_current_active_user),
+    db: Session = Depends(get_db)
+):
+    """获取角色列表"""
+    # 只有管理员可以查看所有角色
+    if current_user.role_name != "admin":
+        raise HTTPException(status_code=403, detail="Not enough permissions")
+    
+    roles = UserService.get_roles(db)
+    
+    return roles
+
+
+@router.get("/roles/{role_id}", response_model=RoleResponse)
+async def get_role(
+    role_id: str,
+    current_user: UserResponse = Depends(get_current_active_user),
+    db: Session = Depends(get_db)
+):
+    """获取角色详情"""
+    # 只有管理员可以查看角色详情
+    if current_user.role_name != "admin":
+        raise HTTPException(status_code=403, detail="Not enough permissions")
+    
+    role = UserService.get_role_by_id(db, role_id)
+    if not role:
+        raise HTTPException(status_code=404, detail="Role not found")
+    
+    return role
+
+
@router.get("/{user_id}", response_model=UserResponse)
 async def get_user(
    user_id: str,
@@ -218,58 +305,3 @@ async def delete_user(
    db.commit()
    
    return {"message": "User deleted successfully"}
-
-
-# 角色管理API
-@router.post("/roles", response_model=RoleResponse)
-async def create_role(
-    role: RoleCreate,
-    current_user: UserResponse = Depends(get_current_active_user),
-    db: Session = Depends(get_db)
-):
-    """创建角色"""
-    # 只有管理员可以创建角色
-    if current_user.role_name != "admin":
-        raise HTTPException(status_code=403, detail="Not enough permissions")
-    
-    # 检查角色名称是否已存在
-    if UserService.get_role_by_name(db, role.name):
-        raise HTTPException(status_code=400, detail="Role name already exists")
-    
-    # 创建角色
-    db_role = UserService.create_role(db, role)
-    
-    return db_role
-
-
-@router.get("/roles", response_model=List[RoleResponse])
-async def get_roles(
-    current_user: UserResponse = Depends(get_current_active_user),
-    db: Session = Depends(get_db)
-):
-    """获取角色列表"""
-    # 只有管理员可以查看所有角色
-    if current_user.role_name != "admin":
-        raise HTTPException(status_code=403, detail="Not enough permissions")
-    
-    roles = UserService.get_roles(db)
-    
-    return roles
-
-
-@router.get("/roles/{role_id}", response_model=RoleResponse)
-async def get_role(
-    role_id: str,
-    current_user: UserResponse = Depends(get_current_active_user),
-    db: Session = Depends(get_db)
-):
-    """获取角色详情"""
-    # 只有管理员可以查看角色详情
-    if current_user.role_name != "admin":
-        raise HTTPException(status_code=403, detail="Not enough permissions")
-    
-    role = UserService.get_role_by_id(db, role_id)
-    if not role:
-        raise HTTPException(status_code=404, detail="Role not found")
-    
-    return role