generated from coulomb/repo-seed
98 lines
3.0 KiB
Python
98 lines
3.0 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
|
|
from api.schemas.workstream import (
|
|
WorkstreamCreate,
|
|
WorkstreamRead,
|
|
WorkstreamStatus,
|
|
WorkstreamUpdate,
|
|
)
|
|
|
|
router = APIRouter(prefix="/workstreams", tags=["workstreams"])
|
|
|
|
|
|
@router.get("/", response_model=list[WorkstreamRead])
|
|
async def list_workstreams(
|
|
topic_id: uuid.UUID | None = None,
|
|
repo_id: uuid.UUID | None = None,
|
|
repo_goal_id: uuid.UUID | None = None,
|
|
status: WorkstreamStatus | None = None,
|
|
owner: str | None = None,
|
|
slug: str | None = None,
|
|
session: AsyncSession = Depends(get_session),
|
|
) -> list[Workstream]:
|
|
q = select(Workstream)
|
|
if topic_id:
|
|
q = q.where(Workstream.topic_id == topic_id)
|
|
if repo_id:
|
|
q = q.where(Workstream.repo_id == repo_id)
|
|
if repo_goal_id:
|
|
q = q.where(Workstream.repo_goal_id == repo_goal_id)
|
|
if status:
|
|
q = q.where(Workstream.status == status)
|
|
if owner:
|
|
q = q.where(Workstream.owner == owner)
|
|
if slug:
|
|
q = q.where(Workstream.slug == slug)
|
|
q = q.order_by(Workstream.updated_at.desc())
|
|
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 = "archived"
|
|
await session.commit()
|
|
await session.refresh(ws)
|
|
return ws
|