รับค่าส่งคืนของโค้ด Javascript ในซีลีเนียม

ฉันใช้ Selenium2 สำหรับการทดสอบอัตโนมัติบางอย่างในเว็บไซต์ของฉันและฉันต้องการรับค่าส่งคืนของโค้ด Javascript บางส่วน หากฉันมีfoobar()ฟังก์ชัน Javascript ในหน้าเว็บของฉันและฉันต้องการเรียกสิ่งนั้นและรับค่าส่งคืนเป็นรหัส Python ของฉันฉันจะเรียกให้ทำอะไรได้บ้าง?

ในการส่งคืนค่าเพียงใช้returnคีย์เวิร์ด JavaScript ในสตริงที่ส่งไปยังexecute_script()เมธอดเช่น

>>> 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'

คุณสามารถส่งคืนค่าได้แม้ว่าคุณจะไม่ได้เขียนโค้ดไว้เป็นฟังก์ชันเหมือนในโค้ดตัวอย่างด้านล่างเพียงแค่เพิ่มreturn var;ที่ส่วนท้ายโดยที่ var เป็นตัวแปรที่คุณต้องการส่งคืน

result = driver.execute_script('''
cells = document.querySelectorAll('a');
URLs = [];
[].forEach.call(cells, function (el) {
    URLs.push(el.href)
});
return URLs
''')

resultจะมีอาร์เรย์ที่อยู่ในURLsกรณีนี้