CmsAnalysisInput.py 429 B

12345678910111213141516
  1. from pydantic import BaseModel, model_validator
  2. from typing import List, Optional
  3. class CmsAnalysisInput(BaseModel):
  4. ids: List[int] = []
  5. windCode: str
  6. analysisType: str
  7. fmin: Optional[int] = None
  8. fmax: Optional[int] = None
  9. @model_validator(mode='before')
  10. def convert_ids(cls, values):
  11. if isinstance(values.get('ids'), int):
  12. values['ids'] = [values['ids']]
  13. return values