2015-05-28 84 views
4

我通过硒webdriver的Python中使用PhantomJS Javascript和我想在返回的数据块的希望在页面上执行一段JavaScript代码:执行硒/ PhantomJS

from selenium import webdriver 

driver = webdriver.PhantomJS("phantomjs.cmd") # or add to your PATH 
driver.set_window_size(1024, 768) # optional 
driver.get('http://google.com') # EXAMPLE, not actual URL 

driver.save_screenshot('screen.png') # save a screenshot to disk 
jsres = driver.execute('$("#list").DataTable().data()') 
print(jsres) 

然而当运行时,它报告KeyError。我无法找到有关可用命令的很多文档,所以我有点卡在这里。

回答

7

用于执行JavaScript的创建的方法被称为execute_script(),不​​:

driver.execute_script('return $("#list").DataTable().data();') 

FYI,​​在内部用于发送的webdriver命令。

请注意,如果您想要javascript代码返回的内容,则需要使用return

另请注意,这可能会导致Can't find variable: $错误消息。在这种情况下,找到与selenium元素并将其传递到脚本中:

# explicitly wait for the element to become present 
wait = WebDriverWait(driver, 10) 
element = wait.until(EC.presence_of_element_located((By.ID, "list"))) 

# pass the found element into the script 
jsres = driver.execute_script('return arguments[0].DataTable().data();', element) 
print(jsres)