python-selenium开发（2：高阶用法）

阳光穿透心脏的1/2处 2022-11-14 13:25 190阅读 0赞

基础用法请参考：[python-selenium开发（1：基础用法）][python-selenium_1]  
**1.表单切换**  
在Web应用中经常会遇到frame/iframe表单嵌套页面的应用，WebDriver只能在一个页面上对元素识别与定位，对于frame/iframe表单内嵌页面上的元素无法直接定位。这时就需要通过  
1.switch\_to.frame()方法将当前定位的主体切换为frame/iframe表单的内嵌页面中。  
2.switch\_to.default\_content()跳回最外层的页面  
[参考链接][Link 1]

**2.窗口切换**  
在页面操作过程中有时候点击某个链接会弹出新的窗口，这时就需要主机切换到新打开的窗口上进行操作。  
WebDriver提供了switch\_to.window()方法，可以实现在不同的窗口之间切换。 以百度首页和百度注册页为例，在两个窗口之间的切换如下图。

current_window_handle：获得当前窗口句柄。
    window_handles：返回所有窗口的句柄到当前会话。
    switch_to.window()：用于切换到相应的窗口，与上一节的switch_to.frame()类似，前者用于不同窗口的切换，后者用于不同表单之间的切换。

from selenium import webdriver
    import time
    
    driver = webdriver.Firefox()
    driver.implicitly_wait(10)
    driver.get("http://www.baidu.com")
    
    # 获得百度搜索窗口句柄
    sreach_windows = driver.current_window_handle
    
    driver.find_element_by_link_text('登录').click()
    driver.find_element_by_link_text("立即注册").click()
    
    # 获得当前所有打开的窗口的句柄
    all_handles = driver.window_handles
    
    # 进入注册窗口
    for handle in all_handles:
        if handle != sreach_windows:
            driver.switch_to.window(handle)
            print('now register window!')
            driver.find_element_by_name("account").send_keys('username')
            driver.find_element_by_name('password').send_keys('password')
            time.sleep(2)
    driver.quit()

**3.警告框处理**

在WebDriver中处理JavaScript所生成的alert、confirm以及prompt十分简单，具体做法是使用 switch\_to.alert 方法定位到 alert/confirm/prompt，然后使用text/accept/dismiss/ send\_keys等方法进行操作。

text：返回 alert/confirm/prompt 中的文字信息。
    accept()：接受现有警告框。
    dismiss()：解散现有警告框。
    send_keys(keysToSend)：发送文本至警告框。keysToSend：将文本发送至警告框。

from selenium import webdriver
    from selenium.webdriver.common.action_chains import ActionChains
    import time
    
    driver = webdriver.Firefox()
    driver.implicitly_wait(10)
    driver.get('http://www.baidu.com')
    
    # 鼠标悬停至“设置”链接
    link = driver.find_element_by_link_text('设置')
    ActionChains(driver).move_to_element(link).perform()
    
    # 打开搜索设置
    driver.find_element_by_link_text("搜索设置").click()
    
    # 保存设置
    driver.find_element_by_class_name("prefpanelgo").click()
    time.sleep(2)
    
    # 接受警告框
    driver.switch_to.alert.accept()
    
    driver.quit()

通过switch\_to\_alert()方法获取当前页面上的警告框，并使用accept()方法接受警告框。

**4.下拉框选择**  
WebDriver提供了Select类来处理下拉框。  
Select类用于定位select标签。  
select\_by\_value() 方法用于定位下接选项中的value值。

from selenium import webdriver
    from selenium.webdriver.support.select import Select
    from time import sleep
    
    driver = webdriver.Chrome()
    driver.implicitly_wait(10)
    driver.get('http://www.baidu.com')
    
    # 鼠标悬停至“设置”链接
    driver.find_element_by_link_text('设置').click()
    sleep(1)
    # 打开搜索设置
    driver.find_element_by_link_text("搜索设置").click()
    sleep(2)
    # 搜索结果显示条数
    sel = driver.find_element_by_xpath("//select[@id='nr'])
    Select(sel).select_by_value('50')  # 显示50条
    driver.quit()

**5.文件上传**  
通过input标签实现上传功能，可以将其看作是一个输入框，即通过send\_keys()指定本地文件路径的方式实现文件上传。

from selenium import webdriver
    import os
    driver = webdriver.Firefox()
    file_path = 'file:///' + os.path.abspath('upfile.html')
    driver.get(file_path)
    # 定位上传按钮，添加本地文件
    driver.find_element_by_name("file").send_keys('D:\\upload_file.txt')
    driver.quit()

**file\_path是网页的网址**

**6.cookie操作**

有时候我们需要验证浏览器中cookie是否正确，因为基于真实cookie的测试是无法通过白盒和集成测试进行的。WebDriver提供了操作Cookie的相关方法，可以读取、添加和删除cookie信息。

WebDriver操作cookie的方法：

get_cookies()： 获得所有cookie信息。
    get_cookie(name)： 返回字典的key为“name”的cookie信息。
    add_cookie(cookie_dict) ： 添加cookie。“cookie_dict”指字典对象，必须有name 和value 值。
    delete_cookie(name,optionsString)：删除cookie信息。“name”是要删除的cookie的名称，“optionsString”是该cookie的选项，目前支持的选项包括“路径”，“域”。
    delete_all_cookies()： 删除所有cookie信息。

from selenium import webdriver
    
    driver = webdriver.Firefox()
    driver.get("http://www.youdao.com")
    
    # 获得cookie信息
    cookie= driver.get_cookies()
    # 将获得cookie的信息打印
    print(cookie)
    driver.quit()

cookie数据是以字典的形式进行存放的，可以按照这种形式向浏览器中写入cookie信息。

from selenium import webdriver
    
    driver = webdriver.Firefox()
    driver.get("http://www.youdao.com")
    # 向cookie的name 和value中添加会话信息
    driver.add_cookie({ 'name': 'key-aaaaaaa', 'value': 'value-bbbbbb'})
    # 遍历cookies中的name 和value信息并打印，当然还有上面添加的信息
    for cookie in driver.get_cookies():
        print("%s -> %s" % (cookie['name'], cookie['value']))
    driver.quit()

**7.窗口截图**  
自动化用例是由程序去执行的，因此有时候打印的错误信息并不十分明确。如果在脚本执行出错的时候能对当前窗口截图保存，那么通过图片就可以非常直观地看出出错的原因。  
WebDriver提供了截图函数get\_screenshot\_as\_file()来截取当前窗口。

from selenium import webdriver
    from time import sleep
    
    driver = webdriver.Firefox()
    driver.get('http://www.baidu.com')
    
    driver.find_element_by_id('kw').send_keys('selenium')
    driver.find_element_by_id('su').click()
    sleep(2)
    
    # 截取当前窗口，并指定截图图片的保存位置
    driver.get_screenshot_as_file("D:\\baidu_img.jpg")
    
    driver.quit()

[python-selenium_1]: https://blog.csdn.net/liulanba/article/details/115294919
[Link 1]: https://blog.csdn.net/liulanba/article/details/115213578