feat: Create a function that can enumerate Oracle classes

gm_assistant/oracle/__init__.py

@@ -2,3 +2,29 @@
 # SPDX-FileContributor: Gergely Polonkai
 #
 # SPDX-License-Identifier: GPL-3.0-or-later
+"""Oracle classes and related functions"""
+
+from typing import Any, Type
+
+from .base import Oracle
+from .object_generator import ObjectGeneratorOracle
+from .random_choice import RandomChoiceOracle
+
+
+def generate_type_classes(class_list: dict[str, Any]) -> dict[str, Type[Oracle]]:
+    """Generate a dictionary of oracle type handlers"""
+
+    ret: dict[str, Type[Oracle]] = {}
+
+    for klass in class_list.values():
+        if not isinstance(klass, type) or klass == Oracle or not issubclass(klass, Oracle):
+            continue
+
+        if klass.TYPE_MARKER in ret:
+            raise KeyError(
+                f"{ret[klass.TYPE_MARKER].__name__} is already registered as a handler for {klass.TYPE_MARKER}"
+            )
+
+        ret[klass.TYPE_MARKER] = klass
+
+    return ret