Spaces:
Sleeping
Sleeping
File size: 1,170 Bytes
177c40c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
"""
Pydantic utilities for JSON serialization.
This provides a base class with camelCase conversion for REST API compatibility.
"""
from solverforge_legacy.solver.score import HardSoftScore
from typing import Any
from pydantic import BaseModel, ConfigDict, PlainSerializer, BeforeValidator
from pydantic.alias_generators import to_camel
# Score serialization (HardSoftScore -> string like "0hard/0soft")
ScoreSerializer = PlainSerializer(
lambda score: str(score) if score is not None else None,
return_type=str | None
)
def validate_score(v: Any) -> Any:
"""Parse score from string format."""
if isinstance(v, HardSoftScore) or v is None:
return v
if isinstance(v, str):
return HardSoftScore.parse(v)
raise ValueError('"score" should be a string')
ScoreValidator = BeforeValidator(validate_score)
class JsonDomainBase(BaseModel):
"""
Base class for Pydantic models with camelCase JSON serialization.
Converts Python snake_case to JavaScript camelCase automatically.
"""
model_config = ConfigDict(
alias_generator=to_camel,
populate_by_name=True,
from_attributes=True,
)
|