Implement TypeMap to enable module specific type map registration

src/magicgui/signature.py

@@ -28,6 +28,7 @@
     from typing_extensions import Unpack
 
     from magicgui.application import AppRef
+    from magicgui.type_map import TypeMap
     from magicgui.widgets import Container, Widget
     from magicgui.widgets.bases._container_widget import ContainerKwargs
 
@@ -188,12 +189,18 @@ def __str__(self) -> str:
             )
         )
 
-    def to_widget(self, app: AppRef | None = None) -> Widget:
+    def to_widget(
+        self,
+        app: AppRef | None = None,
+        type_map: TypeMap | None = None,
+    ) -> Widget:
         """Create and return a widget for this object."""
-        from magicgui.widgets import create_widget
+        from magicgui.type_map import TypeMap
 
         value = Undefined if self.default in (self.empty, TZ_EMPTY) else self.default
-        widget = create_widget(
+        if type_map is None:
+            type_map = TypeMap.global_instance()
+        widget = type_map.create_widget(
             name=self.name,
             value=value,
             annotation=self.annotation,
@@ -286,19 +293,26 @@ def from_signature(
             raise_on_unknown=raise_on_unknown,
         )
 
-    def widgets(self, app: AppRef | None = None) -> MappingProxyType:
+    def widgets(
+        self,
+        app: AppRef | None = None,
+        type_map: TypeMap | None = None,
+    ) -> MappingProxyType:
         """Return mapping from parameters to widgets for all params in Signature."""
         return MappingProxyType(
-            {n: p.to_widget(app) for n, p in self.parameters.items()}
+            {n: p.to_widget(app, type_map=type_map) for n, p in self.parameters.items()}
         )
 
     def to_container(
-        self, app: AppRef | None = None, **kwargs: Unpack[ContainerKwargs]
+        self,
+        app: AppRef | None = None,
+        type_map: TypeMap | None = None,
+        **kwargs: Unpack[ContainerKwargs],
     ) -> Container:
         """Return a ``magicgui.widgets.Container`` for this MagicSignature."""
         from magicgui.widgets import Container
 
-        kwargs["widgets"] = list(self.widgets(app).values())
+        kwargs["widgets"] = list(self.widgets(app, type_map=type_map).values())
         return Container(**kwargs)
 
     def replace(  # type: ignore[override]

src/magicgui/type_map/__init__.py

@@ -1,10 +1,17 @@
 """Functions that map python types to widgets."""
 
-from ._type_map import get_widget_class, register_type, type2callback, type_registered
+from ._type_map import (
+    TypeMap,
+    get_widget_class,
+    register_type,
+    type2callback,
+    type_registered,
+)
 
 __all__ = [
     "get_widget_class",
     "register_type",
     "type_registered",
     "type2callback",
+    "TypeMap",
 ]