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

from __future__ import annotations

from enum import StrEnum
from typing import Literal

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


class Kind(StrEnum):
    LOAD = 'LOAD'
    PV = 'PV'
    PRICE = 'PRICE'


class Unit(StrEnum):
    mw = 'mw'
    yuan_per_mwh = 'yuan_per_mwh'


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


class HistoryItem(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 Exogenous(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 ForecastRequest(BaseModel):
    model_config = ConfigDict(
        extra='forbid',
    )
    kind: Kind
    market_date: constr(pattern=r'^\d{4}-\d{2}-\d{2}$')
    unit: Unit
    history: list[HistoryItem] = Field(..., min_length=1)
    exogenous: dict[str, Exogenous]
    features_snapshot_ref: constr(pattern=r'^[0-9a-f]{64}$')