python - 如何在 Firefox 或 Chrome 中使用 Selenium webdriver 更改屏幕截图的目标目录

我想制作网页的屏幕截图并将其保存在 自定义位置 使用 Selenium webdriver 和 Python。我尝试使用 Firefox 和 Chrome 将屏幕截图保存到自定义位置，但它始终将屏幕截图保存在项目目录中。这是我的 Firefox 版本:

from selenium import webdriver
from selenium.webdriver.firefox.firefox_binary import FirefoxBinary

profile = webdriver.FirefoxProfile()
profile.set_preference("browser.download.folderList", 2)
profile.set_preference("browser.download.dir",
'C:\\Users\\User\\WebstormProjects')
binary = FirefoxBinary("C:\\Program Files\\Mozilla Firefox\\firefox.exe")


def foxScreen():
    driver = webdriver.Firefox(firefox_binary=binary,
    firefox_profile=profile)
    driver.get("http://google.com")
    driver.save_screenshot("foxScreen.png")
    driver.quit()


if __name__ == '__main__':
    foxScreen()

这是我的 Chrome 版本:

from selenium import webdriver

options = webdriver.ChromeOptions()
prefs = {"download.default_directory": r'C:\\Users\\User\\WebstormProjects',
         "directory_upgrade": True}
options.add_experimental_option("prefs", prefs)
chromedriver =
"C:\\Users\\User\\Downloads\\chromedriver_win32\\chromedriver.exe"


def chromeScreen():
    driver = webdriver.Chrome(chrome_options=options,
                              executable_path=chromedriver)
    driver.get("http://google.com")
    driver.save_screenshot("chromeScreen.png")
    driver.quit()


if __name__ == '__main__':
    chromeScreen()

我已经为我想要保存屏幕截图的位置尝试了不同的符号，但这似乎没有帮助。我应该更改什么，以便它不会将屏幕截图保存到项目目录，而是保存到给定的自定义位置？

最佳答案

您需要考虑以下几个事实:

profile.set_preference('key', 'value')

set_preference(key, value) 在 firefox_profile 中设置我们想要的首选项。此首选项在调用特定 Firefox Profile 时生效。

save_screenshot(文件名)

根据文档 save_screenshot(filename) 将当前窗口的屏幕截图保存为 PNG 图像文件。如果有任何 IOError，此方法返回 False，否则返回 True。 在文件名中使用完整路径 。

Args :

filename: The full path you wish to save your screenshot to. This should end with a .png extension.

Usage :

driver.save_screenshot(‘/Screenshots/foo.png’)

因此， save_screenshot(filename) 需要您希望将屏幕截图保存到的完整路径。当您使用时:

driver.save_screenshot("foxScreen.png")

因此，屏幕截图始终保存在项目目录中。

解决方案

要将屏幕截图保存在不同的目录中，您需要传递 绝对路径 如下:

driver.save_screenshot("./my_directory/foo.png")

引用

您可以在 How to take screenshot with Selenium WebDriver 中找到详细的讨论

关于python - 如何在 Firefox 或 Chrome 中使用 Selenium webdriver 更改屏幕截图的目标目录，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/51286846/