PDBA/bot-base
8
2
Fork 0
Code Issues Pull Requests Releases Wiki Activity

[base-storage] Ajout de la possibilité de passer des arguments au décodage

Browse Source
This commit is contained in:
Louis Chauvet 2020-03-22 19:55:05 +01:00
parent 7a34e57a2d
commit 2f1a69e9b6
Signed by: fomys
GPG Key ID: 1ECA046A9615ABA0
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
storage/jsonencoder.py
View File

@ -10,8 +10,8 @@ class Encoder(json.JSONEncoder):
super().__init__()
self.custom = {}
def register(self, type_, encode, decode):
self.custom.update({type_: (encode, decode)})
def register(self, type_, encode, decode, decode_args={}):
self.custom.update({type_: (encode, decode, decode_args)})
def default(self, obj):
if isinstance(obj, tuple(self.custom.keys())):
@ -26,7 +26,7 @@ class Encoder(json.JSONEncoder):
if data_type in dct:
for ty in self.custom.keys():
if str(ty) == dct[data_type]:
return self.custom[ty][1](dct[content])
return self.custom[ty][1](dct[content], **self.custom[ty][2])
if dct[data_type] == "datetime.datetime":
return datetime.datetime.fromisoformat(dct['iso'])
elif dct[data_type] == "datetime.timedelta":