12345678910111213141516 |
- from pydantic import BaseModel, model_validator
- from typing import List, Optional
- class CmsAnalysisInput(BaseModel):
- ids: List[int] = []
- windCode: str
- analysisType: str
- fmin: Optional[int] = None
- fmax: Optional[int] = None
- @model_validator(mode='before')
- def convert_ids(cls, values):
- if isinstance(values.get('ids'), int):
- values['ids'] = [values['ids']]
- return values
|