python - Chrome/Firefox 在新选项卡中打开 PDF，并且不会以 headless 模式保存它 (Selenium+Python)

Question

执行测试时，我遇到 headless Chrome 的问题:单击按钮时会在新选项卡中打开 PDF 文件。如果我在非 headless 模式下运行测试，一切都很好。但是当尝试在 headless 中执行相同操作时 - 文件未下载。

options = ChromeOptions()
            options.add_argument('--no-sandbox')
            options.add_argument('--kiosk-printing')
            options.add_argument('--test-type')
            options.add_argument('--disable-infobars')
            options.add_argument('disable-gpu')
            options.add_argument('--verbose')
            options.add_argument('--disable-extensions')
            options.add_argument('--ignore-certificate-errors')
            options.add_experimental_option("prefs", {
                "profile.default_content_settings.popups": 0,
                "download.default_directory": dwnld_path,
                "download.prompt_for_download": False,
                "download.directory_upgrade": True,
                "safebrowsing.enabled": False,
                "plugins.always_open_pdf_externally": True,
                "plugins.plugins_disabled": ["Chrome PDF Viewer"]
            })

我还发现:

wd.command_executor._commands["send_command"] = ("POST", '/session/$sessionId/chromium/send_command')

params = {'cmd': 'Page.setDownloadBehavior', 'params': {'behavior': 'allow', 'downloadPath': dwnld_path}}
        command_result = wd.execute("send_command", params)

但只有当我收到以 headless 模式下载的请求时它才会有帮助，而不是在浏览器中打开文件时才有帮助。

Answer 1

尝试保存 PDF 文件的 url 并使用 requests 库下载它，我认为它会起作用。

像这样:

import urllib3
import PyPDF2
import certifi
import io

http = urllib3.PoolManager(cert_reqs='CERT_REQUIRED', ca_certs=certifi.where())
pdf_url = "http:\\XXXXXX.pdf"
r3 = http.request('GET', pdf_url)
with io.BytesIO(r3.data) as open_pdf_file:
      read_pdf = PyPDF2.PdfFileReader(open_pdf_file)
      num_pages = read_pdf.getNumPages()

然后我们需要代码的第二部分，您需要以类似的方式保存 pdf