Allow using non-SVG images in toolbar.

Carifio24 · Aug 15, 2024 · 55f2b35 · 55f2b35
1 parent 8434cb4
commit 55f2b35
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/glue_jupyter/common/toolbar_vuetify.py b/glue_jupyter/common/toolbar_vuetify.py
@@ -70,14 +70,17 @@ def _on_change_active_tool(self, change):
     def add_tool(self, tool):
         self.tools[tool.tool_id] = tool
         # TODO: we should ideally just incorporate this check into icon_path directly.
+        ext = os.path.splitext(tool.icon)[1][1:] or "svg"
         if os.path.exists(tool.icon):
             path = tool.icon
         else:
-            path = icon_path(tool.icon, icon_format='svg')
+            path = icon_path(tool.icon, icon_format=ext)
+
+        format = "svg+xml" if ext == "svg" else ext
         self.tools_data = {
             **self.tools_data,
             tool.tool_id: {
                 'tooltip': tool.tool_tip,
-                'img': read_icon(path, 'svg+xml')
+                'img': read_icon(path, format)
             }
         }