selenium 自動下載文件

#coding=utf-8
from selenium import webdriver


#實例化一個火狐配置文件
fp = webdriver.FirefoxProfile()


#設置各項參數，參數可以通過在瀏覽器地址欄中輸入about:config查看。

#設置成0代表下載到瀏覽器默認下載路徑；設置成2則可以保存到指定目錄
fp.set_preference("browser.download.folderList",2)


#是否顯示開始,(個人實驗，不管設成True還是False，都不顯示開始，直接下載)
fp.set_preference("browser.download.manager.showWhenStarting",False)


#下載到指定目錄
fp.set_preference("browser.download.dir","c:\\test")


#不詢問下載路徑；后面的參數為要下載頁面的Content-type的值
fp.set_preference("browser.helperApps.neverAsk.saveToDisk","application/octet-stream")


#啟動一個火狐瀏覽器進程，以剛才的瀏覽器參數
dr = webdriver.Firefox(firefox_profile=fp)


#打開下載頁面
dr.get("https://pypi.python.org/pypi/selenium")


#點擊某個按鈕
dr.find_element_by_xpath("//div[@id='download-button']/a").click()


#如果是某個直接下載鏈接，則缺少了點擊某個鏈接或者按鈕的操作，打開頁面后執行下載動作


#備注：查看Content-type的方法，curl
#curl是采用url命令，在命令行下工作的文件傳輸開源工具
#首先，安裝，到官網下載對應的版本，解壓出來后是一個curl.exe文件。打開cmd，進入該目錄下，則可以使用curl命令了。
#也可以將curl.exe的上層目錄添加到path目錄下去，則在任何地方都可以使用curl命令。
#如查看百度的Content-type ，進入cmd頁面：C:\Users\Administrator>curl -I   http://www.baidu.com
HTTP/1.1 200 OK
Date: Tue, 15 Apr 2014 10:08:24 GMT
Content-Type: text/html
Connection: Keep-Alive
Vary: Accept-Encoding
Set-Cookie: BAIDUID=5A06884F893AB430B9A251E59C27C50F:FG=1;
7 23:55:55 GMT; max-age=2147483647; path=/; domain=.baidu.c
Set-Cookie: BDSVRTM=0; path=/
Set-Cookie: H_PS_PSSID=5552_4578_5231_1455_6052_5223_4760_6
; domain=.baidu.com