33 lines
1.3 KiB
Python
33 lines
1.3 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.beneficio import Beneficio
|
|
from app.models.falecido import Falecido
|
|
from app.schemas.schemas import BeneficioOut
|
|
from app.services.beneficio_scanner import escanear_beneficios
|
|
|
|
router = APIRouter(prefix="/familias/{familia_id}/beneficios", tags=["beneficios"])
|
|
|
|
@router.get("/", response_model=list[BeneficioOut])
|
|
async def listar(familia_id: int, db: AsyncSession = Depends(get_db)):
|
|
result = await db.execute(select(Beneficio).where(Beneficio.familia_id == familia_id))
|
|
return result.scalars().all()
|
|
|
|
@router.post("/scan", response_model=list[BeneficioOut])
|
|
async def scan(familia_id: int, db: AsyncSession = Depends(get_db)):
|
|
# Delete existing and rescan
|
|
result = await db.execute(select(Beneficio).where(Beneficio.familia_id == familia_id))
|
|
for b in result.scalars().all():
|
|
await db.delete(b)
|
|
await db.commit()
|
|
|
|
result = await db.execute(select(Falecido).where(Falecido.familia_id == familia_id))
|
|
falecidos = result.scalars().all()
|
|
all_bens = []
|
|
for f in falecidos:
|
|
bens = await escanear_beneficios(db, familia_id, f)
|
|
all_bens.extend(bens)
|
|
return all_bens
|