Determine the MIME type for the icon path.

glue-viz · Aug 15, 2024 · 93005e7 · 93005e7
1 parent 55f2b35
commit 93005e7
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/glue_jupyter/common/toolbar_vuetify.py b/glue_jupyter/common/toolbar_vuetify.py
@@ -1,3 +1,4 @@
+from mimetypes import guess_type
 import os
 import ipyvuetify as v
 import traitlets
@@ -76,7 +77,11 @@ def add_tool(self, tool):
         else:
             path = icon_path(tool.icon, icon_format=ext)
 
-        format = "svg+xml" if ext == "svg" else ext
+        format = guess_type(path)[0]
+        image_prefix = "image/"
+        if format is None or not format.startswith(image_prefix):
+            raise ValueError(f"Invalid or unknown image MIME type for: {path}")
+        format = format[len(image_prefix):]
         self.tools_data = {
             **self.tools_data,
             tool.tool_id: {