python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > python 通用插件类

python实现一个通用的插件类

作者:会编程的大白熊

插件管理器用于注册、销毁、执行插件,本文主要介绍了python实现一个通用的插件类,文中通过示例代码介绍的非常详细,需要的朋友们下面随着小编来一起学习学习吧

本文提供了一种插件类的实现方案。

定义插件管理器

插件管理器用于注册、销毁、执行插件。

import abc
from functools import wraps
from typing import Callable, Dict

from pydantic import (
    BaseModel,
    validate_arguments,
    ValidationError as PydanticValidationError,
)

def import_string(dotted_path: str) -> Callable:
    """Import a dotted module path and return the attribute/class designated by the
    last name in the path. Raise ImportError if the import failed.

    Args:
        dotted_path: 字符串表示的模块类,module.class
    Returns:
        返回加载的模块中的对象
    """
    try:
        module_path, class_name = dotted_path.rsplit(".", 1)
    except ValueError:
        raise ImportError("{} doesn't look like a module path".format(dotted_path))

    module: ModuleType = import_module(module_path)

    try:
        # 返回模块中的类
        return getattr(module, class_name)
    except AttributeError:
        raise ImportError(
            'Module "{}" does not define a "{}" attribute/class'.format(
                module_path, class_name
            )
        )


class FunctionsManager:
    """函数管理器 ."""
    # 存放注册的可执行对象
    __hub = {}  # type: ignore

    @classmethod
    def register_invocation_cls(cls, invocation_cls: InvocationMeta, name=None) -> None:
        if not name:
            func_name = invocation_cls.Meta.func_name
        else:
            func_name = name
        if not isinstance(func_name, str):
            raise ValueError(f"func_name {func_name} should be string")
        existed_invocation_cls = cls.__hub.get(func_name)
        if existed_invocation_cls:
            raise RuntimeError(
                "func register error, {}'s func_name {} conflict with {}".format(
                    existed_invocation_cls, func_name, invocation_cls
                )
            )

        # 存放类的实例
        cls.__hub[func_name] = invocation_cls()

    @classmethod
    def register_funcs(cls, func_dict) -> None:
        for func_name, func_obj in func_dict.items():
            if not isinstance(func_name, str):
                raise ValueError(f"func_name {func_name} should be string")
            if func_name in cls.__hub:
                raise ValueError(
                    "func register error, {}'s func_name {} conflict with {}".format(
                        func_obj, func_name, cls.__hub[func_name]
                    )
                )
            if isinstance(func_obj, str):
                func = import_string(func_obj)
            elif isinstance(func_obj, Callable):
                func = func_obj
            else:
                raise ValueError(
                    "func register error, {} is not be callable".format(
                        func_obj, func_name
                    )
                )

            cls.__hub[func_name] = func

    @classmethod
    def clear(cls) -> None:
        """清空注册信息 ."""
        cls.__hub = {}

    @classmethod
    def all_funcs(cls) -> Dict:
        """获得所有的注册信息. """
        return cls.__hub

    @classmethod
    def get_func(cls, func_name: str) -> Callable:
        """获得注册的函数 ."""
        func_obj = cls.__hub.get(func_name)
        if not func_obj:
            raise ValueError("func object {} not found".format(func_name))
        return func_obj

    @classmethod
    def func_call(cls, func_name: str, *args, **kwargs):
        """根据函数名执行注册的函数 ."""
        func = cls.get_func(func_name)
        return func(*args, **kwargs)

定义元类

派生的类可自行注册到插件管理器。

class InvocationMeta(type):
    """
    Metaclass for function invocation
    """

    def __new__(cls, name, bases, dct):
        # ensure initialization is only performed for subclasses of Plugin
        parents = [b for b in bases if isinstance(b, InvocationMeta)]
        if not parents:
            return super().__new__(cls, name, bases, dct)

        new_cls = super().__new__(cls, name, bases, dct)

        # meta validation
        meta_obj = getattr(new_cls, "Meta", None)
        if not meta_obj:
            raise AttributeError("Meta class is required")

        func_name = getattr(meta_obj, "func_name", None)
        if not func_name:
            raise AttributeError("func_name is required in Meta")

        desc = getattr(meta_obj, "desc", None)
        if desc is not None and not isinstance(desc, str):
            raise AttributeError("desc in Meta should be str")

        # register func
        FunctionsManager.register_invocation_cls(new_cls)

        return new_cls

定义元类的一个抽象派生类

支持参数验证。

class BaseInvocation(metaclass=InvocationMeta):
    """
    Base class for function invocation
    """

    class Inputs(BaseModel):
        """
        输入校验器
        """

        pass

    @validate_arguments  # type: ignore
    def __call__(self, *args, **kwargs):

        # 输入参数校验, 仅可能是 args 或 kwargs 之一
        try:
            params = {}
            if args:
                inputs_meta = getattr(self.Inputs, "Meta", None)
                inputs_ordering = getattr(inputs_meta, "ordering", None)
                if isinstance(inputs_ordering, list):
                    if len(args) > len(inputs_ordering):
                        raise Exception(f"Too many arguments for inputs: {args}")
                    params = dict(zip(inputs_ordering, args))
            elif kwargs:
                params = kwargs

            # 参数校验
            if params:
                self.Inputs(**params)
        except PydanticValidationError as e:
            raise Exception(e)

        # 执行自定义业务逻辑
        return self.invoke(*args, **kwargs)

    @abc.abstractmethod
    def invoke(self, *args, **kwargs):
        """自定义业务逻辑 ."""
        raise NotImplementedError()

定义装饰器

def register_class(name: str):
    def _register_class(cls: BaseInvocation):
        FunctionsManager.register_invocation_cls(cls, name=name)

        @wraps(cls)
        def wrapper():
            return cls()

        return wrapper

    return _register_class


def register_func(name: str):
    def _register_func(func: Callable):
        FunctionsManager.register_funcs({name: func})

        @wraps(func)
        def wrapper(*args, **kwargs):
            return func(*args, **kwargs)

        return wrapper

    return _register_func

单元测试

from pydantic import BaseModel

from .register import FunctionsManager, register_func, register_class, BaseInvocation


@register_func("add")
def add(x: int, y: int) -> int:
    return x + y


class Add(BaseInvocation):
    class Meta:
        func_name = "multiply"

    class Inputs(BaseModel):
        """
        输入校验器
        """
        x: int
        y: int

        class Meta:
            ordering = ["x", "y"]

    def invoke(self, x: int, y: int) -> int:
        return x * y


@register_class("subtract")
class Subtract:
    class Inputs(BaseModel):
        """
        输入校验器
        """
        x: int
        y: int

        class Meta:
            ordering = ["x", "y"]

    def __call__(self, x: int, y: int) -> int:
        return x - y


class TestFunctionsManager:
    def test_register_func(self):
        func = FunctionsManager.get_func("add")
        assert func(2, 3) == 5

    def test_register_class(self):
        func = FunctionsManager.get_func("subtract")
        assert func(2, 3) == -1

    def test_metaclass(self):
        func = FunctionsManager.get_func("multiply")
        assert func(2, 3) == 6

参考

https://github.com/TencentBlueKing/bkflow-feel/blob/main/bkflow_feel/utils.py

到此这篇关于python实现一个通用的插件类的文章就介绍到这了,更多相关python 通用插件类内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
阅读全文