Files
kmtnflower/backend/app/api/team.py
T

80 lines
2.6 KiB
Python

from typing import Optional
from fastapi import APIRouter, Depends, Query
from sqlalchemy import select
from sqlalchemy.ext.asyncio import AsyncSession
from app.auth import get_current_admin_user
from app.database import get_session
from app.models import TeamMember
from app.schemas import TeamMemberCreate, TeamMemberUpdate, TeamMemberResponse
from app.user_models import User
router = APIRouter()
@router.get("", response_model=list[TeamMemberResponse])
async def list_team_members(
active: Optional[bool] = Query(None, description="Filter by active status"),
session: AsyncSession = Depends(get_session),
):
query = select(TeamMember).order_by(TeamMember.display_order)
if active is not None:
query = query.where(TeamMember.active == active)
result = await session.execute(query)
return list(result.scalars().all())
@router.get("/{member_id}", response_model=TeamMemberResponse)
async def get_team_member(member_id: int, session: AsyncSession = Depends(get_session)):
result = await session.execute(select(TeamMember).where(TeamMember.id == member_id))
member = result.scalar_one_or_none()
if not member:
return {"error": "Team member not found"}
return member
@router.post("", response_model=TeamMemberResponse, status_code=201)
async def create_team_member(
payload: TeamMemberCreate,
session: AsyncSession = Depends(get_session),
_: User = Depends(get_current_admin_user),
):
member = TeamMember(**payload.model_dump(), active=True)
session.add(member)
await session.commit()
await session.refresh(member)
return member
@router.put("/{member_id}", response_model=TeamMemberResponse)
async def update_team_member(
member_id: int,
payload: TeamMemberUpdate,
session: AsyncSession = Depends(get_session),
_: User = Depends(get_current_admin_user),
):
result = await session.execute(select(TeamMember).where(TeamMember.id == member_id))
member = result.scalar_one_or_none()
if not member:
return {"error": "Team member not found"}
for key, value in payload.model_dump(exclude_unset=True).items():
setattr(member, key, value)
await session.commit()
await session.refresh(member)
return member
@router.delete("/{member_id}", status_code=204)
async def delete_team_member(
member_id: int,
session: AsyncSession = Depends(get_session),
_: User = Depends(get_current_admin_user),
):
result = await session.execute(select(TeamMember).where(TeamMember.id == member_id))
member = result.scalar_one_or_none()
if not member:
return {"error": "Team member not found"}
await session.delete(member)
await session.commit()