Selenium API（二）

1.定位一组元素

WebDriver提供了8种定位一组元素的方法。

driver.find_elements_by_css_selector()
driver.find_elements_by_tag_name()
driver.find_elements_by_class_name()
driver.find_elements_by_xpath()
driver.find_elements_by_id()
driver.find_elements_by_link_text()
driver.find_elements_by_name()
driver.find_elements_by_partial_link_text()

# -*- coding:utf-8 -*-
from selenium import webdriver
import os
import time
driver = webdriver.Firefox()
file_path = "file:///" + os.path.abspath('testcheck.html')
driver.get(file_path)
inputs = driver.find_elements_by_tag_name("input")
for input in inputs:
    if input.get_attribute("type") == "checkbox":
        input.click()
        time.sleep(1)
driver.quit()

2.多表单切换

WebDriver只能在一个页面上对元素进行识别和定位，对于frame/iframe表单内嵌页面上的元素无法直接定位，因此WebDriver提供了方法将当前定位的主体切换为frame/iframe表单的内嵌页面中。

switch_to.frame()，默认可以直接取表单的id或者name属性，如果iframe没有可用的id和name属性，可以通过下面的方式定位：

...
# 先通过css定位到iframe
iframe = driver.find_element_by_css_selector("[class='iframe']")
# 再将定位对象传给switch_to.frame()方法
driver.switch_to.frame(iframe)
...
driver.switch_to.parent_frame()

说明：当完成了当前iframe操作后，可以通过driver.switch_to.parent_frame()跳出当前一级表单，该方法默认对应于离他最近的switch_to.frame()方法，此外，在进入多级表单的情况下，还可以通过switch_to.default_content()跳会到最外层的页面。

 3.多页面切换

WebDriver提供了switch_to.window()方法实现在不同的窗口之间切换。

# -*- coding:utf-8 -*-
from selenium import webdriver
import os
import time
driver = webdriver.Firefox()
driver.implicitly_wait(5)
driver.get("http://www.baidu.com")
# 获得百度搜索窗口句柄
search_handle = driver.current_window_handle

driver.find_element_by_link_text("登录").click()
driver.find_element_by_link_text("立即注册").click()
time.sleep(2)
# 获取当前所有打开的窗口的句柄
all_handles = driver.window_handles
for handle in all_handles:
    if handle != search_handle:
        # 如果句柄不为搜索窗口的句柄，则切换到注册页面
        driver.switch_to.window(handle)
        print("now register window!")
        time.sleep(2)

for handle in all_handles:
    if handle == search_handle:
        driver.switch_to.window(handle)
        driver.find_element_by_id("TANGRAM__PSP_4__closeBtn").click()
        driver.find_element_by_id("kw").send_keys("selenium")
        driver.find_element_by_id("su").click()
        time.sleep(2)



driver.quit()

4.警告框处理

在WebDriver中通过使用switch_to_alert()方法定位到alert/confirm/prompt，然后使用text/accept/dismiss/send_keys等方法进行操作。

text：返回alert/confirm/prompt中的文字信息。

accept()：接受现有警告框。

dismiss()：解散现有警告框。

send_keys(keysToSend)：发送文本到警告框。keysToSend：将文本发送至警告框。

5.send_keys()实现文件上传

# 定位上传按钮，然后添加本地文件
driver.find_element_by_name("file").send_keys("f:\\upload_file.txt")

如果想要实现文件上传，只需要找到上传的input标签，基本伤可以同send_keys()方法将文件上传。

6.文件下载

# -*- coding:utf-8 -*-
from selenium import webdriver
import os
import time
firefox_profile = webdriver.FirefoxProfile()
# 设置为0代表下载到浏览器默认下载路径，设置为2则表示可以保存到执行目录中
firefox_profile.set_preference("browser.download.folderList", 2)
# 是否显示开始，True为显示，False为不显示
firefox_profile.set_preference("browser.download.manager.showWhenStarting", False)
# 执行下载的目录，os.getcwd()返回当前目录
firefox_profile.set_preference("browser.download.dir", os.getcwd())
# 设置下载文件的类型
firefox_profile.set_preference("browser.helperApps.neverAsk.saveToDisk", "application/octet-stream")

driver = webdriver.Firefox(firefox_profile=firefox_profile)
driver.get("xxx")
driver.find_element_by_name("xxx").click()
driver.quit()

 7.操作Cookie

WebDriver提供了操作Cookie的方法，可以读取，添加和删除cookie信息：

• get_cookies()：获得所有cookie信息。
• get_cookie()：返回字典的key为“name”的cookie信息。
• add_cookie(cookie_dict)：添加cookie。cookie_dict为字典对象，必须有name和value。
• delete_cookie()：删除cookie信息，需要指定name，即要删除的cookie的名称。
• delete_all_cookies()：删除所有的cookie信息。

8.执行JavaScript

WebDriver提供了execute_script()方法来执行JavaScript代码。

# -*- coding:utf-8 -*-
from selenium import webdriver
import os
import time

driver = webdriver.Firefox()
# 设置浏览器窗口大小
driver.set_window_size(700, 500)
driver.get("http://www.baidu.com")
driver.find_element_by_id("kw").send_keys("selenium")
driver.find_element_by_id("su").click()
time.sleep(3)
js = "window.scrollTo(100, 400)"
# 执行js脚本，设置窗口到滚动条的位置
driver.execute_script(js)
time.sleep(3)
driver.quit()

9.窗口截图

WebDriver提供了get_screenshot_as_file()来截取当前窗口。

# -*- coding:utf-8 -*-
from selenium import webdriver
import os
import time

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()
time.sleep(3)
# 截取当前窗口，并制定截图保存的位置
driver.get_screenshot_as_file("d:\\test.jpg")
driver.quit()

10.关闭浏览器窗口

WebDriver提供了quit()方法，其含义为退出相关的驱动程序和关闭所有窗口，另外提供了close()方法，用来关闭某一个窗口。