from typing import List, Optional

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

from database import get_db
from persona.models import Sexo, Persona
from persona.schemas import SexoSchema, PersonaSchema

router = APIRouter()


@router.get("/sexos/", response_model=List[SexoSchema])
def obtener_sexos(db: Session = Depends(get_db)):
    queryset = db.query(Sexo).all()
    return queryset


@router.get('/personas/', response_model=List[PersonaSchema])
def obtener_personas(
        skip: int = 0,
        limit: int = 50,
        search: Optional[str] = None,
        db: Session = Depends(get_db)):

    if search:
        return db.query(Persona).filter(Persona.documento == search).all()

    return db.query(Persona).offset(skip).limit(limit).all()


@router.get('/personas/{persona_id}/', response_model=PersonaSchema)
def obtener_persona_por_id(persona_id: int, db: Session = Depends(get_db)):
    persona = db.query(Persona).filter(Persona.id == persona_id).first()

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

    return persona