fire-planner/fire_planner/api/life_events.py
Viktor Barzin ee6ed1d3c4
All checks were successful
ci/woodpecker/push/woodpecker Pipeline was successful
api: expand FastAPI surface for scenarios, networth, life-events, goals, simulate
Adds the read+write endpoints the frontend needs to drive a
ProjectionLab-style UX on top of the existing engine.

- /networth, /networth/history    — NW total + per-account from
                                     account_snapshot (frontend chart)
- /scenarios CRUD + projection    — list/get/create/patch/delete user
                                     scenarios; cartesian read-only
- /scenarios/{id}/life-events     — life event CRUD nested under scenario
- /life-events/{id}               — patch + delete by id
- /scenarios/{id}/goals,
  /goals/{id}                     — retirement goal CRUD
- /simulate, /compare             — sync, no-DB-write what-if endpoints

Auth: Bearer-token dependency on writes + simulate when API_BEARER_TOKEN
is set; reads always open (lock down via Authentik-fronted ingress in
prod). Existing /recompute keeps its bearer auth.

CORS middleware reads FRONTEND_ORIGINS (comma-separated) for the dev
SPA. Lifespan now provisions the SQLAlchemy engine + session_factory
on app.state and disposes them on shutdown.

40 new tests covering happy paths and validation. 172 tests total.
mypy strict + ruff clean (B008 ignore added — Depends() in defaults
is the canonical FastAPI pattern, not a bug).

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
2026-05-09 21:48:36 +00:00

93 lines
3 KiB
Python

"""Life-event 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.auth import require_bearer
from fire_planner.api.dependencies import get_session
from fire_planner.api.schemas import LifeEventCreate, LifeEventOut, LifeEventPatch
from fire_planner.db import LifeEvent, Scenario
router = APIRouter(tags=["life-events"])
@router.get(
"/scenarios/{scenario_id}/life-events",
response_model=list[LifeEventOut],
)
async def list_events(
scenario_id: int,
session: AsyncSession = Depends(get_session),
) -> list[LifeEvent]:
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(LifeEvent).where(LifeEvent.scenario_id == scenario_id).order_by(
LifeEvent.year_start, LifeEvent.id))).scalars().all()
return list(rows)
@router.post(
"/scenarios/{scenario_id}/life-events",
response_model=LifeEventOut,
status_code=201,
dependencies=[Depends(require_bearer)],
)
async def create_event(
scenario_id: int,
payload: LifeEventCreate,
session: AsyncSession = Depends(get_session),
) -> LifeEvent:
scen = await session.get(Scenario, scenario_id)
if scen is None:
raise HTTPException(status_code=404, detail="Scenario not found")
if payload.year_end is not None and payload.year_end < payload.year_start:
raise HTTPException(status_code=400, detail="year_end < year_start")
ev = LifeEvent(scenario_id=scenario_id, **payload.model_dump())
session.add(ev)
await session.commit()
await session.refresh(ev)
return ev
@router.patch(
"/life-events/{event_id}",
response_model=LifeEventOut,
dependencies=[Depends(require_bearer)],
)
async def patch_event(
event_id: int,
payload: LifeEventPatch,
session: AsyncSession = Depends(get_session),
) -> LifeEvent:
ev = await session.get(LifeEvent, event_id)
if ev is None:
raise HTTPException(status_code=404, detail="Event not found")
updates = payload.model_dump(exclude_unset=True)
for k, v in updates.items():
setattr(ev, k, v)
if ev.year_end is not None and ev.year_end < ev.year_start:
raise HTTPException(status_code=400, detail="year_end < year_start")
await session.commit()
await session.refresh(ev)
return ev
@router.delete(
"/life-events/{event_id}",
status_code=204,
response_model=None,
dependencies=[Depends(require_bearer)],
)
async def delete_event(
event_id: int,
session: AsyncSession = Depends(get_session),
) -> None:
ev = await session.get(LifeEvent, event_id)
if ev is None:
raise HTTPException(status_code=404, detail="Event not found")
await session.execute(delete(LifeEvent).where(LifeEvent.id == event_id))
await session.commit()