|
from typing import cast |
|
|
|
from core.extension.extensible import Extensible, ExtensionModule, ModuleExtension |
|
from core.external_data_tool.base import ExternalDataTool |
|
from core.moderation.base import Moderation |
|
|
|
|
|
class Extension: |
|
__module_extensions: dict[str, dict[str, ModuleExtension]] = {} |
|
|
|
module_classes = {ExtensionModule.MODERATION: Moderation, ExtensionModule.EXTERNAL_DATA_TOOL: ExternalDataTool} |
|
|
|
def init(self): |
|
for module, module_class in self.module_classes.items(): |
|
m = cast(Extensible, module_class) |
|
self.__module_extensions[module.value] = m.scan_extensions() |
|
|
|
def module_extensions(self, module: str) -> list[ModuleExtension]: |
|
module_extensions = self.__module_extensions.get(module) |
|
|
|
if not module_extensions: |
|
raise ValueError(f"Extension Module {module} not found") |
|
|
|
return list(module_extensions.values()) |
|
|
|
def module_extension(self, module: ExtensionModule, extension_name: str) -> ModuleExtension: |
|
module_extensions = self.__module_extensions.get(module.value) |
|
|
|
if not module_extensions: |
|
raise ValueError(f"Extension Module {module} not found") |
|
|
|
module_extension = module_extensions.get(extension_name) |
|
|
|
if not module_extension: |
|
raise ValueError(f"Extension {extension_name} not found") |
|
|
|
return module_extension |
|
|
|
def extension_class(self, module: ExtensionModule, extension_name: str) -> type: |
|
module_extension = self.module_extension(module, extension_name) |
|
t: type = module_extension.extension_class |
|
return t |
|
|
|
def validate_form_schema(self, module: ExtensionModule, extension_name: str, config: dict) -> None: |
|
module_extension = self.module_extension(module, extension_name) |
|
form_schema = module_extension.form_schema |
|
|
|
|
|
|
