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")