gridappsd-python/gridappsd-python-lib/gridappsd/json_extension.py at 9c349606da7034a275cb564f80d3b248eca6697b · GRIDAPPSD/gridappsd-python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
import dataclasses
import json as _json
from typing import Any, TextIO


@dataclasses.dataclass
class JsonComplex:
    real: float
    imag: float


def jsonDecoderExtension(obj: Any):
    rv = obj
    if isinstance(obj, dict):
        try:
            complexInstance = JsonComplex(**obj)
            rv = complex(complexInstance.real, complexInstance.imag)
        except (TypeError, KeyError):
            rv = obj
    return rv


class JsonEncoderExtension(_json.JSONEncoder):
    def default(self, obj: Any) -> Any:
        rv = None
        if isinstance(obj, complex):
            jsonComplexInstance = JsonComplex(real=obj.real, imag=obj.imag)
            rv = dataclasses.asdict(jsonComplexInstance)
        elif dataclasses.is_dataclass(obj):
            rv = dataclasses.asdict(obj)
        else:
            rv = super().default(obj)
        return rv


def dump(
    data: Any,
    fo: TextIO,
    *,
    skipkeys=False,
    ensure_ascii=True,
    check_circular=True,
    allow_nan=True,
    indent=None,
    separators=None,
    default=None,
    sort_keys=False,
    **kw,
) -> None:
    _json.dump(
        data,
        fo,
        skipkeys=skipkeys,
        ensure_ascii=ensure_ascii,
        check_circular=check_circular,
        allow_nan=allow_nan,
        cls=JsonEncoderExtension,
        indent=indent,
        separators=separators,
        default=default,
        sort_keys=sort_keys,
        **kw,
    )


def dumps(
    data: Any,
    *,
    skipkeys=False,
    ensure_ascii=True,
    check_circular=True,
    allow_nan=True,
    indent=None,
    separators=None,
    default=None,
    sort_keys=False,
    **kw,
) -> str:
    rv = _json.dumps(
        data,
        skipkeys=skipkeys,
        ensure_ascii=ensure_ascii,
        check_circular=check_circular,
        allow_nan=allow_nan,
        cls=JsonEncoderExtension,
        indent=indent,
        separators=separators,
        default=default,
        sort_keys=sort_keys,
        **kw,
    )
    return rv


def load(
    fo: TextIO, *, cls=None, parse_float=None, parse_int=None, parse_constant=None, object_pairs_hook=None, **kw
) -> Any:
    rv = _json.load(
        fo,
        cls=cls,
        object_hook=jsonDecoderExtension,
        parse_float=parse_float,
        parse_int=parse_int,
        parse_constant=parse_constant,
        object_pairs_hook=object_pairs_hook,
        **kw,
    )
    return rv


def loads(
    data: str, *, cls=None, parse_float=None, parse_int=None, parse_constant=None, object_pairs_hook=None, **kw
) -> Any:
    rv = _json.loads(
        data,
        cls=cls,
        object_hook=jsonDecoderExtension,
        parse_float=parse_float,
        parse_int=parse_int,
        parse_constant=parse_constant,
        object_pairs_hook=object_pairs_hook,
        **kw,
    )
    return rv