mirror of https://github.com/craigerl/aprsd.git
81 lines
2.4 KiB
Python
81 lines
2.4 KiB
Python
import datetime
|
|
import decimal
|
|
import json
|
|
import sys
|
|
|
|
from aprsd.packets import core
|
|
|
|
|
|
class EnhancedJSONEncoder(json.JSONEncoder):
|
|
def default(self, obj):
|
|
if isinstance(obj, datetime.datetime):
|
|
args = (
|
|
"year", "month", "day", "hour", "minute",
|
|
"second", "microsecond",
|
|
)
|
|
return {
|
|
"__type__": "datetime.datetime",
|
|
"args": [getattr(obj, a) for a in args],
|
|
}
|
|
elif isinstance(obj, datetime.date):
|
|
args = ("year", "month", "day")
|
|
return {
|
|
"__type__": "datetime.date",
|
|
"args": [getattr(obj, a) for a in args],
|
|
}
|
|
elif isinstance(obj, datetime.time):
|
|
args = ("hour", "minute", "second", "microsecond")
|
|
return {
|
|
"__type__": "datetime.time",
|
|
"args": [getattr(obj, a) for a in args],
|
|
}
|
|
elif isinstance(obj, datetime.timedelta):
|
|
args = ("days", "seconds", "microseconds")
|
|
return {
|
|
"__type__": "datetime.timedelta",
|
|
"args": [getattr(obj, a) for a in args],
|
|
}
|
|
elif isinstance(obj, decimal.Decimal):
|
|
return {
|
|
"__type__": "decimal.Decimal",
|
|
"args": [str(obj)],
|
|
}
|
|
else:
|
|
return super().default(obj)
|
|
|
|
|
|
class SimpleJSONEncoder(json.JSONEncoder):
|
|
def default(self, obj):
|
|
if isinstance(obj, datetime.datetime):
|
|
return obj.isoformat()
|
|
elif isinstance(obj, datetime.date):
|
|
return str(obj)
|
|
elif isinstance(obj, datetime.time):
|
|
return str(obj)
|
|
elif isinstance(obj, datetime.timedelta):
|
|
return str(obj)
|
|
elif isinstance(obj, decimal.Decimal):
|
|
return str(obj)
|
|
elif isinstance(obj, core.Packet):
|
|
return obj.to_dict()
|
|
else:
|
|
return super().default(obj)
|
|
|
|
|
|
class EnhancedJSONDecoder(json.JSONDecoder):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(
|
|
*args, object_hook=self.object_hook,
|
|
**kwargs,
|
|
)
|
|
|
|
def object_hook(self, d):
|
|
if "__type__" not in d:
|
|
return d
|
|
o = sys.modules[__name__]
|
|
for e in d["__type__"].split("."):
|
|
o = getattr(o, e)
|
|
args, kwargs = d.get("args", ()), d.get("kwargs", {})
|
|
return o(*args, **kwargs)
|