1

我想使用 Python Playwright 制作屏幕截图并将该屏幕截图交给 REST API。我在这里找到了一个示例，该示例制作了屏幕截图并将其保存到文件中：

from playwright import sync_playwright
with sync_playwright() as p:
    for browser_type in [p.chromium, p.firefox, p.webkit]:
        browser = browser_type.launch()
        page = browser.newPage()
        page.goto('https://scrapingant.com/')
        page.screenshot(path=f'scrapingant-{browser_type.name}.png')
        browser.close()

如何在不将其保存到磁盘或使用临时文件并将其传递给 REST 调用的情况下制作屏幕截图？

4

1 回答 1

2

您根本不需要将图像保存到文件（cmp. screenshot 文档），而是可以简单地将其存储在变量中，例如img = page.screenshot(). 然后，您可以将该变量传递给您的 REST 请求。我requests在下面的示例中使用了该模块，POST 请求被简化并且可能需要一些额外的参数（取决于您的 API）或者例如不同浏览器类型的不同 URL：

from playwright import sync_playwright
import requests

with sync_playwright() as p:
    for browser_type in [p.chromium, p.firefox, p.webkit]:
        browser = browser_type.launch()
        page = browser.newPage()
        page.goto('https://scrapingant.com/')
        # save screenshot to var
        img = page.screenshot() 
        # pass var directly to your request
        files = {'image': img, 'content-type': 'image/png'}
        requests.post('http://yourresturl.com', files=files)
        browser.close()

如果您出于某种原因确实想将图像保存到临时文件中（据我了解，这并不是真正必要的），您可以例如使用该tempfile模块并创建一个命名的临时文件（cmp。如何使用 tempfile .NamedTemporaryFile()? ):

from playwright import sync_playwright
import tempfile
import requests

tf = tempfile.NamedTemporaryFile(suffix='.png')

with sync_playwright() as p:
    for browser_type in [p.chromium, p.firefox, p.webkit]:
        browser = browser_type.launch()
        page = browser.newPage()
        page.goto('https://scrapingant.com/')
        # save screenshot to temporary file
        page.screenshot(path=tf.name)
        # send request loading temporary file
        requests.post('http://myresturl.com', {'media': open(tf.name, 'rb')})
        browser.close()        

# close temporary file
tf.close()
于 2020-12-20T00:23:00.140 回答