selenium screenshot python

1from selenium import webdriver
2from time import sleep
3
4driver = webdriver.Firefox()
5driver.get('https://www.python.org')
6sleep(1)
7
8driver.get_screenshot_as_file("screenshot.png")
9driver.quit()
10print("end...")