python之selenium调用js(execute_script)

chenjianlin

在编写自动化程序的时候，会遇到弹出新窗口的情况，而程序总是默认在第一个窗口页定位元素，这样，就会定位不到元素，程序报错。
举一个实例说明，如下的网站在第一窗口页面点击“登陆”，会弹出一个新的窗口页输入用户名/密码，如果程序不做出来，在定位用户名的时候，程序会在第一个窗口页中查询，所以会报错
要解决这个问题有两种方法：

首先，弄明白为什么点击“登录”按钮会弹出一个新的窗口，下面我们来检查一下“登录”按钮的代码：
我们可以看到，这是一个HTML a 标签，a标签的 target 属性规定在何处打开链接文档。这里的target属性定义了浏览器将打开一个新的窗口
如果，不想弹出一个新窗口，就需要先删除target属性
怎么删除target？

1、第一种方法
通过执行JavaScript中的代码删除target
代码如下：
document.getElementsByClassName("site-nav-right fr")[0].childNodes[1].removeAttribute("target")
①注：在JavaScript的console中回车执行
②childNodes[1]，表示元素节点位置
③removeAttribute(“target”)，表示删除元素
完整代码：
from selenium import webdriver
driver=webdriver.Chrome()
driver.implicitly_wait(3)
driver.get("http://localhost/")
js = 'document.getElementsByClassName("site-nav-right fr")[0].childNodes[1].removeAttribute("target")'
driver.execute_script(js) #调用js方法，同时执行javascript脚本
driver.find_element_by_link_text('登录').click()
driver.find_element_by_id("username").send_keys("yonghuming")

2、第二种方法
JavaScript比较复杂，我们可不可以通过selenium定位元素来执行JavaScript的脚本，可以的

①通过selenium定位元素登录链接
login_link = driver.find_element_by_link_text(‘登录’)
②把login_link元素传入JavaScript语句，代替
document.getElementsByClassName(“site-nav-right fr”)[0].childNodes[1]，而arguments[0]可以帮我们把selenium的元素传入到JavaScript语句中
arguments指的是execute_script()方法中js代码后面的所有参数，arguments[0]表示第一个参数，argument[1]表示第二个参数
js=’arguments[0].removeAttribute(argument[1])’
driver.execute_script(js,login_link,”target”)
③点击登陆链接
login_link.click()

完整代码：
from selenium import webdriver
driver=webdriver.Chrome()
driver.implicitly_wait(3)
driver.get("http://localhost/")
login_link = driver.find_element_by_link_text('登录')
js='arguments[0].removeAttribute(argument[1])'
driver.execute_script(js,login_link,"target")
login_link.click()

梦想家

赞一个