from selenium import webdriver # Create a new cromedriver driver = webdriver.Chrome() # Go to www.google.com driver.get("https://www.google.com") # Saves a .png file with name my_screenshot_name to the directory that # you are running the program from. screenshot_name = "my_screenshot_name.png" driver.save_screenshot(screenshot_name)
driver.save_screenshot will return 'true' if the screenshot was taken, and 'false' if it was not. Saving screenshots also works with headless browsers. If you want to save a screenshot in a different directory, just add the filepath (relative to where you are running the code from). For example:
screenshot_name = "screenshots/my_screenshot_name.png"
Will save the screenshot in directory "screenshots" inside the directory that python is being run from.