serialize version as an int

AntaresSimulatorTeam · Sep 25, 2024 · d6b4c50 · d6b4c50
1 parent 619946a
commit d6b4c50
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/antarest/study/model.py b/antarest/study/model.py
@@ -19,7 +19,7 @@
 from pathlib import Path
 
 from antares.study.version import StudyVersion
-from pydantic import BaseModel, field_validator
+from pydantic import BaseModel, field_validator, field_serializer
 from sqlalchemy import (  # type: ignore
     Boolean,
     Column,
@@ -359,6 +359,10 @@ class StudyMetadataDTO(BaseModel):
     folder: t.Optional[str] = None
     tags: t.List[str] = []
 
+    @field_serializer('version', when_used='json')
+    def serialize_version(self, version: StudyVersion):
+        return version.__int__()
+
     @field_validator("horizon", mode="before")
     def transform_horizon_to_str(cls, val: t.Union[str, int, None]) -> t.Optional[str]:
         # horizon can be an int.