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

from __future__ import annotations

from enum import StrEnum

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


class Type(StrEnum):
    STORAGE = 'STORAGE'
    PV = 'PV'
    EV_CHARGER = 'EV_CHARGER'
    HVAC = 'HVAC'
    INDUSTRIAL_LOAD = 'INDUSTRIAL_LOAD'
    MICROGRID = 'MICROGRID'
    BASE_STATION = 'BASE_STATION'


class Constraints(BaseModel):
    model_config = ConfigDict(
        extra='forbid',
    )
    min_duration_min: conint(le=9007199254740991, gt=0) | None
    recovery_rate_mw_per_min: constr(pattern=r'^-?\d+(\.\d+)?$') | None


class Profile(BaseModel):
    model_config = ConfigDict(
        extra='forbid',
    )
    resource_id: constr(min_length=1)
    name: constr(min_length=1)
    type: Type
    rated_power_mw: constr(pattern=r'^-?\d+(\.\d+)?$')
    certified_adjustable_mw: constr(pattern=r'^-?\d+(\.\d+)?$')
    confidence: constr(pattern=r'^-?\d+(\.\d+)?$')
    reliability_score: constr(pattern=r'^-?\d+(\.\d+)?$')
    constraints: Constraints
    evidence_refs: list[constr(pattern=r'^[0-9a-f]{64}$')]
    updated_at: AwareDatetime


class FulfillmentItem(BaseModel):
    model_config = ConfigDict(
        extra='forbid',
    )
    market_date: constr(pattern=r'^\d{4}-\d{2}-\d{2}$')
    planned_mwh: constr(pattern=r'^-?\d+(\.\d+)?$')
    delivered_mwh: constr(pattern=r'^-?\d+(\.\d+)?$')


class Resource(BaseModel):
    model_config = ConfigDict(
        extra='forbid',
    )
    profile: Profile
    fulfillment: list[FulfillmentItem]


class PotentialAssessmentRequest(BaseModel):
    model_config = ConfigDict(
        extra='forbid',
    )
    market_date: constr(pattern=r'^\d{4}-\d{2}-\d{2}$')
    resources: list[Resource] = Field(..., min_length=1)
    features_snapshot_ref: constr(pattern=r'^[0-9a-f]{64}$')