TemperatureInput.py 409 B

1234567891011121314151617
  1. from typing import List
  2. from pydantic import BaseModel, model_validator
  3. class TemperatureInput(BaseModel):
  4. windCode: str
  5. windTurbineNumberList: List[str]
  6. startTime: str
  7. endTime: str
  8. @model_validator(mode='before')
  9. def ensure_list(cls, v):
  10. raw = v.get('windTurbineNumberList')
  11. if isinstance(raw, str):
  12. v['windTurbineNumberList'] = [raw]
  13. return v