python代码 playwright库处理勾子函数

Playwright是一个用于自动化浏览器的Python库。要使用勾子函数，可以使用Playwright提供的 `before_request` 和 `after_request` 函数。这些函数允许您在发送请求之前或之后执行代码。

以下是一个示例代码，它使用 `before_request` 函数来添加自定义请求标头，使用 `after_request` 函数来记录请求的响应时间：

import asyncio
from playwright.async_api import Playwright, async_playwright

async def intercept_request(request):
    headers = request.headers()
    headers['X-Custom-Header'] = 'Custom Value'
    await request.continue_(headers=headers)

async def log_response_time(response):
    print(f"Response time: {response.timing()['responseEnd'] - response.timing()['requestStart']} ms")

async def main(playwright: Playwright) -> None:
    browser = await playwright.chromium.launch()
    page = await browser.new_page()

    # 添加 before_request 和 after_request 勾子
    page.on('request', intercept_request)
    page.on('response', log_response_time)

    await page.goto('https://www.example.com')
    await browser.close()

async with async_playwright() as playwright:
    asyncio.get_event_loop().run_until_complete(main(playwright))

在上面的代码中，我们添加了两个勾子函数，一个用于在请求之前添加自定义标头，另一个用于在请求完成后记录响应时间。这些勾子函数都是异步函数，并在创建页面后使用 `page.on` 方法添加到页面中。

当页面开始请求时，`intercept_request` 函数将被调用，并传递请求对象。我们可以在这个函数中修改请求标头，并通过调用 `request.continue_` 方法继续请求。

当页面收到响应时，`log_response_time` 函数将被调用，并传递响应对象。我们可以在这个函数中记录响应时间，并将其打印出来。

总体来说，Playwright的勾子函数提供了很多灵活性，可以让您在自动化浏览器时执行更多自定义操作。