webdriver提供了截图函数get_screenshot_as_file()来截取当前窗口
from selenium import webdriver
from time import sleep
driver = webdriver.Firefox()
driver.get('http://www.test.com')
driver.find_element_by_id('kw').send_keys('selenium')
driver.find_element_by_id('su').click()
sleep(2)
driver.get_screenshot_as_file("D:\\test\\test_img.jpg") #截取当前窗口,并制定截图图片的保存位置
driver.quit()