81 lines
2.6 KiB
Python
81 lines
2.6 KiB
Python
|
import uuid
|
||
|
|
|
||
|
|
from fastapi import APIRouter, Depends, HTTPException, status
|
||
|
|
from sqlalchemy import select
|
||
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
|
|
||
|
|
from api.database import get_session
|
||
|
|
from api.models.workstream import Workstream, WorkstreamStatus
|
||
|
|
from api.schemas.workstream import WorkstreamCreate, WorkstreamRead, WorkstreamUpdate
|
||
|
|
|
||
|
|
router = APIRouter(prefix="/workstreams", tags=["workstreams"])
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("/", response_model=list[WorkstreamRead])
|
||
|
|
async def list_workstreams(
|
||
|
|
topic_id: uuid.UUID | None = None,
|
||
|
|
status: WorkstreamStatus | None = None,
|
||
|
|
session: AsyncSession = Depends(get_session),
|
||
|
|
) -> list[Workstream]:
|
||
|
|
q = select(Workstream)
|
||
|
|
if topic_id:
|
||
|
|
q = q.where(Workstream.topic_id == topic_id)
|
||
|
|
if status:
|
||
|
|
q = q.where(Workstream.status == status)
|
||
|
|
q = q.order_by(Workstream.created_at)
|
||
|
|
result = await session.execute(q)
|
||
|
|
return list(result.scalars().all())
|
||
|
|
|
||
|
|
|
||
|
|
@router.post("/", response_model=WorkstreamRead, status_code=status.HTTP_201_CREATED)
|
||
|
|
async def create_workstream(
|
||
|
|
body: WorkstreamCreate,
|
||
|
|
session: AsyncSession = Depends(get_session),
|
||
|
|
) -> Workstream:
|
||
|
|
ws = Workstream(**body.model_dump())
|
||
|
|
session.add(ws)
|
||
|
|
await session.commit()
|
||
|
|
await session.refresh(ws)
|
||
|
|
return ws
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("/{workstream_id}", response_model=WorkstreamRead)
|
||
|
|
async def get_workstream(
|
||
|
|
workstream_id: uuid.UUID,
|
||
|
|
session: AsyncSession = Depends(get_session),
|
||
|
|
) -> Workstream:
|
||
|
|
ws = await session.get(Workstream, workstream_id)
|
||
|
|
if ws is None:
|
||
|
|
raise HTTPException(status_code=404, detail="Workstream not found")
|
||
|
|
return ws
|
||
|
|
|
||
|
|
|
||
|
|
@router.patch("/{workstream_id}", response_model=WorkstreamRead)
|
||
|
|
async def update_workstream(
|
||
|
|
workstream_id: uuid.UUID,
|
||
|
|
body: WorkstreamUpdate,
|
||
|
|
session: AsyncSession = Depends(get_session),
|
||
|
|
) -> Workstream:
|
||
|
|
ws = await session.get(Workstream, workstream_id)
|
||
|
|
if ws is None:
|
||
|
|
raise HTTPException(status_code=404, detail="Workstream not found")
|
||
|
|
for field, value in body.model_dump(exclude_unset=True).items():
|
||
|
|
setattr(ws, field, value)
|
||
|
|
await session.commit()
|
||
|
|
await session.refresh(ws)
|
||
|
|
return ws
|
||
|
|
|
||
|
|
|
||
|
|
@router.delete("/{workstream_id}", response_model=WorkstreamRead)
|
||
|
|
async def archive_workstream(
|
||
|
|
workstream_id: uuid.UUID,
|
||
|
|
session: AsyncSession = Depends(get_session),
|
||
|
|
) -> Workstream:
|
||
|
|
ws = await session.get(Workstream, workstream_id)
|
||
|
|
if ws is None:
|
||
|
|
raise HTTPException(status_code=404, detail="Workstream not found")
|
||
|
|
ws.status = WorkstreamStatus.archived
|
||
|
|
await session.commit()
|
||
|
|
await session.refresh(ws)
|
||
|
|
return ws
|