Step 3：Study / Site / StudyMember + 项目权限

2025-12-16 16:42:42 +08:00
parent 65df698570
commit f3692e1145
66 changed files with 639 additions and 6 deletions
@@ -0,0 +1,90 @@
+import uuid
+
+from fastapi import APIRouter, Depends, HTTPException, status
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.core.deps import get_db_session, require_study_member, require_study_roles
+from app.crud import member as member_crud
+from app.crud import study as study_crud
+from app.schemas.member import StudyMemberCreate, StudyMemberRead, StudyMemberUpdate
+
+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="Study not found")
+    return study
+
+
+@router.post(
+    "/",
+    response_model=StudyMemberRead,
+    status_code=status.HTTP_201_CREATED,
+    dependencies=[Depends(require_study_roles(["PM"]))],
+)
+async def add_member(
+    study_id: uuid.UUID,
+    member_in: StudyMemberCreate,
+    db: AsyncSession = Depends(get_db_session),
+) -> StudyMemberRead:
+    await _ensure_study_exists(db, study_id)
+    existing = await member_crud.get_member(db, study_id, member_in.user_id)
+    if existing:
+        raise HTTPException(status_code=status.HTTP_409_CONFLICT, detail="Member already exists")
+    member = await member_crud.add_member(db, study_id, member_in)
+    return member
+
+
+@router.get(
+    "/",
+    response_model=list[StudyMemberRead],
+    dependencies=[Depends(require_study_member())],
+)
+async def list_members(
+    study_id: uuid.UUID,
+    skip: int = 0,
+    limit: int = 100,
+    db: AsyncSession = Depends(get_db_session),
+) -> list[StudyMemberRead]:
+    await _ensure_study_exists(db, study_id)
+    members = await member_crud.list_members(db, study_id, skip=skip, limit=limit)
+    return list(members)
+
+
+@router.patch(
+    "/{member_id}",
+    response_model=StudyMemberRead,
+    dependencies=[Depends(require_study_roles(["PM"]))],
+)
+async def update_member(
+    study_id: uuid.UUID,
+    member_id: uuid.UUID,
+    member_in: StudyMemberUpdate,
+    db: AsyncSession = Depends(get_db_session),
+) -> StudyMemberRead:
+    await _ensure_study_exists(db, study_id)
+    member = await member_crud.get_member_by_id(db, member_id)
+    if not member or member.study_id != study_id:
+        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Member not found")
+    updated = await member_crud.update_member(db, member, member_in)
+    return updated
+
+
+@router.delete(
+    "/{member_id}",
+    response_model=StudyMemberRead,
+    dependencies=[Depends(require_study_roles(["PM"]))],
+)
+async def remove_member(
+    study_id: uuid.UUID,
+    member_id: uuid.UUID,
+    db: AsyncSession = Depends(get_db_session),
+) -> StudyMemberRead:
+    await _ensure_study_exists(db, study_id)
+    member = await member_crud.get_member_by_id(db, member_id)
+    if not member or member.study_id != study_id:
+        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Member not found")
+    removed = await member_crud.remove_member(db, member)
+    return removed
@@ -1,7 +1,10 @@
 from fastapi import APIRouter

-from app.api.v1 import auth, users
+from app.api.v1 import auth, users, studies, sites, members

 api_router = APIRouter()
 api_router.include_router(auth.router, prefix="/auth", tags=["auth"])
 api_router.include_router(users.router, prefix="/users", tags=["users"])
+api_router.include_router(studies.router, prefix="/studies", tags=["studies"])
+api_router.include_router(sites.router, prefix="/studies/{study_id}/sites", tags=["sites"])
+api_router.include_router(members.router, prefix="/studies/{study_id}/members", tags=["study-members"])
@@ -0,0 +1,69 @@
+import uuid
+
+from fastapi import APIRouter, Depends, HTTPException, status
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.core.deps import get_db_session, require_study_member, require_study_roles
+from app.crud import site as site_crud
+from app.crud import study as study_crud
+from app.schemas.site import SiteCreate, SiteRead, SiteUpdate
+
+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="Study not found")
+    return study
+
+
+@router.post(
+    "/",
+    response_model=SiteRead,
+    status_code=status.HTTP_201_CREATED,
+    dependencies=[Depends(require_study_roles(["PM"]))],
+)
+async def create_site(
+    study_id: uuid.UUID,
+    site_in: SiteCreate,
+    db: AsyncSession = Depends(get_db_session),
+) -> SiteRead:
+    await _ensure_study_exists(db, study_id)
+    site = await site_crud.create_site(db, study_id, site_in)
+    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,
+    db: AsyncSession = Depends(get_db_session),
+) -> list[SiteRead]:
+    await _ensure_study_exists(db, study_id)
+    sites = await site_crud.list_by_study(db, study_id, skip=skip, limit=limit)
+    return list(sites)
+
+
+@router.patch(
+    "/{site_id}",
+    response_model=SiteRead,
+    dependencies=[Depends(require_study_roles(["PM"]))],
+)
+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="Site not found")
+    updated = await site_crud.update_site(db, site, site_in)
+    return updated
@@ -0,0 +1,65 @@
+import uuid
+
+from fastapi import APIRouter, Depends, HTTPException, status
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.core.deps import get_current_user, get_db_session, require_roles, require_study_member, require_study_roles
+from app.crud import study as study_crud
+from app.schemas.study import StudyCreate, StudyRead, StudyUpdate
+
+router = APIRouter()
+
+
+@router.post("/", response_model=StudyRead, status_code=status.HTTP_201_CREATED, dependencies=[Depends(require_roles(["ADMIN"]))])
+async def create_study(
+    study_in: StudyCreate,
+    db: AsyncSession = Depends(get_db_session),
+    current_user=Depends(get_current_user),
+) -> StudyRead:
+    existing = await study_crud.get_by_code(db, study_in.code)
+    if existing:
+        raise HTTPException(status_code=status.HTTP_409_CONFLICT, detail="Study code already exists")
+    study = await study_crud.create(db, study_in, created_by=current_user.id)
+    return study
+
+
+@router.get("/", response_model=list[StudyRead])
+async def list_studies(
+    skip: int = 0,
+    limit: int = 100,
+    db: AsyncSession = Depends(get_db_session),
+    current_user=Depends(get_current_user),
+) -> list[StudyRead]:
+    if current_user.role == "ADMIN":
+        studies = await study_crud.list_studies(db, skip=skip, limit=limit)
+    else:
+        studies = await study_crud.list_studies_for_user(db, current_user.id, skip=skip, limit=limit)
+    return list(studies)
+
+
+@router.get("/{study_id}", response_model=StudyRead, dependencies=[Depends(require_study_member())])
+async def get_study(
+    study_id: uuid.UUID,
+    db: AsyncSession = Depends(get_db_session),
+) -> StudyRead:
+    study = await study_crud.get(db, study_id)
+    if not study:
+        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Study not found")
+    return study
+
+
+@router.patch(
+    "/{study_id}",
+    response_model=StudyRead,
+    dependencies=[Depends(require_study_roles(["PM"]))],
+)
+async def update_study(
+    study_id: uuid.UUID,
+    study_in: StudyUpdate,
+    db: AsyncSession = Depends(get_db_session),
+) -> StudyRead:
+    study = await study_crud.get(db, study_id)
+    if not study:
+        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Study not found")
+    updated = await study_crud.update(db, study, study_in)
+    return updated