Fix update method on empty and uninitialized array property (#1092)

* Fix for GH1072. Empty array properties are replaced during update

* Don't error when a literal property is set to it's correct value.

This avoids errors when performing updates like figure.update(data=[go.Scatter(...), ...])

* Fix two failing validator tests

_plotly_utils/basevalidators.py

@@ -1658,16 +1658,21 @@ class LiteralValidator(BaseValidator):
     """
     Validator for readonly literal values
     """
-    def __init__(self, plotly_name, parent_name, **kwargs):
-        super(LiteralValidator, self).__init__(plotly_name=plotly_name,
-                         parent_name=parent_name,
-                         **kwargs)
+    def __init__(self, plotly_name, parent_name, val, **kwargs):
+        super(LiteralValidator, self).__init__(
+            plotly_name=plotly_name,
+            parent_name=parent_name,
+            **kwargs)
+        self.val = val
 
     def validate_coerce(self, v):
-        raise ValueError("""\
-The '{plotly_name}' property of {parent_name} is read-only""".format(
-            plotly_name=self.plotly_name, parent_name=self.parent_name
-        ))
+        if v != self.val:
+            raise ValueError("""\
+    The '{plotly_name}' property of {parent_name} is read-only""".format(
+                plotly_name=self.plotly_name, parent_name=self.parent_name
+            ))
+        else:
+            return v
 
 
 class ImageUriValidator(BaseValidator):

_plotly_utils/tests/validators/test_basetraces_validator.py

@@ -10,7 +10,8 @@ def validator():
                                         'bar': 'Bar',
                                         'box': 'Box'},
                              plotly_name='prop',
-                             parent_name='parent')
+                             parent_name='parent',
+                             set_uid=True)
 
 
 # Tests

_plotly_utils/tests/validators/test_infoarray_validator.py

@@ -30,7 +30,7 @@ def validator_number3_free():
 # ----------
 # ### Acceptance ###
 @pytest.mark.parametrize('val', [
-    [1, 'A'], ('hello', 'world!'), [1, ()], [-1, 1]
+    [1, 'A'], ('hello', 'world!'), [1, set()], [-1, 1]
 ])
 def test_validator_acceptance_any2(val, validator_any2: InfoArrayValidator):
     coerce_val = validator_any2.validate_coerce(val)

_plotly_utils/tests/validators/test_literal_validator.py

@@ -0,0 +1,28 @@
+import pytest
+from _plotly_utils.basevalidators import LiteralValidator
+import numpy as np
+
+
+# Fixtures
+# --------
+@pytest.fixture()
+def validator():
+    return LiteralValidator('prop', 'parent', 'scatter')
+
+
+# Tests
+# -----
+# ### Acceptance ###
+@pytest.mark.parametrize('val', ['scatter'])
+def test_acceptance(val, validator):
+    assert validator.validate_coerce(val) is val
+
+
+# ### Test rejection ###
+@pytest.mark.parametrize('val',
+                         ['hello', (), [], [1, 2, 3], set(), '34'])
+def test_rejection(val, validator):
+    with pytest.raises(ValueError) as validation_failure:
+        validator.validate_coerce(val)
+
+    assert 'read-only' in str(validation_failure.value)

codegen/datatypes.py

