【python】使用webdriver在网页输入关键词并截屏

使用webdriver在网页输入关键词并截屏：

# 导入selenium相关的工具
from selenium import webdriver

#通过浏览器加载网页
driver = webdriver.PhantomJS()
#打开网页
driver.get('https://www.baidu.com/')
#截图
driver.save_screenshot('baidu.png')
#找到要搜索的输入框控件
driver.find_element_by_id('kw').send_keys('机器猫')
#截图
driver.save_screenshot('baidu2.png')
#找到要点击的按钮
driver.find_element_by_id('su').click()
#延迟时间
import time
time.sleep(10)
#截图
driver.save_screenshot('baidu3.png')

效果图：