Question

我正在使用Selenium2对我的网站进行一些自动化测试，我希望能够获得一些Javascript代码的返回值。如果我在我的网页上有一个foobar() Javascript函数，并且我想调用它并将返回值放入我的Python代码中，我还可以调用它来做什么？

Answer 1

要返回值，只需在传递给return方法的字符串中使用execute_script() JavaScript关键字，例如

>>> from selenium import webdriver
>>> wd = webdriver.Firefox()
>>> wd.get("http://localhost/foo/bar")
>>> wd.execute_script("return 5")
5
>>> wd.execute_script("return true")
True
>>> wd.execute_script("return {foo: 'bar'}")
{u'foo': u'bar'}
>>> wd.execute_script("return foobar()")
u'eli'

Answer 2

即使你没有像以下示例代码中那样编写代码的剪切代码，也可以返回值，只需在末尾添加return var;，其中var是要返回的变量。

result = driver.execute_script('''cells = document.querySelectorAll('a');
URLs = []
console.log(cells);
[].forEach.call(cells, function (el) {
    if(el.text.indexOf("download") !== -1){
    //el.click();
    console.log(el.href)
    //window.open(el.href, '_blank');
    URLs.push(el.href)
    }
});
return URLs''')

result将包含URLs这种情况下的数组。

获取Selenium中Javascript代码的返回值

2 个答案: