最新国产好看的视频,伊人天堂AV在线,国产Aaaaaa视频,蜜臀视频在线观看一区,人妻av色图,密臀久久久精品影片,青青视频免费观看毛片,久草在线观看视,国产三级精品色情在线

python強大的hook函數(shù)使用及應(yīng)用場景

 更新時間:2023年05月06日 16:02:58   作者:蟲師  
鉤子函數(shù)(hook function)是把我們自己實現(xiàn)的hook函數(shù)在某一時刻掛接到目標(biāo)掛載點上,這篇文章主要介紹了python強大的hook函數(shù)使用及應(yīng)用場景,需要的朋友可以參考下

什么是hook?

鉤子函數(shù)(hook function),可以理解是一個掛鉤,作用是有需要的時候掛一個東西上去。具體的解釋是:鉤子函數(shù)是把我們自己實現(xiàn)的hook函數(shù)在某一時刻掛接到目標(biāo)掛載點上。

hook應(yīng)用場景(一)

相信你對鉤子函數(shù)并不陌生。我在 requests 和 mitmproxy 都有看到類似的設(shè)計。

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)碼,這個動作,我們可以封裝到一個函數(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})

代碼說明:

把打印狀態(tài)碼封裝到一個status_code() 函數(shù)中,在requests.get() 方法中通過hooks 參數(shù)接收鉤子函數(shù)status_code()。

運行結(jié)果:

> python requests_hooks.py
hook status doce: 200

status_code() 作為一個函數(shù),可以做的事情很多,比如,進一步判斷狀態(tài)碼,打印響應(yīng)的數(shù)據(jù),甚至對相應(yīng)的數(shù)據(jù)做加解密等處理。

mitmproxy 中的hook

mitmproxy是一個代理工具,我們這之前的文章也有做過介紹。在抓包的過程中,同樣需要用到 hooks 去對request請求或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()]

運行mitmproxy

> mitmproxy -s anatomy.py

自己實現(xiàn)hook

什么情況下需要實現(xiàn)hook,就是一個功能(類/方法)自身無法滿足所有需求,那么可以通過hook 就提供擴展自身能力的可能。

實現(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()

代碼說明:

在上面的例子中Programmer類實現(xiàn)三個功能:eat、code、sleep,但程序員也是普通人,不能每天都只吃飯、編碼、睡覺,于是通過register_hook() 提供了做別的事情的能力。

那么,看看Tom、Jerry、Spike三位主角,今天都干了什么吧!

運行結(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)用場景(二)

如果把hook理解為:定義一個函數(shù),然后作為參數(shù)塞到另一個類/方法里。 顯然,這只是一種用法。我重新想了一下。httpRunner 的 debugtalk.py 文件; pytest 的 conftest.py 文件,他們本身也是擁有特殊名字的 hook文件。程序在執(zhí)行的過程中,調(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)用 瀏覽器 訪問 baidu_url
    webbrowser.open_new(baidu_url)

兩個文件看似沒有直接的調(diào)用關(guān)系,在執(zhí)行 test_sample.py 文件時,可以間接的調(diào)用 conftest.py 文件中的baidu_url()鉤子函數(shù)。

執(zhí)行測試

> pytest -q test_sample.py

實現(xiàn)動態(tài)調(diào)用hook

接下來,我們來試試做個類似的功能出來。

└───project
    ├───run_conf.py
    ├───loader.py
    └───run.py
  • run_conf.py
def baidu_url():
    """定義鉤子函數(shù)"""
    name = "https://www.baidu.com"
    return name

與 conftest.py 文件類似,在這個文件中實現(xiàn)鉤子函數(shù)。

  • loader.py
import os
import inspect
import importlib
def loader(name):
    """
    動態(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))
    # 動態(tài)加載 *_config.py
    hooks = []
    for module_name in all_hook_module:
        hooks.append(importlib.import_module(module_name))
    # 根據(jù)傳過來的 name 函數(shù)名,從 *_conf.py 文件查找并執(zhí)行。
    for per_hook in hooks:
        # 動態(tài)執(zhí)行 process 函數(shù)
        func = getattr(per_hook, name)
        return func()

這個東西就比較復(fù)雜了,他的作用就是丟給他一個的函數(shù)名, 他能通過*_conf.py文件中查找對應(yīng)的函數(shù)名,并將函數(shù)執(zhí)行結(jié)果返回。

loader() 函數(shù)是一個通用的東西,你可以把他放到任何位置使用。

  • run.py
import webbrowser
from loader import loader
def test_open_url():
    # 調(diào)用 baidu_url 鉤子函數(shù)
    # 調(diào)用 瀏覽器 訪問 baidu_url
    url = loader("baidu_url")
    webbrowser.open_new(url)
if __name__ == '__main__':
    test_open_url()

通過loader() 函數(shù)執(zhí)行baidu_url 鉤子函數(shù),并拿到 url。

注意,我們不需要像傳統(tǒng)的方式一樣from run_conf import baidu_url 導(dǎo)入模塊,只要知道鉤子函數(shù)的名字即可。

這里的實現(xiàn)并沒有 pytest 那么優(yōu)雅,但也比較接近了。

到此這篇關(guān)于python強大的hook函數(shù)使用及應(yīng)用場景的文章就介紹到這了,更多相關(guān)python hook函數(shù)使用內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!

相關(guān)文章

最新評論

渑池县| 抚远县| 叙永县| 沙洋县| 万全县| 禹城市| 铜川市| 扎兰屯市| 绥阳县| 满洲里市| 车致| 闽侯县| 罗定市| 贵港市| 卓尼县| 乌拉特前旗| 博白县| 肥城市| 明溪县| 莱芜市| 兰西县| 衢州市| 双辽市| 吉水县| 大安市| 定结县| 鲁甸县| 宜州市| 偃师市| 九台市| 离岛区| 勐海县| 苏尼特左旗| 南昌县| 桐乡市| 德惠市| 韶关市| 襄汾县| 沙洋县| 盈江县| 长乐市|