python+selenium自动测试之WebDriver的常用API（基础篇二）

本篇介绍一下python+selenium复杂操作的处理，基于python3.6，selenium3.141，详细资料介绍查看官方API文档，点击这里

一、常见特殊情况处理如iframe/弹窗处理

有时候定位不到元素，是因为需要定位的元素在frame里面，这个时候需要切换到对应的frame里面，再去定位元素

# 切换frame
driver.switch_to.frame("frame")  # 切入到id="frame"的嵌套中
driver.switch_to.default_content()  # 退出当前frame, 退回到原来的主body
driver.switch_to.parent_frame()  # 返回上一级父frame
# 弹窗对话框处理
driver.switch_to.alert.text  # 获取网页上的警告信息
driver.switch_to.alert.accept  # 接受警告信息，告警弹窗对话框点击确定
driver.switch_to.alert.dismiss  # 取消告警弹窗对话框
driver.switch_to.alert.send_keys("context")  # 弹窗对话框中输入context
# 切换窗口/浏览器
driver.window_handles  # 获取所有浏览器句柄，一般用于for in 遍历
driver.switch_to.window("handle")  # 切换到handle的窗口

二、等待时间的设置

　　隐形等待，设置全局元素等待超时时间。隐性等待是指当要查找元素，而这个元素没有马上出现时，告诉WebDriver 查询Dom一定时间。默认值是0, 但是设置之后，这个时间将在WebDriver 对象实例整个生命周期都起作用。但这里有一个弊端，那就是程序会在设定的时间内一直等待整个页面加载完成，才会执行下一步，有时候个别JS加载比较慢会比较浪费时间。

# 设置隐性等待时间
driver.implicitly_wait(10)

　　显性等待，wait模块的WebDriverWait()类，配合until或者until_not方法，再辅以一些判断条件expected_conditions模块，就可以构成这样一个场景：每经过多少秒就查看一次locator的元素是否可见，如果可见就停止等待，如果不可见就继续等待直到超过规定的时间后，报超时异常；如果同时设置隐性等待和显性等待，取最大等待时间。

# 设置显性等待
from selenium.webdriver.support import expected_conditions as EC
"""
 __init__(self, driver, timeout, poll_frequency=POLL_FREQUENCY, ignored_exceptions=None)
 - driver - Instance of WebDriver (Ie, Firefox, Chrome or Remote)
 - timeout - 超时时间
 - poll_frequency - 迭代间隔时间，停几秒重新获取一次，默认0.5S
 - ignored_exceptions - 超时时间内忽略的异常，默认只有NoSuchElementException
"""
WebDriverWait(driver, 20, 1).until(EC.presence_of_element_located((By.ID, 'foo')))

　　expected_conditions模块常用的判断条件，这里根据官方API翻译一下，仅供参考

三、Cookie处理

driver.get_cookies()  # 获得所有cookie信息
driver.get_cookie("name")  # 返回指定name的cookie信息
driver.add_cookie({'name': 'key', 'value': 'value'})  # 添加cookie 
driver.delete_cookie("name")  # 删除指定name的cookie 
driver.delete_all_cookies()  # 删除所有cookie

四、其他操作(下拉框/文件上传等)

# 下拉框处理
from selenium.webdriver.support.ui import Select
# when given "foo" this would select an option like: <option value="foo">Bar</option>
Select(driver.find_element_by_xpath('xpath')).select_by_value("foo")
# 上传文件，使用send_keys()发送一个本地的文件路径，从而达到上传文件的目的
driver.find_element_by_name("file").send_keys('D:\\test_file.xlsx')
# 截图,必须以".png"结尾
driver.get_screenshot_as_file("text.png")