plotly.py/plotly/serializers.py at swarm-plot · rl-utility-man/plotly.py · GitHub

Name: plotly.py/plotly/serializers.py at swarm-plot · rl-utility-man/plotly.py · GitHub
Rating: 4.5 (1754 reviews)
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
from .basedatatypesimportUndefined
from .optional_importsimportget_module

np=get_module("numpy")


def_py_to_js(v, widget_manager):
"""
 Python -> Javascript ipywidget serializer

 This function must repalce all objects that the ipywidget library
 can't serialize natively (e.g. numpy arrays) with serializable
 representations

 Parameters
 ----------
 v
 Object to be serialized
 widget_manager
 ipywidget widget_manager (unused)

 Returns
 -------
 any
 Value that the ipywidget library can serialize natively
 """

# Handle dict recursively
# -----------------------
ifisinstance(v, dict):
return {k: _py_to_js(v, widget_manager) fork, vinv.items()}

# Handle list/tuple recursively
# -----------------------------
elifisinstance(v, (list, tuple)):
return [_py_to_js(v, widget_manager) forvinv]

# Handle numpy array
# ------------------
elifnpisnotNoneandisinstance(v, np.ndarray):
# Convert 1D numpy arrays with numeric types to memoryviews with
# datatype and shape metadata.
if (
v.ndim==1
andv.dtype.kindin ["u", "i", "f"]
andv.dtype!="int64"
andv.dtype!="uint64"
 ):

# We have a numpy array the we can directly map to a JavaScript
# Typed array
return {"buffer": memoryview(v), "dtype": str(v.dtype), "shape": v.shape}
else:
# Convert all other numpy arrays to lists
returnv.tolist()

# Handle Undefined
# ----------------
ifvisUndefined:
return"_undefined_"

# Handle simple value
# -------------------
else:
returnv


def_js_to_py(v, widget_manager):
"""
 Javascript -> Python ipywidget deserializer

 Parameters
 ----------
 v
 Object to be deserialized
 widget_manager
 ipywidget widget_manager (unused)

 Returns
 -------
 any
 Deserialized object for use by the Python side of the library
 """
# Handle dict
# -----------
ifisinstance(v, dict):
return {k: _js_to_py(v, widget_manager) fork, vinv.items()}

# Handle list/tuple
# -----------------
elifisinstance(v, (list, tuple)):
return [_js_to_py(v, widget_manager) forvinv]

# Handle Undefined
# ----------------
elifisinstance(v, str) andv=="_undefined_":
returnUndefined

# Handle simple value
# -------------------
else:
returnv


# Custom serializer dict for use in ipywidget traitlet definitions
custom_serializers= {"from_json": _js_to_py, "to_json": _py_to_js}