@@ -254,11 +254,12 @@ def __init__(self""")
         for literal_node in literal_nodes:
             lit_name = literal_node.name_property
             lit_parent = literal_node.parent_path_str
-            lit_val = literal_node.node_data
+            lit_val = repr(literal_node.node_data)
             buffer.write(f"""
-        self._props['{lit_name}'] = '{lit_val}'
+        self._props['{lit_name}'] = {lit_val}
         self._validators['{lit_name}'] =\
-LiteralValidator(plotly_name='{lit_name}', parent_name='{lit_parent}')""")
+LiteralValidator(plotly_name='{lit_name}',\
+    parent_name='{lit_parent}', val={lit_val})""")
 
     buffer.write(f"""
     

codegen/utils.py

@@ -423,6 +423,8 @@ def get_validator_params(self):
                 if colorscale_node_list:
                     colorscale_path = colorscale_node_list[0].path_str
                     params['colorscale_path'] = repr(colorscale_path)
+            elif self.datatype == 'literal':
+                params['val'] = self.node_data
 
         return params
 

plotly/basedatatypes.py

@@ -378,7 +378,16 @@ def update(self, dict1=None, **kwargs):
             for d in [dict1, kwargs]:
                 if d:
                     for k, v in d.items():
-                        BaseFigure._perform_update(self[k], v)
+                        if self[k] == ():
+                            # existing data or frames property is empty
+                            # In this case we accept the v as is.
+                            if k == 'data':
+                                self.add_traces(v)
+                            else:
+                                # Accept v
+                                self[k] = v
+                        else:
+                            BaseFigure._perform_update(self[k], v)
 
         return self
 
@@ -1059,11 +1068,6 @@ def add_traces(self, data, rows=None, cols=None):
         ...                 rows=[1, 2], cols=[1, 1])
         """
 
-        if self._in_batch_mode:
-            self._batch_layout_edits.clear()
-            self._batch_trace_edits.clear()
-            raise ValueError('Traces may not be added in a batch context')
-
         # Validate traces
         data = self._data_validator.validate_coerce(data)
 
@@ -2133,8 +2137,15 @@ def _perform_update(plotly_obj, update_obj):
                     BaseFigure._perform_update(
                         plotly_obj[key], val)
                 elif isinstance(validator, CompoundArrayValidator):
-                    BaseFigure._perform_update(
-                        plotly_obj[key], val)
+                    if plotly_obj[key]:
+                        # plotly_obj has an existing non-empty array for key
+                        # In this case we merge val into the existing elements
+                        BaseFigure._perform_update(
+                            plotly_obj[key], val)
+                    else:
+                        # plotly_obj is an empty or uninitialized list for key
+                        # In this case we accept val as is
+                        plotly_obj[key] = val
                 else:
                     # Assign non-compound value
                     plotly_obj[key] = val

plotly/graph_objs/_area.py

@@ -785,7 +785,7 @@ def __init__(
         from _plotly_utils.basevalidators import LiteralValidator
         self._props['type'] = 'area'
         self._validators['type'] = LiteralValidator(
-            plotly_name='type', parent_name='area'
+            plotly_name='type', parent_name='area', val='area'
         )
 
         # Process unknown kwargs

plotly/graph_objs/_bar.py

@@ -2172,7 +2172,7 @@ def __init__(
         from _plotly_utils.basevalidators import LiteralValidator
         self._props['type'] = 'bar'
         self._validators['type'] = LiteralValidator(
-            plotly_name='type', parent_name='bar'
+            plotly_name='type', parent_name='bar', val='bar'
         )
 
         # Process unknown kwargs

plotly/graph_objs/_box.py

@@ -1574,7 +1574,7 @@ def __init__(
         from _plotly_utils.basevalidators import LiteralValidator
         self._props['type'] = 'box'
         self._validators['type'] = LiteralValidator(
-            plotly_name='type', parent_name='box'
+            plotly_name='type', parent_name='box', val='box'
         )
 
         # Process unknown kwargs

plotly/graph_objs/_candlestick.py

@@ -1234,7 +1234,7 @@ def __init__(
         from _plotly_utils.basevalidators import LiteralValidator
         self._props['type'] = 'candlestick'
         self._validators['type'] = LiteralValidator(
-            plotly_name='type', parent_name='candlestick'
+            plotly_name='type', parent_name='candlestick', val='candlestick'
         )
 
         # Process unknown kwargs

plotly/graph_objs/_carpet.py

@@ -1712,7 +1712,7 @@ def __init__(
         from _plotly_utils.basevalidators import LiteralValidator
         self._props['type'] = 'carpet'
         self._validators['type'] = LiteralValidator(
-            plotly_name='type', parent_name='carpet'
+            plotly_name='type', parent_name='carpet', val='carpet'
         )
 
         # Process unknown kwargs

plotly/graph_objs/_choropleth.py

@@ -1478,7 +1478,7 @@ def __init__(
         from _plotly_utils.basevalidators import LiteralValidator
         self._props['type'] = 'choropleth'
         self._validators['type'] = LiteralValidator(
-            plotly_name='type', parent_name='choropleth'
+            plotly_name='type', parent_name='choropleth', val='choropleth'
         )
 
         # Process unknown kwargs

plotly/graph_objs/_cone.py

@@ -1798,7 +1798,7 @@ def __init__(
         from _plotly_utils.basevalidators import LiteralValidator
         self._props['type'] = 'cone'
         self._validators['type'] = LiteralValidator(
-            plotly_name='type', parent_name='cone'
+            plotly_name='type', parent_name='cone', val='cone'
         )
 
         # Process unknown kwargs

plotly/graph_objs/_contour.py

@@ -2080,7 +2080,7 @@ def __init__(
         from _plotly_utils.basevalidators import LiteralValidator
         self._props['type'] = 'contour'
         self._validators['type'] = LiteralValidator(
-            plotly_name='type', parent_name='contour'
+            plotly_name='type', parent_name='contour', val='contour'
         )
 
         # Process unknown kwargs

plotly/graph_objs/_contourcarpet.py

@@ -1979,7 +1979,9 @@ def __init__(
         from _plotly_utils.basevalidators import LiteralValidator
         self._props['type'] = 'contourcarpet'
         self._validators['type'] = LiteralValidator(
-            plotly_name='type', parent_name='contourcarpet'
+            plotly_name='type',
+            parent_name='contourcarpet',
+            val='contourcarpet'
         )
 
         # Process unknown kwargs

plotly/graph_objs/_heatmap.py

@@ -1877,7 +1877,7 @@ def __init__(
         from _plotly_utils.basevalidators import LiteralValidator
         self._props['type'] = 'heatmap'
         self._validators['type'] = LiteralValidator(
-            plotly_name='type', parent_name='heatmap'
+            plotly_name='type', parent_name='heatmap', val='heatmap'
         )
 
         # Process unknown kwargs

plotly/graph_objs/_heatmapgl.py

@@ -1646,7 +1646,7 @@ def __init__(
         from _plotly_utils.basevalidators import LiteralValidator
         self._props['type'] = 'heatmapgl'
         self._validators['type'] = LiteralValidator(
-            plotly_name='type', parent_name='heatmapgl'
+            plotly_name='type', parent_name='heatmapgl', val='heatmapgl'
         )
 
         # Process unknown kwargs

plotly/graph_objs/_histogram.py

@@ -1754,7 +1754,7 @@ def __init__(
         from _plotly_utils.basevalidators import LiteralValidator
         self._props['type'] = 'histogram'
         self._validators['type'] = LiteralValidator(
-            plotly_name='type', parent_name='histogram'
+            plotly_name='type', parent_name='histogram', val='histogram'
         )
 
         # Process unknown kwargs

plotly/graph_objs/_histogram2d.py

@@ -1941,7 +1941,7 @@ def __init__(
         from _plotly_utils.basevalidators import LiteralValidator
         self._props['type'] = 'histogram2d'
         self._validators['type'] = LiteralValidator(
-            plotly_name='type', parent_name='histogram2d'
+            plotly_name='type', parent_name='histogram2d', val='histogram2d'
         )
 
         # Process unknown kwargs

plotly/graph_objs/_histogram2dcontour.py

@@ -2090,7 +2090,9 @@ def __init__(
         from _plotly_utils.basevalidators import LiteralValidator
         self._props['type'] = 'histogram2dcontour'
         self._validators['type'] = LiteralValidator(
-            plotly_name='type', parent_name='histogram2dcontour'
+            plotly_name='type',
+            parent_name='histogram2dcontour',
+            val='histogram2dcontour'
         )
 
         # Process unknown kwargs

plotly/graph_objs/_mesh3d.py

@@ -2268,7 +2268,7 @@ def __init__(
         from _plotly_utils.basevalidators import LiteralValidator
         self._props['type'] = 'mesh3d'
         self._validators['type'] = LiteralValidator(
-            plotly_name='type', parent_name='mesh3d'
+            plotly_name='type', parent_name='mesh3d', val='mesh3d'
         )
 
         # Process unknown kwargs

plotly/graph_objs/_ohlc.py

@@ -1225,7 +1225,7 @@ def __init__(
         from _plotly_utils.basevalidators import LiteralValidator
         self._props['type'] = 'ohlc'
         self._validators['type'] = LiteralValidator(
-            plotly_name='type', parent_name='ohlc'
+            plotly_name='type', parent_name='ohlc', val='ohlc'
         )
 
         # Process unknown kwargs

plotly/graph_objs/_parcoords.py

@@ -1026,7 +1026,7 @@ def __init__(
         from _plotly_utils.basevalidators import LiteralValidator
         self._props['type'] = 'parcoords'
         self._validators['type'] = LiteralValidator(
-            plotly_name='type', parent_name='parcoords'
+            plotly_name='type', parent_name='parcoords', val='parcoords'
         )
 
         # Process unknown kwargs

plotly/graph_objs/_pie.py

@@ -1505,7 +1505,7 @@ def __init__(
         from _plotly_utils.basevalidators import LiteralValidator
         self._props['type'] = 'pie'
         self._validators['type'] = LiteralValidator(
-            plotly_name='type', parent_name='pie'
+            plotly_name='type', parent_name='pie', val='pie'
         )
 
         # Process unknown kwargs

plotly/graph_objs/_pointcloud.py

@@ -1207,7 +1207,7 @@ def __init__(
         from _plotly_utils.basevalidators import LiteralValidator
         self._props['type'] = 'pointcloud'
         self._validators['type'] = LiteralValidator(
-            plotly_name='type', parent_name='pointcloud'
+            plotly_name='type', parent_name='pointcloud', val='pointcloud'
         )
 
         # Process unknown kwargs

plotly/graph_objs/_sankey.py

@@ -981,7 +981,7 @@ def __init__(
         from _plotly_utils.basevalidators import LiteralValidator
         self._props['type'] = 'sankey'
         self._validators['type'] = LiteralValidator(
-            plotly_name='type', parent_name='sankey'
+            plotly_name='type', parent_name='sankey', val='sankey'
         )
 
         # Process unknown kwargs

plotly/graph_objs/_scatter.py

@@ -2145,7 +2145,7 @@ def __init__(
         from _plotly_utils.basevalidators import LiteralValidator
         self._props['type'] = 'scatter'
         self._validators['type'] = LiteralValidator(
-            plotly_name='type', parent_name='scatter'
+            plotly_name='type', parent_name='scatter', val='scatter'
         )
 
         # Process unknown kwargs

plotly/graph_objs/_scatter3d.py

@@ -1925,7 +1925,7 @@ def __init__(
         from _plotly_utils.basevalidators import LiteralValidator
         self._props['type'] = 'scatter3d'
         self._validators['type'] = LiteralValidator(
-            plotly_name='type', parent_name='scatter3d'
+            plotly_name='type', parent_name='scatter3d', val='scatter3d'
         )
 
         # Process unknown kwargs

plotly/graph_objs/_scattercarpet.py

@@ -1580,7 +1580,9 @@ def __init__(
         from _plotly_utils.basevalidators import LiteralValidator
         self._props['type'] = 'scattercarpet'
         self._validators['type'] = LiteralValidator(
-            plotly_name='type', parent_name='scattercarpet'
+            plotly_name='type',
+            parent_name='scattercarpet',
+            val='scattercarpet'
         )
 
         # Process unknown kwargs

plotly/graph_objs/_scattergeo.py

@@ -1589,7 +1589,7 @@ def __init__(
         from _plotly_utils.basevalidators import LiteralValidator
         self._props['type'] = 'scattergeo'
         self._validators['type'] = LiteralValidator(
-            plotly_name='type', parent_name='scattergeo'
+            plotly_name='type', parent_name='scattergeo', val='scattergeo'
         )
 
         # Process unknown kwargs

plotly/graph_objs/_scattergl.py

@@ -1942,7 +1942,7 @@ def __init__(
         from _plotly_utils.basevalidators import LiteralValidator
         self._props['type'] = 'scattergl'
         self._validators['type'] = LiteralValidator(
-            plotly_name='type', parent_name='scattergl'
+            plotly_name='type', parent_name='scattergl', val='scattergl'
         )
 
         # Process unknown kwargs

plotly/graph_objs/_scattermapbox.py

@@ -1430,7 +1430,9 @@ def __init__(
         from _plotly_utils.basevalidators import LiteralValidator
         self._props['type'] = 'scattermapbox'
         self._validators['type'] = LiteralValidator(
-            plotly_name='type', parent_name='scattermapbox'
+            plotly_name='type',
+            parent_name='scattermapbox',
+            val='scattermapbox'
         )
 
         # Process unknown kwargs