schemas.py

from typing import List, Union
import datetime
from pydantic import BaseModel


class EnvMonitorBase(BaseModel):
    temperature01: int
    temperature02: int


class EnvMonitorCreate(EnvMonitorBase):
    pass


class EnvMonitor(EnvMonitorBase):
    id: int
    timestamp: datetime.datetime

    class Config:
        orm_mode = True


class UBLBase(BaseModel):
    redshift_entitled: int
    redshift_used: int
    redshift_available: int

    # allow None since a migration happened and None was default
    nuke_entitled: int | None = ...
    nuke_used: int | None = ...
    nuke_available: int | None = ...

    vray_entitled: int | None = ...
    vray_used: int | None = ...
    vray_available: int | None = ...


class UBLCreate(UBLBase):
    pass


class UBL(UBLBase):
    id: int
    timestamp: datetime.datetime

    class Config:
        orm_mode = True