# generated by datamodel-codegen:
#   filename:  flexibility_envelope.json

from __future__ import annotations

from enum import StrEnum
from typing import Literal

from pydantic import AwareDatetime, BaseModel, ConfigDict, Field, RootModel, constr


class Level(StrEnum):
    SITE = 'SITE'
    AGGREGATION_UNIT = 'AGGREGATION_UNIT'
    PROVINCE = 'PROVINCE'


class Provider(BaseModel):
    model_config = ConfigDict(
        extra='forbid',
    )
    unit_id: constr(min_length=1)
    level: Level


class Value(RootModel[constr(pattern=r'^-?\d+(\.\d+)?$')]):
    root: constr(pattern=r'^-?\d+(\.\d+)?$')


class UpMw(BaseModel):
    model_config = ConfigDict(
        extra='forbid',
    )
    interval_minutes: Literal[15]
    date: constr(pattern=r'^\d{4}-\d{2}-\d{2}$')
    values: list[Value] = Field(..., max_length=96, min_length=96)


class DownMw(BaseModel):
    model_config = ConfigDict(
        extra='forbid',
    )
    interval_minutes: Literal[15]
    date: constr(pattern=r'^\d{4}-\d{2}-\d{2}$')
    values: list[Value] = Field(..., max_length=96, min_length=96)


class FlexibilityEnvelope(BaseModel):
    model_config = ConfigDict(
        extra='forbid',
    )
    id: constr(min_length=1)
    provider: Provider
    market_date: constr(pattern=r'^\d{4}-\d{2}-\d{2}$')
    up_mw: UpMw
    down_mw: DownMw
    valid_until: AwareDatetime
    schema_version: constr(min_length=1)
    signature: str | None