# coding=utf-8
from selenium import webdriver
import time
driver = webdriver.Chrome(r"C:\Program Files (x86)\Google\Chrome\Application\chromedriver.exe")
driver.get('https://www.baidu.com/')
# ----------------------------------
driver.get_screenshot_as_file('shot.png')
ele = driver.find_element_by_id('su')
ele.screenshot('e:\\button.png')
# ----------------------------------
input('press any key to quit...')
driver.quit() # 浏览器退出
