Cheng Zhou
118 lines
3.8 KiB
Python
118 lines
3.8 KiB
Python
import uuid
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException, status
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.core.deps import get_cra_site_scope, get_current_user, get_db_session, require_study_member, require_study_roles, require_study_not_locked
|
|
from app.crud import site as site_crud
|
|
from app.crud import startup as startup_crud
|
|
from app.crud import study as study_crud
|
|
from app.schemas.site import SiteCreate, SiteRead, SiteUpdate
|
|
from app.schemas.startup import StartupEthicsCreate, StartupFeasibilityCreate
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
async def _ensure_study_exists(db: AsyncSession, study_id: uuid.UUID):
|
|
study = await study_crud.get(db, study_id)
|
|
if not study:
|
|
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="项目不存在")
|
|
return study
|
|
|
|
|
|
@router.post(
|
|
"/",
|
|
response_model=SiteRead,
|
|
status_code=status.HTTP_201_CREATED,
|
|
dependencies=[Depends(require_study_roles(["PM"])), Depends(require_study_not_locked())],
|
|
)
|
|
async def create_site(
|
|
study_id: uuid.UUID,
|
|
site_in: SiteCreate,
|
|
db: AsyncSession = Depends(get_db_session),
|
|
current_user=Depends(get_current_user),
|
|
) -> SiteRead:
|
|
await _ensure_study_exists(db, study_id)
|
|
site = await site_crud.create_site(db, study_id, site_in)
|
|
await startup_crud.create_feasibility(
|
|
db,
|
|
study_id,
|
|
StartupFeasibilityCreate(site_id=site.id),
|
|
created_by=getattr(current_user, "id", None),
|
|
)
|
|
await startup_crud.create_ethics(
|
|
db,
|
|
study_id,
|
|
StartupEthicsCreate(site_id=site.id),
|
|
created_by=getattr(current_user, "id", None),
|
|
)
|
|
return site
|
|
|
|
|
|
@router.get(
|
|
"/",
|
|
response_model=list[SiteRead],
|
|
dependencies=[Depends(require_study_member())],
|
|
)
|
|
async def list_sites(
|
|
study_id: uuid.UUID,
|
|
skip: int = 0,
|
|
limit: int = 100,
|
|
include_inactive: bool = False,
|
|
db: AsyncSession = Depends(get_db_session),
|
|
current_user=Depends(get_current_user),
|
|
) -> list[SiteRead]:
|
|
await _ensure_study_exists(db, study_id)
|
|
cra_scope = await get_cra_site_scope(db, study_id, current_user)
|
|
site_ids = cra_scope[0] if cra_scope else None
|
|
sites = await site_crud.list_by_study(
|
|
db,
|
|
study_id,
|
|
skip=skip,
|
|
limit=limit,
|
|
include_inactive=include_inactive,
|
|
site_ids=site_ids,
|
|
)
|
|
return list(sites)
|
|
|
|
|
|
@router.patch(
|
|
"/{site_id}",
|
|
response_model=SiteRead,
|
|
dependencies=[Depends(require_study_roles(["PM"])), Depends(require_study_not_locked())],
|
|
)
|
|
async def update_site(
|
|
study_id: uuid.UUID,
|
|
site_id: uuid.UUID,
|
|
site_in: SiteUpdate,
|
|
db: AsyncSession = Depends(get_db_session),
|
|
) -> SiteRead:
|
|
await _ensure_study_exists(db, study_id)
|
|
site = await site_crud.get_site(db, site_id)
|
|
if not site or site.study_id != study_id:
|
|
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="分中心不存在")
|
|
updated = await site_crud.update_site(db, site, site_in)
|
|
return updated
|
|
|
|
|
|
@router.delete(
|
|
"/{site_id}",
|
|
status_code=status.HTTP_204_NO_CONTENT,
|
|
dependencies=[Depends(require_study_not_locked())],
|
|
)
|
|
async def delete_site(
|
|
study_id: uuid.UUID,
|
|
site_id: uuid.UUID,
|
|
db: AsyncSession = Depends(get_db_session),
|
|
current_user=Depends(get_current_user),
|
|
) -> None:
|
|
await _ensure_study_exists(db, study_id)
|
|
role_value = current_user.role.value if hasattr(current_user.role, "value") else current_user.role
|
|
if role_value != "ADMIN":
|
|
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail="仅管理员可删除分中心")
|
|
site = await site_crud.get_site(db, site_id)
|
|
if not site or site.study_id != study_id:
|
|
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="分中心不存在")
|
|
await site_crud.delete_site_and_related(db, site)
|
|
return None
|