插件-playwright截图

提供palywright的页面句柄或者元素句柄即可截图，返回图片路径 。

import asyncio
import datetime
import logging
import os
import random

from playwright.async_api import async_playwright, ElementHandle, Page


async def screenshot_page(page: Page | ElementHandle,
                          folder_path: str | None = None,
                          file_name: str = datetime.datetime.now().strftime(
                              '%Y-%m-%d_%H-%M-%S-%f') + f'[{random.randint(0, 99999)}]') -> str:
    """
    :param page 页面句柄或者定位器
    :param folder_path 文件夹地址
    :param file_name 文件名称
    """
    screenshot_path = f'{file_name}_element.png'
    if isinstance(page, ElementHandle):
        if folder_path:
            screenshot_path = os.path.join(folder_path, f'{file_name}_element.png')
        await page.screenshot(path=screenshot_path)
    elif isinstance(page, Page):
        if folder_path:
            screenshot_path = os.path.join(folder_path, f'{file_name}_page.png')
        await page.screenshot(path=screenshot_path, full_page=True)
    else:
        if folder_path:
            screenshot_path = os.path.join(folder_path, f'{file_name}_element.png')
        await page.screenshot(path=screenshot_path)
    return screenshot_path


# 使用示例
async def main():
    async def start_chrome():
        command = r'"C:\Users\Lenovo\AppData\Local\Google\Chrome\Application\chrome.exe" --remote-debugging-port=9998 --user-data-dir=D:/chrome_user_data/15925828527 --window-position=10000,10000 --start-maximized'
        process = await asyncio.create_subprocess_shell(command)
        return process

    # 启动 Chrome
    chrome_process = await start_chrome()

    async with async_playwright() as p:
        browser = await p.chromium.connect_over_cdp(f"http://localhost:{9998}")
        content = browser.contexts[0]
        page = await content.new_page()
        await page.goto("https://www.baidu.com", timeout=60 * 1000)
        page_path = await screenshot_page(page=page, file_name=datetime.datetime.now().strftime('%Y-%m-%d_%H-%M-%S-%f'))
        logging.info(f"Page screenshot saved to {page_path}")
        await asyncio.sleep(1)
        # 等待页面加载完成
        await page.wait_for_load_state('networkidle')
        # 选择要截图的元素（例如百度搜索框）
        element = await page.query_selector('input[name="wd"]')
        element_path = await screenshot_page(page=element,
                                             file_name=datetime.datetime.now().strftime('%Y-%m-%d_%H-%M-%S-%f'))
        logging.info(f"Element screenshot saved to {element_path}")
        await page.close()
        await content.close()
        await browser.close()


if __name__ == '__main__':
    asyncio.run(main())