欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

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

 更新時(shí)間:2023年05月06日 16:02:58   作者:蟲(chóng)師  
鉤子函數(shù)(hook function)是把我們自己實(shí)現(xiàn)的hook函數(shù)在某一時(shí)刻掛接到目標(biāo)掛載點(diǎn)上,這篇文章主要介紹了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)單石頭剪刀布小游戲的示例代碼

    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ī)則的介紹

    這篇文章主要介紹了基于pycharm 項(xiàng)目和項(xiàng)目文件命名規(guī)則的介紹,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
    2021-01-01
  • Python使用chardet判斷字符編碼

    Python使用chardet判斷字符編碼

    這篇文章主要介紹了Python使用chardet判斷字符編碼的方法,較為詳細(xì)的分析了Python中chardet的功能、安裝及使用技巧,需要的朋友可以參考下
    2015-05-05
  • Windows下Anaconda安裝、換源與更新的方法

    Windows下Anaconda安裝、換源與更新的方法

    這篇文章主要介紹了Windows下Anaconda安裝、換源與更新的方法,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧
    2020-04-04
  • python3 動(dòng)態(tài)模塊導(dǎo)入與全局變量使用實(shí)例

    python3 動(dòng)態(tài)模塊導(dǎo)入與全局變量使用實(shí)例

    今天小編就為大家分享一篇python3 動(dòng)態(tài)模塊導(dǎo)入與全局變量使用實(shí)例,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
    2019-12-12
  • PyTorch中的padding(邊緣填充)操作方式

    PyTorch中的padding(邊緣填充)操作方式

    今天小編就為大家分享一篇PyTorch中的padding(邊緣填充)操作方式,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
    2020-01-01
  • python實(shí)現(xiàn)AES和RSA加解密的方法

    python實(shí)現(xiàn)AES和RSA加解密的方法

    這篇文章主要為大家詳細(xì)介紹了python實(shí)現(xiàn)AES和RSA加解密的方法,具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下
    2019-03-03
  • python取代netcat過(guò)程分析

    python取代netcat過(guò)程分析

    本篇文章通過(guò)代碼實(shí)例給大家詳細(xì)分析了python取代netcat過(guò)程,希望我們整理的內(nèi)容能夠幫助到你。
    2018-02-02
  • Python實(shí)現(xiàn)的爬蟲(chóng)刷回復(fù)功能示例

    Python實(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-06
  • python可視化篇之流式數(shù)據(jù)監(jiān)控的實(shí)現(xiàn)

    python可視化篇之流式數(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

最新評(píng)論