python - Webdriver 截图

Question

用python在windows上使用Selenium Webdriver截图时，截图直接保存到程序的路径下，有没有办法将.png文件保存到特定目录？

Answer 1

使用 driver.save_screenshot('/path/to/file') 或 driver.get_screenshot_as_file('/path/to/file'):

import selenium.webdriver as webdriver
import contextlib

@contextlib.contextmanager
def quitting(thing):
    yield thing
    thing.quit()

with quitting(webdriver.Firefox()) as driver:
    driver.implicitly_wait(10)
    driver.get('http://www.google.com')
    driver.get_screenshot_as_file('/tmp/google.png') 
    # driver.save_screenshot('/tmp/google.png')