selenium2 怎樣使用定制的 Firefox profile自動保存文件

第一步：確定你想要自己保存的文件類型。要確認你想自動下載的內(nèi)容類型，你可以使用 ?curl?:

curl -I URL|grep "Content-Type"

另外一個找出內(nèi)容類型的方法是使用 ?requests?模塊:

import requests
content_type = requests.head('http://www.python.org').headers['content-type']
print(content_type)

一旦你確定了內(nèi)容類型，你就可以設(shè)置? firefox profile?相關(guān)：?browser.helperApps.neverAsk.saveToDisk?

下面是一個實例：

import os

from selenium import webdriver

fp = webdriver.FirefoxProfile()

fp.set_preference("browser.download.folderList",2)
fp.set_preference("browser.download.manager.showWhenStarting",False)
fp.set_preference("browser.download.dir",os.getcwd())
fp.set_preference("browser.helperApps.neverAsk.saveToDisk","application/octet-stream")

browser = webdriver.Firefox(firefox_profile=fp)
browser.get("http://pypi.python.org/pypi/selenium")
browser.find_element_by_partial_link_text("selenium-2").click()

上面的例子中，我們用了?application/octet-stream?這個內(nèi)容類型，?browser.download.dir?指定了你想要保存下載文件的路徑。