from datetime import date

from sqlalchemy import select
from sqlalchemy.ext.asyncio import AsyncSession

from app.models.objekt import Objekt
from app.models.personal import (
    BenutzerQualifikation,
    ObjekttypQualifikationsanforderung,
    Qualifikationstyp,
)


async def pruefe_berechtigung(db: AsyncSession, *, benutzer_id: int, objekt_id: int) -> list[Qualifikationstyp]:
    """Nutzer-Beispiel Personal-Modul: "wer darf dieses Fahrzeug fahren?".
    Liefert die Qualifikationstypen, die der Objekttyp verlangt, aber der
    Benutzer nicht (mehr) gültig nachgewiesen hat - leere Liste = berechtigt."""
    objekt = await db.get(Objekt, objekt_id)
    if objekt is None:
        return []

    erforderlich = (
        (
            await db.execute(
                select(Qualifikationstyp)
                .join(
                    ObjekttypQualifikationsanforderung,
                    ObjekttypQualifikationsanforderung.qualifikationstyp_id == Qualifikationstyp.id,
                )
                .where(ObjekttypQualifikationsanforderung.objekttyp_id == objekt.objekttyp_id)
            )
        )
        .scalars()
        .all()
    )
    if not erforderlich:
        return []

    heute = date.today()
    gueltige_ids = set(
        (
            await db.execute(
                select(BenutzerQualifikation.qualifikationstyp_id).where(
                    BenutzerQualifikation.benutzer_id == benutzer_id,
                    (BenutzerQualifikation.gueltig_bis.is_(None))
                    | (BenutzerQualifikation.gueltig_bis >= heute),
                )
            )
        )
        .scalars()
        .all()
    )
    return [q for q in erforderlich if q.id not in gueltige_ids]