- 我们在做UI自动化测试的过程中,某些情况会遇到,需要操作WebElement属性的情况。
假设现在我们需要获取一个元素的title属性,我们可以先找到这个元素,然后利用get_attribute方法获取属性的值。
举个栗子:
from selenium import webdriver
driver = webdriver.Chrome()
driver.get("http://www.baidu.com")
search_button = driver.find_element_by_id("su") # 百度搜索按钮
# 现在我们获取百度一下的值
value = search_button.get_attribute("value") # 获取input标签的value,也就是百度一下那4个字
print(value) # 打印 百度一下
但是现在我们有了新的需求,我们需要改变百度一下这个按钮里边显示的值。
先说一下原理,原理是利用js的dom(document object model),也就是文档对象模型,获取到input标签, 然后通过js来改变这个input标签的value属性。
js如下:
var button = document.getElementById("su");
button.setAttribute("su", "你猜一下");
//或者直接给value属性赋值
document.getElementById("su").value = "你猜一下";
我们在Chrome DevelopmentTools里边可以看到,“百度一下”变成了“你猜一下”~
那么为什么我们不直接用driver.execute_script()这个方法来执行上述js语句呢,但是要知道,因为dom里获取元素的方式有限,并不如selenium那么方便,什么link_text这类的api都是无法使用的。
昨天偶然发现2个问题,第一是execute_script函数是可以传脚本参数进去的,第二个是selenium抓取到的元素可以作为js的dom元素处理。有了这2点之后呢,就可以干活了!
现在用WebElement的方法做到同样的事情
from selenium import webdriver
driver = webdriver.Chrome()
driver.get("http://www.baidu.com")
search_button = driver.find_element_by_id("su") # 百度搜索按钮
# arguments[0]对应的是第一个参数,可以理解为python里的%s传参,与之类似
driver.execute_script("arguments[0].value = '你猜一下';", search_button)
补充一点, 如果需要获取js语句执行后的返回值,在js语句前加"return" 就行了,例如:
button_value = driver.execute_script("return arguments[0].value;", search_button)