【pytest框架源码分析三】pluggy源码分析之hook注册调用流程

embedded/2025/3/5 4:25:30/

pluggy的hook调用，最重要的就是使用了__call__魔法函数，这个函数能够在我们调用实例时，自动调用这个函数，无需自己手动调用。
前面介绍了各个类的方法，这里简述下pluggy的调用流程（主要介绍主流程，分支处理忽略）。
前面我们介绍了HookspecMarker和HookimplMarker这两个类，这两个类是用来作装饰器用的，spec相当于接口类，impl为实现类。我们这边还用之前的例子，来介绍下插件的注册调用流程。

import pluggyspec_test = pluggy.HookspecMarker('test11')
impl_test = pluggy.HookimplMarker('test11')class Spec:@spec_testdef pluggy_test(self, arg1):print(f'this is spec test and arg is {arg1}')passclass Impl1:@impl_testdef pluggy_test(self, arg1):print(f'this is test1 and arg is {arg1}')return arg1class Impl2:@impl_testdef pluggy_test(self, arg1):print(f'this is test2 and arg is {arg1}')return -1 * arg1pm = pluggy.PluginManager('test11')
pm.add_hookspecs(Spec)
pm.register(Impl1())
pm.register(Impl2())
res = pm.hook.pluggy_test(arg1=1)

pluggy.PluginManager(‘test11’)调用后实例化一个pm，注册执行流程都依靠pm管理。
add_hookspecs(Spec)在pm中增加一个spec。
在这里插入图片描述
这个方法中最重要的就是实例化了一个HookCaller的对象，并且把它放到self.hook中。
然后调用pm.register(Impl1())注册插件

注册方法中比较重要的就是实例化了一个hookimpl对象，并且把它添加到hook中，具体看就是添加到_hookimpls的list中（整体在我们上面添加的HookCaller下）。
在这里插入图片描述
接下继续注册第二个插件Impl2，注意hook添加spec时，是按照方法名添加的。impl1和impl2中两个方法名是一样的，所以不会添加新的name来存放impl2下的pluggy_test。Impl2的pluggy_test也和impl1的pluggy_test注册在同一个HookCaller下。
最后调用 pm.hook.pluggy_test(arg1=1)来执行插件。这里的pluggy_test就是HookCaller对象（add_hookspecs步骤）。因为HookCaller类中有__call__函数，所以调用该对象时，会自动调用该方法。
在这里插入图片描述
执行方法就是最下面这行。_hookexec方法即是我们初始化HookCaller传入的。