Python pluggy 库

前言

什么是 pluggy

pluggy 是 pytest 团队开发的库，
用来解决给 pytest 写扩展插件的。
pluggy 提供了方便的 API 可以很轻松的扩展框架的逻辑。

Github仓库
 文档地址

pluggy 的调度方式

draw.io链接

pluggy 有三大要素

HookspecMarker(hookspec) 定义
HookimplMarker(hookimpl) 实现
PluginManager(pm) 插件管理器

spec 定义了可以暴露的方法，可以通过 pm.hook 去调用对应的方法
调用的时候会逐个执行 impl 注册的插件，按 LIFO 后进先出的方式进行。

pluggy 案例代码

import pluggy

hookspec = pluggy.HookspecMarker("myproject")
hookimpl = pluggy.HookimplMarker("myproject")


class MySpec:
    """A hook specification namespace."""

    @hookspec
    def myhook(self, arg1, arg2):
        """My special little hook that you can customize."""


class Plugin_1:
    """A hook implementation namespace."""

    @hookimpl
    def myhook(self, arg1, arg2):
        print("inside Plugin_1.myhook()")
        return arg1 + arg2


class Plugin_2:
    """A 2nd hook implementation namespace."""

    @hookimpl
    def myhook(self, arg1, arg2):
        print("inside Plugin_2.myhook()")
        return arg1 - arg2


# create a manager and add the spec
pm = pluggy.PluginManager("myproject")
pm.add_hookspecs(MySpec)
# register plugins
pm.register(Plugin_1())
pm.register(Plugin_2())
# call our `myhook` hook
results = pm.hook.myhook(arg1=1, arg2=2)
print(results)