66 lines
2 KiB
Python
66 lines
2 KiB
Python
|
"""Retirement-goal CRUD nested under a scenario."""
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from fastapi import APIRouter, Depends, HTTPException
|
||
|
|
from sqlalchemy import delete, select
|
||
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
|
|
||
|
|
from fire_planner.api.dependencies import get_session
|
||
|
|
from fire_planner.api.schemas import GoalCreate, GoalOut
|
||
|
|
from fire_planner.db import RetirementGoal, Scenario
|
||
|
|
|
||
|
|
router = APIRouter(tags=["goals"])
|
||
|
|
|
||
|
|
|
||
|
|
@router.get(
|
||
|
|
"/scenarios/{scenario_id}/goals",
|
||
|
|
response_model=list[GoalOut],
|
||
|
|
)
|
||
|
|
async def list_goals(
|
||
|
|
scenario_id: int,
|
||
|
|
session: AsyncSession = Depends(get_session),
|
||
|
|
) -> list[RetirementGoal]:
|
||
|
|
scen = await session.get(Scenario, scenario_id)
|
||
|
|
if scen is None:
|
||
|
|
raise HTTPException(status_code=404, detail="Scenario not found")
|
||
|
|
rows = (await session.execute(
|
||
|
|
select(RetirementGoal).where(RetirementGoal.scenario_id == scenario_id).order_by(
|
||
|
|
RetirementGoal.id))).scalars().all()
|
||
|
|
return list(rows)
|
||
|
|
|
||
|
|
|
||
|
|
@router.post(
|
||
|
|
"/scenarios/{scenario_id}/goals",
|
||
|
|
response_model=GoalOut,
|
||
|
|
status_code=201,
|
||
|
|
)
|
||
|
|
async def create_goal(
|
||
|
|
scenario_id: int,
|
||
|
|
payload: GoalCreate,
|
||
|
|
session: AsyncSession = Depends(get_session),
|
||
|
|
) -> RetirementGoal:
|
||
|
|
scen = await session.get(Scenario, scenario_id)
|
||
|
|
if scen is None:
|
||
|
|
raise HTTPException(status_code=404, detail="Scenario not found")
|
||
|
|
goal = RetirementGoal(scenario_id=scenario_id, **payload.model_dump())
|
||
|
|
session.add(goal)
|
||
|
|
await session.commit()
|
||
|
|
await session.refresh(goal)
|
||
|
|
return goal
|
||
|
|
|
||
|
|
|
||
|
|
@router.delete(
|
||
|
|
"/goals/{goal_id}",
|
||
|
|
status_code=204,
|
||
|
|
response_model=None,
|
||
|
|
)
|
||
|
|
async def delete_goal(
|
||
|
|
goal_id: int,
|
||
|
|
session: AsyncSession = Depends(get_session),
|
||
|
|
) -> None:
|
||
|
|
goal = await session.get(RetirementGoal, goal_id)
|
||
|
|
if goal is None:
|
||
|
|
raise HTTPException(status_code=404, detail="Goal not found")
|
||
|
|
await session.execute(delete(RetirementGoal).where(RetirementGoal.id == goal_id))
|
||
|
|
await session.commit()
|