add plugin registry

raven/pluginregistry.py

@@ -0,0 +1,55 @@
+from types import ModuleType
+from typing import Dict
+from inspect import isclass
+from pkgutil import iter_modules
+from pathlib import Path
+from os.path import dirname
+from importlib import import_module
+from raven.pluginbase import PluginBase
+
+
+class PluginRegistry():
+    plugins: Dict[str, PluginBase] = {}
+
+    modules: [ModuleType] = []
+
+    @staticmethod
+    def register_plugin(name: str, plugin: PluginBase):
+        PluginRegistry.plugins[name] = plugin
+
+    @staticmethod
+    def get_plugins_by_function(function_name: str) -> [PluginBase]:
+        result = []
+        for plugin in PluginRegistry.plugins.values():
+            fun = getattr(plugin, function_name, None)
+            if callable(fun):
+                result.append(plugin)
+        return result
+
+    @staticmethod
+    def load_module(module_name: str) -> ModuleType:
+        module_name = f"plugins.{module_name}"
+        # import the module and iterate through its attributes
+        module = import_module(module_name)
+        PluginRegistry.modules.append(module)
+        return module
+
+    @staticmethod
+    def load_plugin(plugin_name: str):
+        module_name = f"plugins.{plugin_name.lower()}"
+        module = import_module(module_name)
+        if plugin_name in dir(module):
+            plugin_class = getattr(module, plugin_name)
+
+            if isclass(plugin_class) and issubclass(plugin_class, PluginBase):
+                PluginRegistry.register_plugin(plugin_name, plugin_class)
+                print("%s -> %s :: %s in %s" % (plugin_name, plugin_class, module_name, module))
+                return plugin_class
+
+    @staticmethod
+    def get_modules() -> [ModuleType]:
+        return PluginRegistry.modules.copy()
+
+    @staticmethod
+    def get_plugins() -> [ModuleType]:
+        return PluginRegistry.modules.copy()