def marshall(proto, spec=None, **kwargs):
"""Marshall a proto with DeckGL chart info.
See DeltaGenerator.deck_gl_chart for docs.
"""
if "data" in kwargs:
# TODO: Remove this check after 2019-11-28.
raise Exception(
"The `data` parameter is deprecated. Use `st.map` or"
"specify a spec dict in `st.deck_gl_chart`."
)
data = []
if spec is None:
spec = dict()
# Merge spec with unflattened kwargs, where kwargs take precedence.
# This only works for string keys, but kwarg keys are strings anyways.
spec = dict(spec, **dicttools.unflatten(kwargs))
if "layers" not in spec:
spec["layers"] = []
# Syntax sugar: if no layers defined and data is passed at the top
# level, create a scatterplot layer with the top-level data by default.
if data is not None:
spec["layers"].append({"data": data, "type": "ScatterplotLayer"})
for layer in spec["layers"]:
# Don't add layers that have no data.
if "data" not in layer:
continue
# Remove DataFrame because it's not JSON-serializable
data = layer.pop("data")
layer_proto = proto.layers.add()
fixed_layer = case_converters.convert_dict_keys(
case_converters.to_lower_camel_case, layer
)
layer_proto.spec = json.dumps(fixed_layer)
# TODO: If several layers use the same data frame, the data gets resent
# for each layer. Need to improve this.
data_frame_proto.marshall_data_frame(data, layer_proto.data)
del spec["layers"]
# Dump JSON after removing DataFrames (see loop above), because DataFrames
# are not JSON-serializable.
proto.spec = json.dumps(spec)
def marshall(proto, data=None, spec=None, **kwargs):
"""Marshall a proto with DeckGL chart info.
See DeltaGenerator.deck_gl_chart for docs.
"""
if data is None:
data = []
if spec is None:
spec = dict()
# Merge spec with unflattened kwargs, where kwargs take precedence.
# This only works for string keys, but kwarg keys are strings anyways.
spec = dict(spec, **dicttools.unflatten(kwargs))
if 'layers' not in spec:
spec['layers'] = []
# Syntax sugar: if no layers defined and data is passed at the top
# level, create a scatterplot layer with the top-level data by default.
if data is not None:
spec['layers'].append({
'data': data,
'type': 'ScatterplotLayer',
})
for layer in spec['layers']:
# Don't add layers that have no data.
if 'data' not in layer:
continue
# Remove DataFrame because it's not JSON-serializable
data = layer.pop('data')
layer_proto = proto.layers.add()
fixed_layer = case_converters.convert_dict_keys(
case_converters.to_lower_camel_case, layer)
layer_proto.spec = json.dumps(fixed_layer)
# TODO: If several layers use the same data frame, the data gets resent
# for each layer. Need to improve this.
data_frame_proto.marshall_data_frame(data, layer_proto.data)
del spec['layers']
# Dump JSON after removing DataFrames (see loop above), because DataFrames
# are not JSON-serializable.
proto.spec = json.dumps(spec)