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, SexoSchemaBase

router = APIRouter()


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


@router.post('/sexos/', response_model=SexoSchema)
def crear_sexo(sexo: SexoSchemaBase, db: Session = Depends(get_db)):
    instancia = Sexo(**sexo.dict())
    db.add(instancia)
    db.commit()
    db.refresh(instancia)

    return instancia


@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)):

    queryset = db.query(Persona)

    if search:
        queryset = queryset.filter(Persona.documento == search)

    return queryset.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