fix(chart.py): fix detecting axis type for default axes - i.e. when x…

…axis or yaxis is not defined (#12)
slaclau · Nov 4, 2024 · af9083d · af9083d
1 parent aaf609d
commit af9083d
Showing 1 changed file with 17 additions and 6 deletions.
diff --git a/src/plotly_gtk/chart.py b/src/plotly_gtk/chart.py
@@ -541,28 +541,39 @@ def _update_layout(self):
             ),
         )
         for xaxis in xaxes:
+            if xaxis not in self.layout:
+                self.layout[xaxis] = {}
             if "type" not in self.layout[xaxis]:
                 first_plot_on_axis = [
                     trace
                     for trace in self.data
-                    if trace["xaxis"] == xaxis.replace("axis", "")
+                    if "xaxis" not in trace
+                    or trace["xaxis"] == xaxis.replace("axis", "")
                 ][0]
-                self.layout[xaxis]["_type"] = self._detect_axis_type(
-                    first_plot_on_axis["x"]
+                self.layout[xaxis]["_type"] = (
+                    self._detect_axis_type(first_plot_on_axis["x"])
+                    if "x" in first_plot_on_axis
+                    else "linear"
                 )
             else:
                 self.layout[xaxis]["_type"] = self.layout[xaxis]["type"]
             template[xaxis] = template["xaxis"]
             defaults[xaxis] = defaults["xaxis"]
         for yaxis in yaxes:
+            if yaxis not in self.layout:
+                self.layout[yaxis] = {}
             if "type" not in self.layout[yaxis]:
                 first_plot_on_axis = [
                     trace
                     for trace in self.data
-                    if trace["yaxis"] == yaxis.replace("axis", "")
+                    if "yaxis" not in trace
+                    or trace["yaxis"] == yaxis.replace("axis", "")
                 ][0]
-                self.layout[yaxis]["_type"] = self._detect_axis_type(
-                    first_plot_on_axis["y"]
+                print(first_plot_on_axis)
+                self.layout[yaxis]["_type"] = (
+                    self._detect_axis_type(first_plot_on_axis["y"])
+                    if "y" in first_plot_on_axis
+                    else "linear"
                 )
             else:
                 self.layout[yaxis]["_type"] = self.layout[yaxis]["type"]