Cheng Zhou
64 lines
2.4 KiB
Python
64 lines
2.4 KiB
Python
"""项目角色生效管理API"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import uuid
|
|
from typing import Annotated
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException, status
|
|
from sqlalchemy import select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.core.deps import get_db_session, require_study_roles
|
|
from app.models.study import Study
|
|
|
|
router = APIRouter(prefix="/active-roles", tags=["active-roles"])
|
|
|
|
|
|
def _normalize_active_roles(value: object) -> list[str]:
|
|
if not isinstance(value, list):
|
|
raise HTTPException(status_code=status.HTTP_422_UNPROCESSABLE_ENTITY, detail="active_roles 必须是数组")
|
|
roles: list[str] = []
|
|
for item in value:
|
|
role = str(item or "").strip()
|
|
if not role:
|
|
continue
|
|
if role == "ADMIN":
|
|
raise HTTPException(status_code=status.HTTP_422_UNPROCESSABLE_ENTITY, detail="ADMIN 不能作为项目角色")
|
|
if role == "QA":
|
|
raise HTTPException(status_code=status.HTTP_422_UNPROCESSABLE_ENTITY, detail="QA 不能作为项目角色")
|
|
if len(role) > 20:
|
|
raise HTTPException(status_code=status.HTTP_422_UNPROCESSABLE_ENTITY, detail="项目角色长度不能超过20个字符")
|
|
if role not in roles:
|
|
roles.append(role)
|
|
return roles
|
|
|
|
|
|
@router.get("", summary="获取项目已生效角色列表")
|
|
async def get_active_roles(
|
|
study_id: uuid.UUID,
|
|
_=Depends(require_study_roles(["PM"])),
|
|
db: Annotated[AsyncSession, Depends(get_db_session)] = None,
|
|
) -> dict:
|
|
result = await db.execute(select(Study).where(Study.id == study_id))
|
|
study = result.scalar_one_or_none()
|
|
if not study:
|
|
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="项目不存在")
|
|
return {"active_roles": study.active_roles or []}
|
|
|
|
|
|
@router.put("", summary="更新项目已生效角色列表")
|
|
async def update_active_roles(
|
|
study_id: uuid.UUID,
|
|
payload: dict,
|
|
_=Depends(require_study_roles(["PM"])),
|
|
db: Annotated[AsyncSession, Depends(get_db_session)] = None,
|
|
) -> dict:
|
|
result = await db.execute(select(Study).where(Study.id == study_id))
|
|
study = result.scalar_one_or_none()
|
|
if not study:
|
|
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="项目不存在")
|
|
study.active_roles = _normalize_active_roles(payload.get("active_roles", []))
|
|
await db.commit()
|
|
return {"active_roles": study.active_roles}
|