I'm using Selenium2 for some automated tests of my website, and I'd like to be able to get the return value of some Javascript code. If I have a foobar() Javascript function in my webpage and I want to call that and get the return value into my Python code, what can I call to do that?
Add a comment
|
2 Answers
To return a value, simply use the return JavaScript keyword in the string passed to the execute_script() method, e.g.
>>> 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'
4 Comments
Alex
if the variable is not defined by javascript, what would be the return value?Does it throw an exception or simply an empty string ?
dbJones
if variable is not defined, it returns
None
ntk4
quick note for those newbs,
return_value = wd.execute_script("return {foo: 'bar'}") would store the returned value to be used later in your program.
You can return values even if you don't have your snippet of code written as a function like in the below example code, by just adding return var; at the end where var is the variable you want to return.
result = driver.execute_script('''
cells = document.querySelectorAll('a');
URLs = [];
[].forEach.call(cells, function (el) {
URLs.push(el.href)
});
return URLs
''')
result will contain the array that is in URLs this case.
answered Dec 28, 2017 at 12:46
Eduard Florinescu
17.7k30 gold badges123 silver badges189 bronze badges