-
Notifications
You must be signed in to change notification settings - Fork 43
/
Copy pathcustom_class_serializer.py
43 lines (32 loc) · 1.02 KB
/
custom_class_serializer.py
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
from plum import dispatch
from datetime import datetime
from serde import (
serde,
field,
)
from serde.json import from_json, to_json
from typing import Type, Any
class MySerializer:
@dispatch
def serialize(self, value: datetime) -> str:
return value.strftime("%d/%m/%y")
class MyDeserializer:
@dispatch
def deserialize(self, cls: Type[datetime], value: Any) -> datetime:
return datetime.strptime(value, "%d/%m/%y")
@serde(class_serializer=MySerializer(), class_deserializer=MyDeserializer())
class Foo:
a: datetime
b: datetime = field(
serializer=lambda x: x.strftime("%y.%m.%d"),
deserializer=lambda x: datetime.strptime(x, "%y.%m.%d"),
)
c: list[datetime] = field(default_factory=list)
def main() -> None:
dt = datetime(2021, 1, 1, 0, 0, 0)
f = Foo(dt, dt, [dt])
print(f"Into Json: {to_json(f)}")
s = '{"a": "01/01/21", "b": "01.01.21", "c": ["01/01/21"]}'
print(f"From Json: {from_json(Foo, s)}")
if __name__ == "__main__":
main()