WD.py是一个Python WebDriver客户端，实现了WebDriver协议中的大部分API。 它最初是为Macaca（一个Node.js驱动的WebDriver服务器）而设计的，但也可以应用于WebDriver服务器的其他实现，比如Selenium，Appium等。

自动化测试

查找元素

WebDriver的查找元素命令允许分别查找单个元素和元素集合，定位策略如下：

• 文本
• id
• XPath
• 链接文本
• 部分链接文本
• 标签名
• 类名
• CSS选择器

找到元素的基本方法是element()，例如查找id为“login”的元素：

driver.element('id', 'login')

但是在大多数情况下，不需要使用这个基本的方法，为了方便起见，有很多扩展方法。

文本

例如，查找页面上name属性为“Login failed!”的元素：

driver.element_by_name('Login failed!')

id

例如，查找id为“login”的元素：

driver.element_by_id('login')

XPath

XPath是XML Path的简称，由于HTML文档本身就是一个标准的XML页面，所以可以使用XPath的语法来定位页面元素。这个方法是非常强大的元素查找方式，使用这种方法几乎可以定位到页面上的任意元素。

例如，查找页面上id为“finding-elements-to-interact”的元素下的第4个table元素：

driver.element_by_xpath('//*[@id="finding-elements-to-interact"]/table[4]')

链接文本

这个方法比较直接，即通过超文本链接上的文字信息来定位元素，这种方式一般专门用于定位页面上的超文本链接。

例如，查找页面上文字为“macaca”的超文本链接：

driver.element_by_link_text('macaca')

部分链接文本

这个方法是上一个方法的扩展，当不能准确知道超链接上的文本信息或者只想通过一些关键字进行匹配时，可以使用这个方法来通过部分链接文字进行匹配。

例如，查找页面上部分文字为“maca”的超文本链接：

driver.element_by_partial_link_text('maca')

标签名

该方法可以通过元素的标签名称来查找元素，需要注意的是，这个方法搜索到的元素通常不止一个。

例如，查找页面上的“ input”标签：

driver.element_by_tag_name('input')

此外，WebDriver上的所有元素方法都可以在WebElement上使用，也就是从当前Web元素中查找元素：

web_element.element_by_id('ss')

类名

一般程序员或页面设计师会给元素直接赋予一个样式属性或者利用css文件里的伪类来定义元素样式，这个方法可以利用元素的css样式表所引用的伪类名称来进行元素查找。

例如，查找页面上className属性为“btn”的元素：

driver.element_by_class_name('btn')

CSS选择器

这种元素定位方式跟XPath比较类似，但执行速度较快，所以功能也是蛮强大的。

例如，查找页面上样式表为“.btn”的元素：

driver.element_by_css_selector('.btn')

异常处理

当没有找到元素时，会引发WebDriverException异常。为了避免这种情况，可是使用element_if_exists方法，如果元素存在，则返回True，否则返回False，例如：

driver.element_by_id_if_exists('login')

也可以使用element_or_none方法，如果元素存在，则返回元素，否则返回None，例如：

driver.element_by_id_or_none('login')

此外，还有wait_for方法等待元素满足给定条件，默认等待10秒，每个间隔1秒，断言器函数默认为asserters.is_displayed，例如：

driver.wait_for_element_by_id('login')

按键输入

当需要完成一个输入字段的操作时，可以将一系列的按键行为发送给一个元素：

driver.web_element.send_keys('123456')

send_keys方法也接受一个数组，这在发送特殊键（不是文本的按键）时非常有用：

driver.web_element.send_keys([1, 2, 3, 4, 5, 6])

PC按键映射：

Android按键映射：

iOS按键映射：

使用数组发送特殊键非常方便：

driver.web_element.send_keys([1, DELETE, 1, 2, 3, 4, 5, 6])

屏幕快照

截图时可以返回截图的base64编码字符串：

base64_str = driver.take_screenshot()

或者保存截图到给定的路径：

driver.save_screenshot('./screen.png')

save_screenshot方法具有可选的第二个参数来决定是否由于某种原因无法保存到文件系统时忽略IOError。例如，没有读写权限时，忽略异常信息：

driver.save_screenshot('/etc/screen.png', True)

切换环境

对于移动端测试，可能需要在Native（原生）和Webview（H5）之间切换环境，首先获取现有的环境：

ctxs = driver.contexts
print(ctxs) # ['NATIVE', 'WEBVIEW_1', 'WEBVIEW_2']

然后切换到指定的环境：

driver.context = 'WEBVIEW_1'
print(driver.context) # WEBVIEW_1

执行JS片段

在一些复杂的情况下，可能需要在页面中插入一段JavaScript代码，并得到想要的任何东西。可以在脚本中使用arguments来表示脚本之后的索引参数。

script = 'return document.querySelector(".btn").tagName === arguments[0]'
args = ['div']
result = driver.execute_script(script, *args)

上面的脚本等于JavaScript中的IIFE：

function () {
  return document.querySelector(".btn").tagName === "div"
}()

WebElement方法

WebElement的实例方法主要与行为元素有关，比如点击元素、获取标签名或元素的内部文本。WebElement实例是通过查找元素命令返回的，例如通过id检索元素：

web_element = driver.element_by_id('login')
print(type(web_element) == WebElement) # True

例如点击元素：

web_element.click()

例如获取元素的标签名称：

tag_name = web_element.tag_name

例如获取元素的内部文本：

text = web_element.text

编码风格

建议使用官方推荐的两种编码风格，第一种是使用额外的括号：

(
    driver
        .get('https://www.google.com')
        .element_by_id('login')
        .click()
)

第二种是使用反斜杠：

driver                             \
    .get('https://www.google.com') \
    .element_by_id('login')        \
    .click()