62 lines
2.6 KiB
Python
62 lines
2.6 KiB
Python
from fastapi import APIRouter, Depends, HTTPException
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
from sqlalchemy import select
|
|
from app.core.database import get_db
|
|
from app.core.security import get_current_user_id
|
|
from app.models.familia import Familia, MembroFamilia
|
|
from app.models.falecido import Falecido
|
|
from app.schemas.schemas import FamiliaCreate, FamiliaOut, MembroCreate, MembroOut, FalecidoCreate, FalecidoOut
|
|
from app.services.checklist_engine import gerar_checklist
|
|
from app.services.beneficio_scanner import escanear_beneficios
|
|
|
|
router = APIRouter(prefix="/familias", tags=["familias"])
|
|
|
|
@router.get("/", response_model=list[FamiliaOut])
|
|
async def listar(user_id: int = Depends(get_current_user_id), db: AsyncSession = Depends(get_db)):
|
|
result = await db.execute(select(Familia).where(Familia.usuario_id == user_id))
|
|
return result.scalars().all()
|
|
|
|
@router.post("/", response_model=FamiliaOut)
|
|
async def criar(data: FamiliaCreate, user_id: int = Depends(get_current_user_id), db: AsyncSession = Depends(get_db)):
|
|
f = Familia(nome=data.nome, usuario_id=user_id)
|
|
db.add(f)
|
|
await db.commit()
|
|
await db.refresh(f)
|
|
return f
|
|
|
|
@router.get("/{id}", response_model=FamiliaOut)
|
|
async def detalhe(id: int, user_id: int = Depends(get_current_user_id), db: AsyncSession = Depends(get_db)):
|
|
result = await db.execute(select(Familia).where(Familia.id == id, Familia.usuario_id == user_id))
|
|
f = result.scalar_one_or_none()
|
|
if not f:
|
|
raise HTTPException(404, "Família não encontrada")
|
|
return f
|
|
|
|
@router.post("/{id}/membros", response_model=MembroOut)
|
|
async def add_membro(id: int, data: MembroCreate, db: AsyncSession = Depends(get_db)):
|
|
m = MembroFamilia(familia_id=id, **data.model_dump())
|
|
db.add(m)
|
|
await db.commit()
|
|
await db.refresh(m)
|
|
return m
|
|
|
|
@router.get("/{id}/membros", response_model=list[MembroOut])
|
|
async def listar_membros(id: int, db: AsyncSession = Depends(get_db)):
|
|
result = await db.execute(select(MembroFamilia).where(MembroFamilia.familia_id == id))
|
|
return result.scalars().all()
|
|
|
|
@router.post("/{id}/falecido", response_model=FalecidoOut)
|
|
async def registrar_falecido(id: int, data: FalecidoCreate, db: AsyncSession = Depends(get_db)):
|
|
f = Falecido(familia_id=id, **data.model_dump())
|
|
db.add(f)
|
|
await db.commit()
|
|
await db.refresh(f)
|
|
await gerar_checklist(db, id, f)
|
|
await escanear_beneficios(db, id, f)
|
|
return f
|
|
|
|
@router.get("/{id}/falecidos", response_model=list[FalecidoOut])
|
|
async def listar_falecidos(id: int, db: AsyncSession = Depends(get_db)):
|
|
result = await db.execute(select(Falecido).where(Falecido.familia_id == id))
|
|
return result.scalars().all()
|