首页 > > 网络编程 > 其它 >

Python：Selenium 2：使用

2018-08-10 11:27:38来源：博客园阅读 ()

创建一个浏览器对象

from selenium import webdriver
browser = webdriver.Chrome()

WebDriver在将控制权返回给测试脚本之前，会一直等待到页面完全加载完毕，但如果页面使用了很多AJAX，WebDriver可能无法准确判断页面是否完全加载。

打开页面

browser.get("https://www.baidu.com")

不能省略http:// 或 https://

获取页面源码

html = browser.page_source

定位HTML元素

定位一个元素	含义
find_element_by_tag_name	通过标签定位
find_element_by_id	通过元素id定位
find_element_by_name	通过元素name定位
find_element_by_class_name	通过类名进行定位
find_element_by_css_selector	通过css选择器进行定位
find_element_by_xpath	通过xpath表达式定位（xpath和xml中的类似）
find_element_by_link_text	通过完整超链接文本定位（不是通过超链接的地址，而是超链接上面的文字描述）
find_element_by_partial_link_text	通过部分链接定位

如果没有发现任何元素，将会抛出NoSuchElementException异常。

以上的方法仅返回第一个满足条件的元素，如果要返回多个元素，需要使用find_elements_by_xx。

输入文本

element.send_keys("selenium")

你输入的字符将会被添加在已有文本之后。如果传入多个文本，将依次添加。

在输入框中输入的字符不会自动清空，你可以使用clear方法输入区域的内容。

element.Clear()

快捷键

from selenium.webdriver.common.keys import Keys
element.send_keys(Keys.BACKSPACE)    # 删除键
element.send_keys(keys.ENTER)        # 回车键
element.send_keys(Keys.SPACE)        # 空格键
element.send_keys(Keys.CONTROL, 'a')    # Ctrl+A
element.send_keys(Keys.CONTROL, 'c')    # Ctrl+C

点击按钮

element.click()

定位下拉框（select）

方法一：先定位select，再定位option选项。

browser.find_element_by_id("sel").find_element_by_xpath("//option[@value='晴']").click()

方法二：直接xpath定位。

browser.find_element_by_xpath("//*[@id='sel']/option[3]").click()

方法三：select模块定位

from selenium.webdriver.support.select import Select
element = browser.find_element_by_id("sel")
Select(element).select_by_index(2).click() # index 从0开始
Select(element).select_by_value('晴')
Select(element).select_by_visible_text('雨')

用Select定位后会直接选中不需要click()方法选中，也没有click()方法。

Select中的其他方法如下：

deselect_all() # 取消所有选项
deselect_by_index() # 取消对应 index 选项
deselect_by_value() # 取消对应 value 选项
deselect_by_visible_text() # 取消对应文本选项

以上方法需要select支持多选（multiple="true"）

单选框（radio）复选框（checkbox）

element = browser.find_element_by_id('yes')
element.click()

就是定位HTML元素，没啥花样。

判断是否选中

browser.find_element_by_id('yes').is_selected()

判断是否可见

有些元素设置了display:none，这时候如果直接操作这种隐藏的元素，会报错：element not visible。

element.is_displayed()

定位弹框

1、alert弹框

alertElement = browser.switch_to_alert()
print(alertElement.text) # 获取对话框文本值
alertElement.accept() # 点击确认按钮

在现在的版本中，browser.switch_to_alert()已经过时了，

改成：alertElement = browser.switch_to.alert

2、confirm弹框

confirm 和 alert 是一样的，只是多了个点击取消按钮的方法。

confirmElement = browser.switch_to.alert
print(confirmElement.text) # 获取对话框文本值
confirmElement.dismiss() # 点击取消按钮

3、prompt弹框

prompt 又多了个输入框，send_keys 方法即可。

执行JS

js = "$('#sp').toggle();"
browser.execute_script(js)

退出

browser.quit()

示例：

为了测试，我们新建一个htnl网页

<html>
<head>
<title>SeleniumTest</title>
<script src="Scripts/jquery-1.10.2.min.js" type="text/javascript"></script>

