♻️ (schema.py): refactor serialize_model method in Record class to use self instead of cls and improve method signature for better readability and maintainability

This commit is contained in:
ogabrielluiz 2024-06-06 11:52:03 -03:00
commit f87e49e0dc

View file

@ -30,9 +30,9 @@ class Record(BaseModel):
values["data"][key] = values[key]
return values
@model_serializer(mode="json")
def serialize_model(cls, obj):
data = {k: v.to_json() if hasattr(v, "to_json") else v for k, v in obj.data.items()}
@model_serializer(mode="plain", when_used="json")
def serialize_model(self):
data = {k: v.to_json() if hasattr(v, "to_json") else v for k, v in self.data.items()}
return data
def get_text(self):