feat: support highlight in line, bar and histogram plots using maidr ts

Author: dakshpokar

Diff for: example/bar/example_bar_plot.ipynb

@@ -4,7 +4,7 @@
 import maidr
 
 # To use maidr's typescript build set the following command to use maidr ts
-# maidr.set_engine("ts")
+maidr.set_engine("ts")
 
 # Load dataset
 tips = sns.load_dataset("tips")

Diff for: example/bar/matplotlib/example_mpl_bar_plot.py

@@ -3,6 +3,8 @@
 
 import maidr
 
+maidr.set_engine("ts")
+
 # Load the penguins dataset
 penguins = sns.load_dataset("penguins")
 

Diff for: example/bar/seaborn/example_sns_bar_plot.py

@@ -3,7 +3,7 @@
 
 import maidr
 
-# maidr.set_engine("ts")
+maidr.set_engine("ts")
 
 # Load the flights dataset from seaborn
 flights = sns.load_dataset("flights")

Diff for: example/histogram/example_histogram_plot.ipynb

@@ -3,6 +3,8 @@
 
 import maidr
 
+maidr.set_engine("ts")
+
 # Load the 'tips' dataset from seaborn
 tips = sns.load_dataset("tips")
 

Diff for: example/line/example_line_plot.ipynb

@@ -5,7 +5,7 @@ class MaidrKey(str, Enum):
     # Maidr info keys.
     ID = "id"
     ORIENTATION = "orientation"
-    SELECTOR = "selector"
+    SELECTOR = "selectors"
     TYPE = "type"
 
     # Plot data keys.

Diff for: example/line/matplotlib/example_mpl_line.py

@@ -160,10 +160,18 @@ def _flatten_maidr(self) -> dict | list[dict]:
 
         for plot in self._plots:
             schema = plot.schema
+
             if MaidrKey.SELECTOR in schema:
-                schema[MaidrKey.SELECTOR] = schema[MaidrKey.SELECTOR].replace(
-                    "maidr='true'", f"maidr='{self.selector_id}'"
-                )
+                if isinstance(schema[MaidrKey.SELECTOR], str):
+                    schema[MaidrKey.SELECTOR] = schema[MaidrKey.SELECTOR].replace(
+                        "maidr='true'", f"maidr='{self.selector_id}'"
+                    )
+                if isinstance(schema[MaidrKey.SELECTOR], list):
+                    for i in range(len(schema[MaidrKey.SELECTOR])):
+                        schema[MaidrKey.SELECTOR][i] = schema[MaidrKey.SELECTOR][
+                            i
+                        ].replace("maidr='true'", f"maidr='{self.selector_id}'")
+
             plot_schemas.append(
                 {
                     "schema": schema,

Diff for: example/line/seaborn/example_sns_line.py

@@ -42,8 +42,8 @@ class MultiLinePlot(MaidrPlot, LineExtractorMixin):
     def __init__(self, ax: Axes, **kwargs):
         super().__init__(ax, PlotType.LINE)
 
-    def _get_selector(self) -> str:
-        return "g[maidr='true'] > path"
+    def _get_selector(self) -> list[str]:
+        return ["g[maidr='true'] > path"]
 
     def _extract_plot_data(self) -> list[dict]:
         plot = self.extract_lines(self.ax)