<style type="text/css">
tr{
height: 40px}
</style>
</head>
<body>
<table style="margin: 200px auto;">
<tr>
<td><input type="text" /></td>
</tr>
<tr>
<td><input type="text" id="txt1" /></td>
</tr>
<tr>
<td><input type="text" id="txt2" name="txt2" /></td>
</tr>
<tr>
<td><input type="text" class="c1" /></td>
</tr>
<tr>
<td><input type="text" class="c2" /></td>
</tr>
<tr>
<td><input type="text" id="txt6" /></td>
</tr>
<tr>
<td><span id="sp" style="display:none">隐藏的标签</span></td>
</tr>
<tr>
<td>
<a href="javascript:;">aaa</a>
<a href="https://www.baidu.com" target="_blank">bbb</a>
<a href="https://baike.baidu.com/" target="_blank">ccc</a>
</td>
</tr>
<tr>
<td>
<select id="sel">
<option value="无"></option>
<option value="晴">晴</option>
<option value="阴">阴</option>
<option value="雨">雨</option>
<option value="多云">多云</option>
</select>
</td>
</tr>
<tr>
<td>
<input type="radio" id="yes" />是
<input type="radio" id="no" />否
</td>
</tr>
<tr>
<td>
<input type="checkbox" id="chk1" name="chk" />语文
<input type="checkbox" id="chk2" name="chk" />数学
<input type="checkbox" id="chk3" name="chk" />英语
</td>
</tr>
<tr>
<td>
<input type="button" id="btn1" value="警告消息框" onclick="alert('警告警告')" />
<input type="button" id="btn2" value="确认消息框" onclick="confirm('是否确认')" />
<input type="button" id="btn3" value="提示消息框" onclick="prompt('提示注意')" />
<input type="button" id="btn4" value="显示" onclick="$('#sp').toggle();" />
</td>
</tr>
</table>
</body>
</html>

Python代码：

from selenium import webdriver
from selenium.webdriver.support.select import Select
from selenium.webdriver.common.keys import Keys
import time

browser = webdriver.Chrome() # 创建浏览器对象

browser.get("http://localhost:8001/") # 打开网址
time.sleep(2)

# 定位html元素
txt1 = browser.find_element_by_tag_name('input')
txt1.send_keys('tag_name=input') # 输入文本

txt2 = browser.find_element_by_id('txt1')
txt2.send_keys('id=txt1')

txt3 = browser.find_element_by_name('txt2')
txt3.send_keys('name=txt2')

txt4 = browser.find_element_by_class_name('c1')
txt4.send_keys('class_name=c1')

txt5 = browser.find_element_by_css_selector('.c2')
txt5.send_keys('css=.c2')

txt6 = browser.find_element_by_xpath("//input[@id='txt6']")
txt6.send_keys("xpath=//input[@id='txt6']")

a1 = browser.find_element_by_link_text('bbb')
# a1.click()

txt1.clear() # 清空文本
time.sleep(2)

# 快捷键
txt2.send_keys(Keys.CONTROL, 'a')
time.sleep(1)
txt2.send_keys(Keys.CONTROL, 'c')
time.sleep(1)
txt2.send_keys(Keys.BACKSPACE)
time.sleep(1)
txt2.send_keys(Keys.CONTROL, 'v')
time.sleep(2)

# 定位下拉框
browser.find_element_by_id("sel")\
.find_element_by_xpath("//option[@value='晴']").click()
time.sleep(1)
browser.find_element_by_xpath("//*[@id='sel']/option[3]").click()

# 用Select模块定位
element = browser.find_element_by_id("sel")
Select(element).select_by_index(2) # index 从0开始
time.sleep(1)
Select(element).select_by_value('晴')
time.sleep(1)
Select(element).select_by_visible_text('雨')

# 单选框
element = browser.find_element_by_id('yes')
element.click()
print(element.is_selected())

# 复选框
browser.find_element_by_id('chk1').click()

# 判断元素是否可见
print(browser.find_element_by_id('sp').is_displayed())

# alert弹框
browser.find_element_by_id('btn1').click()
alertElement = browser.switch_to.alert
print(alertElement.text) # 获取对话框文本值
alertElement.accept() # 点击确认按钮

# confirm弹框
browser.find_element_by_id('btn2').click()
confirmElement = browser.switch_to.alert
print(confirmElement.text) # 获取对话框文本值
confirmElement.dismiss() # 点击取消按钮

# 执行JS
js = "$('#sp').toggle();"
browser.execute_script(js)

# 退出
browser.quit()