Python 钩子函数详解

###### 钩子函数

```

import pluggy

hookspec = pluggy.HookspecMarker('aaa')

hookimpl = pluggy.HookimplMarker('aaa')

class MySpec():

@hookspec() # firstresult=True设置之后，等到第一个返回非空结果的hookimpl，就返回（hookwrapper还是正常执行）

def myhook(self, arg_one, arg_two):

pass

class MyPlugin():

@hookimpl() # tryfirst=True,trylast=True定义执行顺序

def myhook(self, arg_one):

print('arg_one')

return arg_one

class MyPluginOther():

@hookimpl()

def myhook(self, arg_one, arg_two):

print('arg_one, arg_two')

return arg_one, arg_two

class MyPluginWrapper():

'''

hookwrapper作为每个钩子函数的上下文运行

在所有hook之前先运行

'''

@hookimpl(hookwrapper=True)

def myhook(self):

print('-------- hool wrapper before -----------')

outcome = yield # 生成器会给outcome send一个pluggy.callers._Result对象

print('-------- hook wrapper after -----------')

res = outcome.get_result() # 通过get_result拿到所有钩子函数的结果

print('get all hook return ', res)

outcome.force_result(1) # 强制更改钩子函数返回结果

print('-------- hook wrapper end-----------')

test1 = pluggy.PluginManager('aaa')

test1.add_hookspecs(MySpec)

test1.register(MyPlugin())

test1.register(MyPluginOther()) # 后注册的先执行，收集进结果

test1.register(MyPluginWrapper())

result = test1.hook.myhook(arg_one=1, arg_two=2) # hookimpl调用，必须用关键字语法

print(result)

------------------

输出结果：

-------- hool wrapper before -----------

arg_one, arg_two

arg_one

-------- hook wrapper after -----------

get all hook return [(1, 2), 1]

-------- hook wrapper end-----------

1

```