from fastapi import APIRouter, Depends, HTTPException
from sqlalchemy.orm import Session

from agente.models import Agente
from agente.schemas import AgenteSchema
from database import get_db

router = APIRouter()


@router.get('/agentes/{persona_id}/', response_model=AgenteSchema)
def obtener_agente_por_persona_id(persona_id: int, db: Session = Depends(get_db)):
    agente = db.query(Agente).filter(Agente.persona_id == persona_id).first()

    if agente is None:
        raise HTTPException(status_code=404, detail=f"No se encontró el Agente para la Persona con id {persona_id}")

    return agente