|
13 | 13 | ) |
14 | 14 |
|
15 | 15 |
|
| 16 | +def _maybe_parse_json(value): |
| 17 | + if isinstance(value, str): |
| 18 | + try: |
| 19 | + return json.loads(value) |
| 20 | + except json.JSONDecodeError as exc: |
| 21 | + raise ValueError('Invalid JSON payload') from exc |
| 22 | + return value |
| 23 | + |
| 24 | + |
16 | 25 | class RocketModel(ApiBaseModel): |
17 | 26 | NAME: ClassVar = "rocket" |
18 | 27 | METHODS: ClassVar = ("POST", "GET", "PUT", "DELETE") |
@@ -40,6 +49,42 @@ class RocketModel(ApiBaseModel): |
40 | 49 | rail_buttons: Optional[RailButtons] = None |
41 | 50 | tail: Optional[Tail] = None |
42 | 51 |
|
| 52 | + @field_validator('motor', mode='before') |
| 53 | + @classmethod |
| 54 | + def _coerce_motor(cls, value): |
| 55 | + return _maybe_parse_json(value) |
| 56 | + |
| 57 | + @field_validator('nose', mode='before') |
| 58 | + @classmethod |
| 59 | + def _coerce_nose(cls, value): |
| 60 | + return _maybe_parse_json(value) |
| 61 | + |
| 62 | + @field_validator('fins', mode='before') |
| 63 | + @classmethod |
| 64 | + def _coerce_fins(cls, value): |
| 65 | + value = _maybe_parse_json(value) |
| 66 | + if isinstance(value, dict): |
| 67 | + value = [value] |
| 68 | + return value |
| 69 | + |
| 70 | + @field_validator('parachutes', mode='before') |
| 71 | + @classmethod |
| 72 | + def _coerce_parachutes(cls, value): |
| 73 | + value = _maybe_parse_json(value) |
| 74 | + if isinstance(value, dict): |
| 75 | + value = [value] |
| 76 | + return value |
| 77 | + |
| 78 | + @field_validator('rail_buttons', mode='before') |
| 79 | + @classmethod |
| 80 | + def _coerce_rail_buttons(cls, value): |
| 81 | + return _maybe_parse_json(value) |
| 82 | + |
| 83 | + @field_validator('tail', mode='before') |
| 84 | + @classmethod |
| 85 | + def _coerce_tail(cls, value): |
| 86 | + return _maybe_parse_json(value) |
| 87 | + |
43 | 88 | @staticmethod |
44 | 89 | def UPDATED(): |
45 | 90 | return |
|
0 commit comments