from pydantic import BaseModel, ConfigDict

from app.models.auth import RolleTyp


class BenutzerCreate(BaseModel):
    name: str
    login: str
    passwort: str
    email: str | None = None
    rollen: list[RolleTyp] = []
    einheit_id: int | None = None


class BenutzerRead(BaseModel):
    model_config = ConfigDict(from_attributes=True)
    id: int
    name: str
    login: str
    email: str | None
    aktiv: bool
    rollen: list[str]
    einheit_id: int | None

    @classmethod
    def from_orm_benutzer(cls, benutzer) -> "BenutzerRead":
        return cls(
            id=benutzer.id,
            name=benutzer.name,
            login=benutzer.login,
            email=benutzer.email,
            aktiv=benutzer.aktiv,
            rollen=benutzer.rollen_namen,
            einheit_id=benutzer.einheit_id,
        )


class BenutzerUpdate(BaseModel):
    name: str | None = None
    email: str | None = None
    aktiv: bool | None = None
    rollen: list[RolleTyp] | None = None
    einheit_id: int | None = None