python強(qiáng)大的hook函數(shù)使用及應(yīng)用場(chǎng)景
什么是hook?
鉤子函數(shù)(hook function),可以理解是一個(gè)掛鉤,作用是有需要的時(shí)候掛一個(gè)東西上去。具體的解釋是:鉤子函數(shù)是把我們自己實(shí)現(xiàn)的hook函數(shù)在某一時(shí)刻掛接到目標(biāo)掛載點(diǎn)上。
hook應(yīng)用場(chǎng)景(一)
相信你對(duì)鉤子函數(shù)并不陌生。我在 requests 和 mitmproxy 都有看到類(lèi)似的設(shè)計(jì)。
requests 使用hook
例如 requests 中需要打印狀態(tài)碼:
# requests_hooks.py import requests r = requests.get("https://httpbin.org/get") print(f"status doce: {r.status_code}")
打印狀態(tài)碼,這個(gè)動(dòng)作,我們可以封裝到一個(gè)函數(shù)里,然后作為鉤子函數(shù)傳給requests 使用。
# requests_hooks.py import requests def status_code(response, *args, **kwargs): print(f"hook status doce: {response.status_code}") r = requests.get("https://httpbin.org/get", hooks={"response": status_code})
代碼說(shuō)明:
把打印狀態(tài)碼封裝到一個(gè)status_code() 函數(shù)中,在requests.get() 方法中通過(guò)hooks 參數(shù)接收鉤子函數(shù)status_code()。
運(yùn)行結(jié)果:
> python requests_hooks.py
hook status doce: 200
status_code()
作為一個(gè)函數(shù),可以做的事情很多,比如,進(jìn)一步判斷狀態(tài)碼,打印響應(yīng)的數(shù)據(jù),甚至對(duì)相應(yīng)的數(shù)據(jù)做加解密等處理。
mitmproxy 中的hook
mitmproxy是一個(gè)代理工具,我們這之前的文章也有做過(guò)介紹。在抓包的過(guò)程中,同樣需要用到 hooks 去對(duì)request請(qǐng)求或response響應(yīng)做一些額外的處理。
# anatomy.py """ Basic skeleton of a mitmproxy addon. Run as follows: mitmproxy -s anatomy.py """ import logging class Counter: def __init__(self): self.num = 0 def request(self, flow): self.num = self.num + 1 logging.info("We've seen %d flows" % self.num) addons = [Counter()]
運(yùn)行mitmproxy
> mitmproxy -s anatomy.py
自己實(shí)現(xiàn)hook
什么情況下需要實(shí)現(xiàn)hook,就是一個(gè)功能(類(lèi)/方法)自身無(wú)法滿(mǎn)足所有需求,那么可以通過(guò)hook 就提供擴(kuò)展自身能力的可能。
實(shí)現(xiàn)hook 并不難,看例子:
import time class Programmer(object): """程序員""" def __init__(self, name, hook=None): self.name = name self.hooks_func = hook self.now_date = time.strftime("%Y-%m-%d") def get_to_eat(self): print(f"{self.name} - {self.now_date}: eat.") def go_to_code(self): print(f"{self.name} - {self.now_date}: code.") def go_to_sleep(self): print(f"{self.name} - {self.now_date}: sleep.") def everyday(self): # 程序員日常三件事 self.get_to_eat() self.go_to_code() self.go_to_sleep() # check the register_hook(hooked or unhooked) # hooked if self.hooks_func is not None: self.hooks_func(self.name) def play_game(name): now_date = time.strftime("%Y-%m-%d") print(f"{name} - {now_date}: play game.") def shopping(name): now_date = time.strftime("%Y-%m-%d") print(f"{name} - {now_date}: shopping.") if __name__ == "__main__": # hook 作為參數(shù)傳入 tom = Programmer("Tom", hook=play_game) jerry = Programmer("Jerry", hook=shopping) spike = Programmer("Spike") # 今日事情 tom.everyday() jerry.everyday() spike.everyday()
代碼說(shuō)明:
在上面的例子中Programmer類(lèi)實(shí)現(xiàn)三個(gè)功能:eat、code、sleep,但程序員也是普通人,不能每天都只吃飯、編碼、睡覺(jué),于是通過(guò)register_hook() 提供了做別的事情的能力。
那么,看看Tom、Jerry、Spike三位主角,今天都干了什么吧!
運(yùn)行結(jié)果:
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應(yīng)用場(chǎng)景(二)
如果把hook理解為:定義一個(gè)函數(shù),然后作為參數(shù)塞到另一個(gè)類(lèi)/方法里。 顯然,這只是一種用法。我重新想了一下。httpRunner 的 debugtalk.py
文件; pytest 的 conftest.py
文件,他們本身也是擁有特殊名字的 hook文件。程序在執(zhí)行的過(guò)程中,調(diào)用這些文件中的鉤子函數(shù)完成一些特殊的任務(wù)。
以pytest為例子
└───project ├───conftest.py └───test_sample.py
- conftest.py
import pytest @pytest.fixture() def baidu_url(): """定義鉤子函數(shù)""" return "https://www.baidu.com"
- test_sample.py
import webbrowser def test_open_url(baidu_url): # 調(diào)用 baidu_url 鉤子函數(shù) # 調(diào)用 瀏覽器 訪(fǎng)問(wèn) baidu_url webbrowser.open_new(baidu_url)
兩個(gè)文件看似沒(méi)有直接的調(diào)用關(guān)系,在執(zhí)行 test_sample.py
文件時(shí),可以間接的調(diào)用 conftest.py
文件中的baidu_url()
鉤子函數(shù)。
執(zhí)行測(cè)試
> pytest -q test_sample.py
實(shí)現(xiàn)動(dòng)態(tài)調(diào)用hook
接下來(lái),我們來(lái)試試做個(gè)類(lèi)似的功能出來(lái)。
└───project ├───run_conf.py ├───loader.py └───run.py
- run_conf.py
def baidu_url(): """定義鉤子函數(shù)""" name = "https://www.baidu.com" return name
與 conftest.py
文件類(lèi)似,在這個(gè)文件中實(shí)現(xiàn)鉤子函數(shù)。
- loader.py
import os import inspect import importlib def loader(name): """ 動(dòng)態(tài)執(zhí)行 hook 函數(shù) """ # 被調(diào)用文件的目錄 stack_t = inspect.stack() ins = inspect.getframeinfo(stack_t[1][0]) file_dir = os.path.dirname(os.path.abspath(ins.filename)) # 被調(diào)用文件目錄下面 *_conf.py 文件 all_hook_files = list(filter(lambda x: x.endswith("_conf.py"), os.listdir(file_dir))) all_hook_module = list(map(lambda x: x.replace(".py", ""), all_hook_files)) # 動(dòng)態(tài)加載 *_config.py hooks = [] for module_name in all_hook_module: hooks.append(importlib.import_module(module_name)) # 根據(jù)傳過(guò)來(lái)的 name 函數(shù)名,從 *_conf.py 文件查找并執(zhí)行。 for per_hook in hooks: # 動(dòng)態(tài)執(zhí)行 process 函數(shù) func = getattr(per_hook, name) return func()
這個(gè)東西就比較復(fù)雜了,他的作用就是丟給他一個(gè)的函數(shù)名, 他能通過(guò)*_conf.py
文件中查找對(duì)應(yīng)的函數(shù)名,并將函數(shù)執(zhí)行結(jié)果返回。
loader()
函數(shù)是一個(gè)通用的東西,你可以把他放到任何位置使用。
- run.py
import webbrowser from loader import loader def test_open_url(): # 調(diào)用 baidu_url 鉤子函數(shù) # 調(diào)用 瀏覽器 訪(fǎng)問(wèn) baidu_url url = loader("baidu_url") webbrowser.open_new(url) if __name__ == '__main__': test_open_url()
通過(guò)loader()
函數(shù)執(zhí)行baidu_url
鉤子函數(shù),并拿到 url。
注意,我們不需要像傳統(tǒng)的方式一樣from run_conf import baidu_url
導(dǎo)入模塊,只要知道鉤子函數(shù)的名字即可。
這里的實(shí)現(xiàn)并沒(méi)有 pytest 那么優(yōu)雅,但也比較接近了。
到此這篇關(guān)于python強(qiáng)大的hook函數(shù)使用及應(yīng)用場(chǎng)景的文章就介紹到這了,更多相關(guān)python hook函數(shù)使用內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
Python實(shí)現(xiàn)簡(jiǎn)單石頭剪刀布小游戲的示例代碼
石頭剪刀布是一種簡(jiǎn)單而又經(jīng)典的游戲,常常用于決定勝負(fù)或者娛樂(lè)消遣,本文將使用Python實(shí)現(xiàn)一個(gè)簡(jiǎn)單的石頭剪刀布游戲,需要的可以參考一下2023-06-06基于pycharm 項(xiàng)目和項(xiàng)目文件命名規(guī)則的介紹
這篇文章主要介紹了基于pycharm 項(xiàng)目和項(xiàng)目文件命名規(guī)則的介紹,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2021-01-01python3 動(dòng)態(tài)模塊導(dǎo)入與全局變量使用實(shí)例
今天小編就為大家分享一篇python3 動(dòng)態(tài)模塊導(dǎo)入與全局變量使用實(shí)例,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-12-12python實(shí)現(xiàn)AES和RSA加解密的方法
這篇文章主要為大家詳細(xì)介紹了python實(shí)現(xiàn)AES和RSA加解密的方法,具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下2019-03-03Python實(shí)現(xiàn)的爬蟲(chóng)刷回復(fù)功能示例
這篇文章主要介紹了Python實(shí)現(xiàn)的爬蟲(chóng)刷回復(fù)功能,結(jié)合實(shí)例形式分析了Python2.7基于爬蟲(chóng)實(shí)現(xiàn)的模擬登陸、刷帖、回復(fù)等功能相關(guān)實(shí)現(xiàn)技巧,需要的朋友可以參考下2018-06-06python可視化篇之流式數(shù)據(jù)監(jiān)控的實(shí)現(xiàn)
這篇文章主要介紹了python可視化篇之流式數(shù)據(jù)監(jiān)控的實(shí)現(xiàn),文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2019-08-08