bdlg2023-sms-back/app/importData/validation.py

from typing import Optional

import pandas as pd
import pandera as pa

from pandera.typing import Series


class ContactSchema(pa.DataFrameModel):
    key: Series[str] = pa.Field(unique=True)
    Prénom: Series[str]
    Nom: Series[str]
    Mail: Series[str]
    Tél: Series[str]
    SMS: Series[bool]


class CreneauDataSchema(pa.DataFrameModel):
    title: Series[str]
    lieu: Optional[Series[str]]
    description: Series[str]
    responsable: Series[str]
    tags: Optional[Series[str]]


class CreneauSchema(pa.DataFrameModel):
    id: Series[str]
    template_id: Series[str]
    nom: Series[str]
    benevole_nom: Series[str]
    ligne: Series[int]
    start: Series[pd.DatetimeTZDtype] = pa.Field(
        dtype_kwargs={"unit": "ns", "tz": "UTC"}, coerce=True
    )
    end: Series[pd.DatetimeTZDtype] = pa.Field(
        dtype_kwargs={"unit": "ns", "tz": "UTC"}, coerce=True
    )