使用Selenium通过PhantomJS中的超链接下载文件

Question

我正在使用硒在超链接上执行单击功能，该超链接加载到某个页面上。该脚本适用于谷歌浏览器，但不适用于phantomjs。为什么这不起作用？

from selenium import webdriver
driver = webdriver.Chrome()   
#driver = webdriver.PhantomJS(executable_path = "/Users/jameslemieux/PythonProjects/phantomjs-1.9.8-macosx/bin/phantomjs")
driver.get("http://www.youtube-mp3.org/?e=t_exp&r=true#v=hC-T0rC6m7I")
elem = driver.find_element_by_link_text('Download')
elem.click()

driver.save_screenshot('/Users/jameslemieux/Desktop/Misc./test_image.png')
driver.quit()

这在 chrome 中有效，但它总是会打开一个新的 chrome 窗口来完成任务。我读到我应该使用 phantomjs 让它在幕后运行，但是当我将驱动程序切换到 phantomjs 时，下载似乎没有完成。屏幕截图抓住了，它确实在正确的页面上，"下载"肯定在那里。所以

elem.click()

没有做它应该做的事情，或者它正在点击，但 phantomjs 不知道如何处理直接下载链接。请帮忙，我已经连续几个小时了。

Answer 1

由于PhantomJS永远不会继续下载请求，因此我们需要手动下载文件。

这里的想法是单击"转换"按钮，等待"下载"链接出现，获取 href 属性，其中包含指向生成的mp3文件的链接，然后通过 urllib.urlretrieve() 下载：

import urllib
from urlparse import urljoin
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
base_url = 'http://www.youtube-mp3.org/'
driver = webdriver.PhantomJS()
driver.get("http://www.youtube-mp3.org/?e=t_exp&r=true#v=hC-T0rC6m7I")
# convert the video to mp3
driver.find_element_by_id('submit').click()
# wait for download link to appear
element = WebDriverWait(driver, 10).until(EC.presence_of_element_located((By.LINK_TEXT, "Download")))
link = element.get_attribute('href')
url = urljoin(base_url, link)
# download the song
urllib.urlretrieve(url, 'song.mp3')
driver.quit()
# enjoy the great song