# 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()   # 浏览器退出