python selenium實現百度搜索

1.環境

　　python2.7+selenium+phantomjs+linux

2.代碼

#-*-coding:utf-8 -*-
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
import time

driver = webdriver.PhantomJS()
driver.maximize_window() #窗口最大化
driver.get('http://www.baidu.com')
driver.save_screenshot('1.png') #保存圖片
element = driver.find_element_by_id('kw')
element.send_keys('sina') #輸入搜索內容
element.send_keys(Keys.RETURN) #模擬回車
time.sleep(1) #延時長短跟頁面加載時間有關
driver.save_screenshot('2.png')
driver.quit()