bot-base/storage/jsonencoder.py

import datetime
import json

data_type = "__data_type"
content = "__content"


class Encoder(json.JSONEncoder):
    def __init__(self):
        super().__init__()
        self.custom = {}

    def register(self, type_, encode, decode):
        self.custom.update({type_: (encode, decode)})

    def default(self, obj):
        if isinstance(obj, tuple(self.custom.keys())):
            return {data_type: f'{type(obj)}', content: self.custom[type(obj)][0](obj)}
        if isinstance(obj, (datetime.datetime)):
            return {data_type: 'datetime.datetime', 'iso': obj.isoformat()}
        if isinstance(obj, (datetime.timedelta)):
            return {data_type: 'datetime.timedelta', 'totalseconds': obj.total_seconds()}
        return json.JSONEncoder.default(self, obj)

    def hook(self, dct):
        if data_type in dct:
            for ty in self.custom.keys():
                if str(ty) == dct[data_type]:
                    return self.custom[ty][1](dct[content])
            if dct[data_type] == "datetime.datetime":
                return datetime.datetime.fromisoformat(dct['iso'])
            elif dct[data_type] == "datetime.timedelta":
                return datetime.timedelta(seconds=dct['totalseconds'])
        return dct