playwright图片爬取到本地

from playwright.sync_api import sync_playwright


def save_image_sync(url, path):
    with sync_playwright() as p:
        browser = p.chromium.launch()
        page = browser.new_page()
        page.goto(url)

        # 假设图片在第一个 <img> 标签中
        img_handle = page.query_selector('img')
        img_src = img_handle.get_attribute('src')

        # 下载图片
        if img_src:
            response = page.request.get(img_src)
            with open(path, 'wb') as f:
                f.write(response.body())
            print(f'图片已保存到 {path}')

        browser.close()


if __name__ == '__main__':
    url = 'https://image.baidu.com/'  # 目标网页
    path = r"C:\Users\MR\Desktop\local_image.png"  # 保存路径
    save_image_sync(url, path)