mirror of
https://github.com/PaiGramTeam/genshin-wiki.git
synced 2024-11-23 08:00:59 +00:00
33 lines
824 B
Python
33 lines
824 B
Python
from typing import Any
|
|
from datetime import datetime
|
|
from json import JSONEncoder as _JSONEncoder
|
|
|
|
from orjson import dumps as json_dumps
|
|
from orjson import loads as json_loads
|
|
from orjson import JSONDecodeError as _JSONDecodeError
|
|
from orjson import JSONEncodeError as _JSONEncodeError
|
|
|
|
__all__ = ("JSONEncodeError", "JSONDecodeError", "dumps", "loads")
|
|
|
|
JSONEncodeError = _JSONEncodeError
|
|
JSONDecodeError = _JSONDecodeError
|
|
|
|
|
|
def _default(value: Any):
|
|
if isinstance(value, datetime):
|
|
return value.timestamp()
|
|
return value
|
|
|
|
|
|
def loads(*args, **kwargs) -> Any:
|
|
return json_loads(*args, **kwargs)
|
|
|
|
|
|
def dumps(*args, **kwargs) -> str:
|
|
default = kwargs.pop("default", _default)
|
|
return json_dumps(*args, default=default, **kwargs).decode(encoding="utf-8")
|
|
|
|
|
|
class JSONEncoder(_JSONEncoder):
|
|
pass
|