[TOC]
Install
pip install selenium
selenium 是自动测试工具，支持各种界面式浏览器，实际上就是在这些浏览器里面安装一个 selenium 的插件，方便地实现 Web 界面测试，即 selenium 支持这些浏览器驱动。PhantomJS 也是一个浏览器，selenium 也能够支持。
Demo
from selenium import webdriver
browser = webdriver.Chrome()
browser.get('http://www.baidu.com/')
驱动文件
放在 path 路径下
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
driver = webdriver.Chrome()
driver.get("http://www.python.org")
assert "Python" in driver.title
elem = driver.find_element_by_name("q")
elem.send_keys("pycon")
elem.send_keys(Keys.RETURN)
print driver.page_source
测试用例：
import unittest
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
class PythonOrgSearch(unittest.TestCase):
def setUp(self):
self.driver = webdriver.Chrome()
def test_search_in_python_org(self):
driver = self.driver
driver.get("http://www.python.org")
self.assertIn("Python", driver.title)
elem = driver.find_element_by_name("q")
elem.send_keys("pycon")
elem.send_keys(Keys.RETURN)
assert "No results found." not in driver.page_source
def tearDown(self):
self.driver.close()
if __name__ == "__main__":
unittest.main()
测试用例是继承了 unittest.TestCase 类，继承这个类表明这是一个测试类。setUp方法是初始化的方法，这个方法会在每个测试类中自动调用。每一个测试方法命名都有规范，必须以 test 开头，会自动执行。最后的 tearDown 方法会在每一个测试方法结束之后调用。这相当于最后的析构方法。在这个方法里写的是 close 方法，你还可以写 quit 方法。不过 close 方法相当于关闭了这个 TAB 选项卡，然而 quit 是退出了整个浏览器。当你只开启了一个 TAB 选项卡的时候，关闭的时候也会将整个浏览器关闭。
页面交互
假如有一个表单输入框：
<input type="text" name="passwd" id="passwd-id" />
元素选取
单个元素选取：
- find_element_by_id
- find_element_by_name
- find_element_by_xpath
- find_element_by_link_text
- find_element_by_partial_link_text
- find_element_by_tag_name
- find_element_by_class_name
- find_element_by_css_selector
多个元素选取：
- find_elements_by_name
- find_elements_by_xpath
- find_elements_by_link_text
- find_elements_by_partial_link_text
- find_elements_by_tag_name
- find_elements_by_class_name
- find_elements_by_css_selector
还可以用 By 类确定哪种选择方式：
from selenium.webdriver.common.by import By
driver.find_element(By.XPATH, '//button[text()="Some text"]')
driver.find_elements(By.XPATH, '//button')
By 类的一些属性如下：
ID = "id"
XPATH = "xpath"
LINK_TEXT = "link text"
PARTIAL_LINK_TEXT = "partial link text"
NAME = "name"
TAG_NAME = "tag name"
CLASS_NAME = "class name"
CSS_SELECTOR = "css selector"
可以获取这个元素：
element = driver.find_element_by_id("passwd-id")
element = driver.find_element_by_name("passwd")
element = driver.find_elements_by_tag_name("input")
element = driver.find_element_by_xpath("//input[@id='passwd-id']")
element = driver.find_element_by_id("q")
element.size() //大小 dict
element.get_attribute("type") //获取元素属性
element.is_displayed() //是否可见
element.text //获取文本信息
driver.current_url //当前url
用 xpath 时需要注意，如果有多个元素匹配了 xpath，只会返回第一个匹配的元素。如果没有找到，会抛出 NoSuchElementException 的异常。
填充表单
向文本输入内容
element.send_keys("some text")
还可以利用 Keys 这个类模拟点击某个按键
element.send_keys("and some", Keys.ARROW_DOWN)
element.clear() //输入的文本会被清除
Select 方法
下拉选项卡处理：
element = driver.find_element_by_xpath("//select[@name='name']")
all_options = element.find_elements_by_tag_name("option")
for option in all_options:
print("Value is: %s" % option.get_attribute("value"))
option.click()
WebDriver 提供更简单的方法 Select：
from selenium.webdriver.support.ui import Select
select = Select(driver.find_element_by_name('name'))
select.select_by_index(index)
select.select_by_visible_text("text")
select.select_by_value(value)
可以根据索引来选择，根据值来选择，根据文字来选择。
全部取消选择：
select = Select(driver.find_element_by_id('id'))
select.deselect_all()
获取所有已选选项：
select = Select(driver.find_element_by_xpath("xpath"))
all_selected_options = select.all_selected_options
获取所有可选选项：
options = select.options
提交表单
driver.find_element_by_id("submit").click()
相当于模拟点击 submit 按钮提交表单
也可以单独提交某个元素
element.submit() WebDriver 会在表单中寻找该元素所在表单，如果发现该元素没有被表单包围，程序会抛出 NoSuchElementException
元素拖拽
指定被拖动的元素和拖动目标元素，然后利用 ActionChains 类实现
element = driver.find_element_by_name("source")
target = driver.find_element_by_name("target")
from selenium.webdriver import ActionChains
action_chains = ActionChains(driver)
action_chains.drag_and_drop(element, target).perform()
元素从 source 拖动到 target
页面窗口切换
一个浏览器肯定有很多窗口，所以肯定有方法实现窗口的切换：
driver.switch_to_window("windowName")
另外可以使用 window_handles 方法获取每个窗口的操作对象：
for handle in driver.window_handles:
driver.switch_to_window(handle)
切换 frame：
driver.switch_to_frame("frameName.0.child")
之后焦点会切换到 name 为 child 的 frame 上
弹窗处理
当触发了某个事件之后，页面出现了弹框提示，如何处理呢？
alert = driver.switch_to_alert() 可以获取弹窗对象
.get_text()
.accept() //点击确定
.dismiss() //点击取消
send_keys() //输入值
历史纪录
前进和后台：
driver.forward()
driver.back()
Cookies处理
给页面添加 Cookies：
# Go to the correct domain
driver.get("http://www.example.com")
# Now set the cookie. This one's valid for the entire domain
cookie = {'name' : 'foo', 'value' : 'bar'}
driver.add_cookie(cookie)
获取页面 Cookies：
# Go to the correct domain
driver.get("http://www.example.com")
# And now output all the available cookies for the current URL
driver.get_cookies()
页面等待
这是非常重要的一部分，现代网页越来越多采用 Ajax 技术，这样程序就不能确定某个元素何时加载完毕，会让元素定位困难并且报出 ElementNotVisibleException 的概率提高
隐式等待：等待特定的时间
显式等待：指定某一条件，直到这个条件成立时继续执行
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
try:
element = WebDriverWait(driver, 10).until(
EC.presence_of_element_located((By.ID, "myDynamicElement"))
)
finally:
driver.quit()
显式等待
显式等待指定某个条件，设置最长等待时间，如果这个时间内还没有找到元素，便会抛出异常
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
driver = webdriver.Chrome()
driver.get("http://somedomain/url_that_delays_loading")
try:
element = WebDriverWait(driver, 10).until(
EC.presence_of_element_located((By.ID, "myDynamicElement"))
)
finally:
driver.quit()
程序默认会 500ms 调用一次来查看元素是否已经生成，如果本来元素就是存在的，那么会立即返回
下面是一些内置的等待条件，可以直接调用这些条件，而不用自己写某些等待条件了：
title_is
title_contains
presence_of_element_located
visibility_of_element_located
visibility_of
presence_of_all_elements_located
text_to_be_present_in_element
text_to_be_present_in_element_value
frame_to_be_available_and_switch_to_it
invisibility_of_element_located
element_to_be_clickable – it is Displayed and Enabled.
staleness_of
element_to_be_selected
element_located_to_be_selected
element_selection_state_to_be
element_located_selection_state_to_be
alert_is_present
from selenium.webdriver.support import expected_conditions as EC
wait = WebDriverWait(driver, 10)
element = wait.until(EC.element_to_be_clickable((By.ID,'someid')))
隐式等待
隐式等待比较简单，就是简单地设置一个等待时间，单位为秒。
from selenium import webdriver
driver = webdriver.Chrome()
driver.implicitly_wait(10) # seconds
driver.get("http://somedomain/url_that_delays_loading")
myDynamicElement = driver.find_element_by_id("myDynamicElement")
Demo
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
driver = wendriver.Firefox()
driver.get("http://www.python.org")
assert "Python" in
elem = driver.find_element_by_name("q")
elem.clear()
elem.send_keys("pycon")
elem.send_keys(Keys.RETURN)
assert "No results found." not in driver.page_source
driver.close()
Http 协议分析
GET / HTTP/1.1\r\n
Host: rk-zyk.zx.ga\r\n
Connection: Keep-Alive\r\n
例子
from selenium import webdriver
from selenium.webdriver.common.action_chains import ActionChains
driver = webdriver.Firefox()
driver.get('http://www.baidu.com')
driver.maximize_window()
setting = driver.find_element_by_link_text(u'设置')
ActionChains(driver).move_to_element(setting).perform()
driver.find_element_by_link_text(u'高级搜索'),click()
