Extract processor registry into new module

Moves processor registration and discovery helpers out of `dlc_processor_socket.py` into a new `registry.py` module so registry access no longer depends on importing socket logic. `dlc_processor_socket.py` now imports the shared registry helpers and adds a safe fallback when `dlclive` is unavailable, reducing import-time failures in environments without that dependency. Package exports were updated to expose registry APIs from the new module.

Author: C-Achard

dlclivegui/processors/__init__.py

@@ -1,3 +1,3 @@
-from .dlc_processor_socket import PROCESSOR_REGISTRY, BaseProcessorSocket, register_processor
+from .registry import PROCESSOR_REGISTRY, register_processor
 
-__all__ = ["register_processor", "BaseProcessorSocket", "PROCESSOR_REGISTRY"]
+__all__ = ["register_processor", "PROCESSOR_REGISTRY"]

dlclivegui/processors/dlc_processor_socket.py

@@ -13,7 +13,11 @@
 
 import numpy as np
 import pandas as pd
-from dlclive.processor import Processor  # type: ignore
+
+try:
+    from dlclive.processor import Processor  # type: ignore
+except ImportError:
+    Processor = object  # Fallback for type checking if dlclive is not installed
 
 logger = logging.getLogger("dlc_processor_socket")
 
@@ -23,21 +27,6 @@
     _handler.setFormatter(logging.Formatter("%(asctime)s [%(levelname)s] %(message)s"))
     logger.addHandler(_handler)
 
-# Registry for GUI discovery
-PROCESSOR_REGISTRY = {}
-
-
-def register_processor(cls):
-    registry_key = getattr(cls, "PROCESSOR_ID", cls.__name__)
-    if registry_key in PROCESSOR_REGISTRY:
-        msg = (
-            f"Duplicate processor registration key '{registry_key}': "
-            f"{PROCESSOR_REGISTRY[registry_key].__name__} vs {cls.__name__}"
-        )
-        logger.warning(msg)
-    PROCESSOR_REGISTRY[registry_key] = cls
-    return cls
-
 
 # pragma: cover
 class BaseProcessorSocket(Processor):
@@ -435,38 +424,3 @@ def get_data(self):
         if self.dlc_cfg is not None:
             save_dict["dlc_cfg"] = self.dlc_cfg
         return save_dict
-
-
-def get_available_processors():
-    """
-    Get list of available processor classes.
-
-    Returns:
-        dict: Dictionary mapping registry keys to processor info.
-    """
-    return {
-        name: {
-            "class": cls,
-            "name": getattr(cls, "PROCESSOR_NAME", name),
-            "description": getattr(cls, "PROCESSOR_DESCRIPTION", ""),
-            "params": getattr(cls, "PROCESSOR_PARAMS", {}),
-        }
-        for name, cls in PROCESSOR_REGISTRY.items()
-    }
-
-
-def instantiate_processor(class_name, **kwargs):
-    """
-    Instantiate a processor by class name with given parameters.
-
-    Args:
-        class_name: Registry key (e.g., "MyProcessorSocket")
-        **kwargs: Constructor kwargs
-
-    Raises:
-        ValueError: If class_name is not in registry
-    """
-    if class_name not in PROCESSOR_REGISTRY:
-        available = ", ".join(PROCESSOR_REGISTRY.keys())
-        raise ValueError(f"Unknown processor '{class_name}'. Available: {available}")
-    return PROCESSOR_REGISTRY[class_name](**kwargs)

dlclivegui/processors/examples.py

@@ -6,7 +6,8 @@
 
 import numpy as np
 
-from dlclivegui.processors import BaseProcessorSocket, register_processor
+from dlclivegui.processors import register_processor
+from dlclivegui.processors.dlc_processor_socket import BaseProcessorSocket
 
 logger = logging.getLogger(__name__)
 

dlclivegui/processors/registry.py

@@ -0,0 +1,53 @@
+import logging
+
+logger = logging.getLogger(__name__)
+
+# Registry for GUI discovery
+PROCESSOR_REGISTRY = {}
+
+
+def register_processor(cls):
+    registry_key = getattr(cls, "PROCESSOR_ID", cls.__name__)
+    if registry_key in PROCESSOR_REGISTRY:
+        msg = (
+            f"Duplicate processor registration key '{registry_key}': "
+            f"{PROCESSOR_REGISTRY[registry_key].__name__} vs {cls.__name__}"
+        )
+        logger.warning(msg)
+    PROCESSOR_REGISTRY[registry_key] = cls
+    return cls
+
+
+def get_available_processors():
+    """
+    Get list of available processor classes.
+
+    Returns:
+        dict: Dictionary mapping registry keys to processor info.
+    """
+    return {
+        name: {
+            "class": cls,
+            "name": getattr(cls, "PROCESSOR_NAME", name),
+            "description": getattr(cls, "PROCESSOR_DESCRIPTION", ""),
+            "params": getattr(cls, "PROCESSOR_PARAMS", {}),
+        }
+        for name, cls in PROCESSOR_REGISTRY.items()
+    }
+
+
+def instantiate_processor(class_name, **kwargs):
+    """
+    Instantiate a processor by class name with given parameters.
+
+    Args:
+        class_name: Registry key (e.g., "MyProcessorSocket")
+        **kwargs: Constructor kwargs
+
+    Raises:
+        ValueError: If class_name is not in registry
+    """
+    if class_name not in PROCESSOR_REGISTRY:
+        available = ", ".join(PROCESSOR_REGISTRY.keys())
+        raise ValueError(f"Unknown processor '{class_name}'. Available: {available}")
+    return PROCESSOR_REGISTRY[class_name](**kwargs)