80 lines
2.6 KiB
Python
80 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.managed_repo import ManagedRepo
|
||
|
|
from api.models.repo_goal import RepoGoal, RepoGoalStatus
|
||
|
|
from api.schemas.repo_goal import RepoGoalCreate, RepoGoalRead, RepoGoalUpdate
|
||
|
|
|
||
|
|
router = APIRouter(prefix="/repo-goals", tags=["repo-goals"])
|
||
|
|
|
||
|
|
|
||
|
|
async def _resolve_repo(repo_slug: str, session: AsyncSession) -> ManagedRepo:
|
||
|
|
result = await session.execute(select(ManagedRepo).where(ManagedRepo.slug == repo_slug))
|
||
|
|
repo = result.scalar_one_or_none()
|
||
|
|
if repo is None:
|
||
|
|
raise HTTPException(status_code=404, detail=f"Repo '{repo_slug}' not found")
|
||
|
|
return repo
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("/", response_model=list[RepoGoalRead])
|
||
|
|
async def list_repo_goals(
|
||
|
|
repo_slug: str | None = None,
|
||
|
|
domain_goal_id: uuid.UUID | None = None,
|
||
|
|
status: str | None = None,
|
||
|
|
session: AsyncSession = Depends(get_session),
|
||
|
|
) -> list[RepoGoal]:
|
||
|
|
q = select(RepoGoal)
|
||
|
|
if repo_slug:
|
||
|
|
repo = await _resolve_repo(repo_slug, session)
|
||
|
|
q = q.where(RepoGoal.repo_id == repo.id)
|
||
|
|
if domain_goal_id:
|
||
|
|
q = q.where(RepoGoal.domain_goal_id == domain_goal_id)
|
||
|
|
if status:
|
||
|
|
q = q.where(RepoGoal.status == status)
|
||
|
|
q = q.order_by(RepoGoal.priority.asc(), RepoGoal.created_at.asc())
|
||
|
|
result = await session.execute(q)
|
||
|
|
return list(result.scalars().all())
|
||
|
|
|
||
|
|
|
||
|
|
@router.post("/", response_model=RepoGoalRead, status_code=status.HTTP_201_CREATED)
|
||
|
|
async def create_repo_goal(
|
||
|
|
body: RepoGoalCreate,
|
||
|
|
session: AsyncSession = Depends(get_session),
|
||
|
|
) -> RepoGoal:
|
||
|
|
goal = RepoGoal(**body.model_dump())
|
||
|
|
session.add(goal)
|
||
|
|
await session.commit()
|
||
|
|
await session.refresh(goal)
|
||
|
|
return goal
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("/{goal_id}", response_model=RepoGoalRead)
|
||
|
|
async def get_repo_goal(
|
||
|
|
goal_id: uuid.UUID,
|
||
|
|
session: AsyncSession = Depends(get_session),
|
||
|
|
) -> RepoGoal:
|
||
|
|
goal = await session.get(RepoGoal, goal_id)
|
||
|
|
if goal is None:
|
||
|
|
raise HTTPException(status_code=404, detail="Repo goal not found")
|
||
|
|
return goal
|
||
|
|
|
||
|
|
|
||
|
|
@router.patch("/{goal_id}", response_model=RepoGoalRead)
|
||
|
|
async def update_repo_goal(
|
||
|
|
goal_id: uuid.UUID,
|
||
|
|
body: RepoGoalUpdate,
|
||
|
|
session: AsyncSession = Depends(get_session),
|
||
|
|
) -> RepoGoal:
|
||
|
|
goal = await session.get(RepoGoal, goal_id)
|
||
|
|
if goal is None:
|
||
|
|
raise HTTPException(status_code=404, detail="Repo goal not found")
|
||
|
|
for field, value in body.model_dump(exclude_unset=True).items():
|
||
|
|
setattr(goal, field, value)
|
||
|
|
await session.commit()
|
||
|
|
await session.refresh(goal)
|
||
|
|
return goal
|