55 lines
1.3 KiB
Python
55 lines
1.3 KiB
Python
|
# 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}$')
|