Move HelperHandlerRegistry to helper_registry.py

Author: r41k0u

pythonbpf/helper/__init__.py

@@ -1,4 +1,5 @@
-from .helper_utils import HelperHandlerRegistry, reset_scratch_pool
+from .helper_registry import HelperHandlerRegistry
+from .helper_utils import reset_scratch_pool
 from .bpf_helper_handler import handle_helper_call
 from .helpers import ktime, pid, deref, XDP_DROP, XDP_PASS
 

pythonbpf/helper/bpf_helper_handler.py

@@ -1,8 +1,9 @@
 import ast
 from llvmlite import ir
 from enum import Enum
+
+from .helper_registry import HelperHandlerRegistry
 from .helper_utils import (
-    HelperHandlerRegistry,
     get_or_create_ptr_from_arg,
     get_flags_val,
     handle_fstring_print,

pythonbpf/helper/helper_registry.py

@@ -0,0 +1,27 @@
+from typing import Callable
+
+
+class HelperHandlerRegistry:
+    """Registry for BPF helpers"""
+
+    _handlers: dict[str, Callable] = {}
+
+    @classmethod
+    def register(cls, helper_name):
+        """Decorator to register a handler function for a helper"""
+
+        def decorator(func):
+            cls._handlers[helper_name] = func
+            return func
+
+        return decorator
+
+    @classmethod
+    def get_handler(cls, helper_name):
+        """Get the handler function for a helper"""
+        return cls._handlers.get(helper_name)
+
+    @classmethod
+    def has_handler(cls, helper_name):
+        """Check if a handler function is registered for a helper"""
+        return helper_name in cls._handlers

pythonbpf/helper/helper_utils.py

@@ -1,6 +1,5 @@
 import ast
 import logging
-from collections.abc import Callable
 
 from llvmlite import ir
 from pythonbpf.expr import (
@@ -13,32 +12,6 @@
 logger = logging.getLogger(__name__)
 
 
-class HelperHandlerRegistry:
-    """Registry for BPF helpers"""
-
-    _handlers: dict[str, Callable] = {}
-
-    @classmethod
-    def register(cls, helper_name):
-        """Decorator to register a handler function for a helper"""
-
-        def decorator(func):
-            cls._handlers[helper_name] = func
-            return func
-
-        return decorator
-
-    @classmethod
-    def get_handler(cls, helper_name):
-        """Get the handler function for a helper"""
-        return cls._handlers.get(helper_name)
-
-    @classmethod
-    def has_handler(cls, helper_name):
-        """Check if a handler function is registered for a helper"""
-        return helper_name in cls._handlers
-
-
 class ScratchPoolManager:
     """Manage the temporary helper variables in local_sym_tab"""