Add colorbar options to coloraxis.colorbar instead of to individual figures

Author: joelostblom

Diff for: packages/python/plotly/templategen/utils/__init__.py

@@ -29,18 +29,6 @@
 ]
 
 
-def set_all_colorbars(template, colorbar):
-    for parent_path in colorscale_parent_paths:
-        if not template.data[parent_path[0]]:
-            template.data[parent_path[0]] = [{}]
-
-        for trace in template.data[parent_path[0]]:
-            parent = trace[parent_path[1:]]
-
-            if "colorbar" in parent:
-                parent.colorbar = colorbar
-
-
 def initialize_template(
     annotation_defaults,
     axis_common,
@@ -82,10 +70,11 @@ def initialize_template(
     template.layout.plot_bgcolor = panel_background_clr
     template.layout.polar.bgcolor = panel_background_clr
     template.layout.ternary.bgcolor = panel_background_clr
-    set_all_colorbars(template, colorbar_common)
+
     cartesian_axis = dict(axis_common, zerolinecolor=zerolinecolor_clr)
 
     # Colorscales
+    template.layout.coloraxis.colorbar = colorbar_common
     template.layout.colorscale.sequential = colorscale
     if colorscale_minus is not None:
         template.layout.colorscale.sequentialminus = colorscale_minus
@@ -95,13 +84,6 @@ def initialize_template(
     if colorscale_diverging is not None:
         template.layout.colorscale.diverging = colorscale_diverging
 
-    template.data.heatmap[0].colorscale = colorscale
-    template.data.histogram2d[0].colorscale = colorscale
-    template.data.histogram2dcontour[0].colorscale = colorscale
-    template.data.contour[0].colorscale = colorscale
-    template.data.surface[0].colorscale = colorscale
-    template.data.heatmapgl[0].colorscale = colorscale
-
     # Cartesian
     template.layout.xaxis = cartesian_axis
     template.layout.yaxis = cartesian_axis