什么是hook?
钩子函数(hook function),可以理解是一个挂钩,作用是有需要的时候挂一个东西上去。具体的解释是:钩子函数是把我们自己实现的hook函数在某一时刻挂接到目标挂载点上。
hook应用场景(一)
相信你对钩子函数并不陌生。我在 requests 和 mitmproxy 都有看到类似的设计。
requests 使用hook
例如 requests 中需要打印状态码:
打印状态码,这个动作,我们可以封装到一个函数里,然后作为钩子函数传给requests 使用。
代码说明:
把打印状态码封装到一个status_code() 函数中,在requests.get() 方法中通过hooks 参数接收钩子函数status_code()。
运行结果:
> python requests_hooks.pyhook status doce: 200
status_code() 作为一个函数,可以做的事情很多,比如,进一步判断状态码,打印响应的数据,甚至对相应的数据做加解密等处理。
mitmproxy 中的hook
mitmproxy是一个代理工具,我们这之前的文章也有做过介绍。在抓包的过程中,同样需要用到 hooks 去对request请求或response响应做一些额外的处理。
运行mitmproxy
> mitmproxy -s anatomy.py
自己实现hook
什么情况下需要实现hook,就是一个功能(类/方法)自身无法满足所有需求,那么可以通过hook 就提供扩展自身能力的可能。
实现hook 并不难,看例子:
代码说明:
在上面的例子中Programmer类实现三个功能:eat、code、sleep,但程序员也是普通人,不能每天都只吃饭、编码、睡觉,于是通过register_hook() 提供了做别的事情的能力。
那么,看看Tom、Jerry、Spike三位主角,今天都干了什么吧!
运行结果:
Tom - 2022-12-01: eat.Tom - 2022-12-01: code.Tom - 2022-12-01: sleep.Tom - 2022-12-01: play game.Jerry - 2022-12-01: eat.Jerry - 2022-12-01: code.Jerry - 2022-12-01: sleep.Jerry - 2022-12-01: shopping.Spike - 2022-12-01: eat.Spike - 2022-12-01: code.Spike - 2022-12-01: sleep.
hook应用场景(二)
如果把hook理解为:定义一个函数,然后作为参数塞到另一个类/方法里。 显然,这只是一种用法。我重新想了一下。httpRunner 的 debugtalk.py 文件; pytest 的 conftest.py 文件,他们本身也是拥有特殊名字的 hook文件。程序在执行的过程中,调用这些文件中的钩子函数完成一些特殊的任务。
以pytest为例子
conftest.py
test_sample.py
两个文件看似没有直接的调用关系,在执行 test_sample.py 文件时,可以间接的调用 conftest.py 文件中的baidu_url()钩子函数。
执行测试
> pytest -q test_sample.py
实现动态调用hook
接下来,我们来试试做个类似的功能出来。
run_conf.py
与 conftest.py 文件类似,在这个文件中实现钩子函数。
loader.py
这个东西就比较复杂了,他的作用就是丢给他一个的函数名, 他能通过*_conf.py文件中查找对应的函数名,并将函数执行结果返回。
loader() 函数是一个通用的东西,你可以把他放到任何位置使用。
run.py
通过loader() 函数执行baidu_url 钩子函数,并拿到 url。
注意,我们不需要像传统的方式一样from run_conf import baidu_url 导入模块,只要知道钩子函数的名字即可。
这里的实现并没有 pytest 那么优雅,但也比较接近了。
到此这篇关于python强大的hook函数使用及应用场景的文章就介绍到这了,更多相关python hook函数使用内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!
您可能感兴趣的文章:5 分钟读懂Python 中的 Hook 钩子函数python hook监听事件详解python学习之hook钩子的原理和使用详解Python开发中如何使用Hook技巧Python中使用PyHook监听鼠标和键盘事件实例python中使用pyhook实现键盘监控的例子python利用hook技术破解https的实例代码
python强大的hook函数使用及应用场景
发布日期:2025-01-03 18:36 点击次数:93
上一篇:没有了
下一篇:骁龙8 联发科新品今天亮相!
下一篇:骁龙8 联发科新品今天亮相!