53 lines
1.3 KiB
Python
53 lines
1.3 KiB
Python
|
import copy
|
||
|
|
import datetime
|
||
|
|
import io
|
||
|
|
from json import *
|
||
|
|
from typing import *
|
||
|
|
|
||
|
|
from .const import UTF_8
|
||
|
|
from .type import T
|
||
|
|
|
||
|
|
|
||
|
|
class JSON(object):
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def marshal(obj: Any, indent=None) -> Optional[str]:
|
||
|
|
if obj is None:
|
||
|
|
return None
|
||
|
|
return dumps(obj, cls=Encoder, indent=indent, ensure_ascii=False)
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def unmarshal(json_str: str, clazz: Type[T]) -> T:
|
||
|
|
dict_obj = loads(json_str)
|
||
|
|
return clazz(dict_obj)
|
||
|
|
|
||
|
|
|
||
|
|
class Encoder(JSONEncoder):
|
||
|
|
def default(self, o: Any) -> Any:
|
||
|
|
if isinstance(o, io.BufferedReader):
|
||
|
|
return o.__str__()
|
||
|
|
if hasattr(o, "__dict__"):
|
||
|
|
return filter_null(copy.deepcopy(vars(o)))
|
||
|
|
if isinstance(o, datetime.datetime):
|
||
|
|
return o.strftime("%Y-%m-%d %H:%M:%S")
|
||
|
|
if isinstance(o, bytes):
|
||
|
|
return str(o, encoding=UTF_8)
|
||
|
|
if isinstance(o, int):
|
||
|
|
return int(o)
|
||
|
|
if isinstance(o, float):
|
||
|
|
return float(o)
|
||
|
|
if isinstance(o, set):
|
||
|
|
return list(o)
|
||
|
|
return super().default(o)
|
||
|
|
|
||
|
|
|
||
|
|
def filter_null(d: Dict) -> Dict:
|
||
|
|
if isinstance(d, dict):
|
||
|
|
for k, v in list(d.items()):
|
||
|
|
if isinstance(v, dict):
|
||
|
|
filter_null(v)
|
||
|
|
elif v is None:
|
||
|
|
del d[k]
|
||
|
|
|
||
|
|
return d
|