Step 3:Study / Site / StudyMember + 项目权限
This commit is contained in:
@@ -0,0 +1,64 @@
|
||||
import uuid
|
||||
from typing import Sequence
|
||||
|
||||
from sqlalchemy import select, update
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
from app.models.study_member import StudyMember
|
||||
from app.schemas.member import StudyMemberCreate, StudyMemberUpdate
|
||||
|
||||
|
||||
async def get_member(db: AsyncSession, study_id: uuid.UUID, user_id: uuid.UUID) -> StudyMember | None:
|
||||
result = await db.execute(
|
||||
select(StudyMember).where(StudyMember.study_id == study_id, StudyMember.user_id == user_id)
|
||||
)
|
||||
return result.scalar_one_or_none()
|
||||
|
||||
|
||||
async def get_member_by_id(db: AsyncSession, member_id: uuid.UUID) -> StudyMember | None:
|
||||
result = await db.execute(select(StudyMember).where(StudyMember.id == member_id))
|
||||
return result.scalar_one_or_none()
|
||||
|
||||
|
||||
async def add_member(db: AsyncSession, study_id: uuid.UUID, member_in: StudyMemberCreate) -> StudyMember:
|
||||
member = StudyMember(
|
||||
study_id=study_id,
|
||||
user_id=member_in.user_id,
|
||||
role_in_study=member_in.role_in_study,
|
||||
is_active=member_in.is_active,
|
||||
)
|
||||
db.add(member)
|
||||
await db.commit()
|
||||
await db.refresh(member)
|
||||
return member
|
||||
|
||||
|
||||
async def list_members(db: AsyncSession, study_id: uuid.UUID, skip: int = 0, limit: int = 100) -> Sequence[StudyMember]:
|
||||
result = await db.execute(
|
||||
select(StudyMember).where(StudyMember.study_id == study_id).offset(skip).limit(limit)
|
||||
)
|
||||
return result.scalars().all()
|
||||
|
||||
|
||||
async def update_member(db: AsyncSession, member: StudyMember, member_in: StudyMemberUpdate) -> StudyMember:
|
||||
update_data = {k: v for k, v in member_in.model_dump(exclude_unset=True).items() if v is not None}
|
||||
if update_data:
|
||||
await db.execute(
|
||||
update(StudyMember)
|
||||
.where(StudyMember.id == member.id)
|
||||
.values(**update_data)
|
||||
)
|
||||
await db.commit()
|
||||
await db.refresh(member)
|
||||
return member
|
||||
|
||||
|
||||
async def remove_member(db: AsyncSession, member: StudyMember) -> StudyMember:
|
||||
await db.execute(
|
||||
update(StudyMember)
|
||||
.where(StudyMember.id == member.id)
|
||||
.values(is_active=False)
|
||||
)
|
||||
await db.commit()
|
||||
await db.refresh(member)
|
||||
return member
|
||||
Reference in New Issue
Block a user