Selenium+Python：下載文件(Firefox 和 Chrome)

引自  https://blog.csdn.net/Momorrine/article/details/79794146

1.      環境

操作系統	Win10
IDE	Eclipse (Oxygen 4.7)+ PyDev 5.9.2 (JDK1.8)
Python	3.5
Selenium	selenium-3.9.0-py2.py3-none-any.whl
FirefoxDriver	0.20.0
Firefox瀏覽器	59.0.2(32位)
ChromeDriver	2.34
Chrome瀏覽器	63.0.3239.84

  

2.      Firefox

    這是蟲師書里面的一個例子，我直接copy下來了

1. from selenium import webdriver
2.  
3. fp = webdriver.FirefoxProfile()
4. fp.set_preference( "browser.download.folderList", 2)
5. fp.set_preference( "browser.download.manager.showWhenStarting", False) # 不起作用
6. fp.set_preference( "browser.download.dir", os.getcwd())
7. fp.set_preference( "browser.helperApps.neverAsk.saveToDisk", "application/octet-stream")
8.  
9. driver = webdriver.Firefox(firefox_profile=fp)
10. driver.get( "http://pypi.Python.org/pypi/selenium")
11. driver.find_element_by_partial_link_text( "selenium-3.11.0-py2.py3-none-any").click()

    但是很不幸，我這兒運行了，還是會彈出確認下載對話框，雖然設置了browser.download.manager.showWhenStarting為Flase，而由於Selenium無法操作該對話框，程序就卡在那兒了沒能下載文件。

 

    我發現確認下載對話框，默認的焦點就在【確定】上，就試着在代碼的最后添加這樣一句：

ActionChains(driver).key_down(Keys.ENTER).perform()

當這個對話框出現的時候，手動鍵入ENTER是可以下載的，但這句代碼並沒有起作用，仍然沒有下載文件。

 

3.      Chrome

    又試着增加了一些等待時間，始終沒成功，就想着先換一個瀏覽器試試吧。

1. options = webdriver.ChromeOptions()
2. prefs = { 'profile.default_content_settings.popups': 0, 'download.default_directory': os.getcwd()}
3. options.add_experimental_option( 'prefs', prefs)
4.  
5. driver = webdriver.Chrome(chrome_options=options)
6. driver.get( "http://pypi.Python.org/pypi/selenium")driver.find_element_by_partial_link_text("selenium-3.11.0-py2.py3-none-any").click()

    對Chrome瀏覽器設置了profile.default_content_settings.popups為0是起作用的，運行這段代碼，並沒有彈出確認提示框，文件可以下載下來。

  

4.      Firefox+Pywin32

總覺得Firefox應該也會有辦法來操作那個確認下載對話框，百度了一下，有提到Pywin32，可以獲取並操作Windows窗口。那就先安裝一個吧。 

    Pywin32，我是直接在命令行切換到Python根目錄\Scripts下，用pip install pywn32來安裝的：

                          

     此外還用到一個可以獲取Windows窗口屬性的小工具Sky++，下載了解壓縮就可以用了。運行Sky++后，點擊【搜索】-【窗口搜索】菜單項打開【窗口搜索】對話框

                                            

按住其中的查找程序工具拖動到確認下載對話框后釋放，【窗口搜索】對話框中會顯示該窗口的句柄、標題、類等屬性

                                              

                                            

通過后兩個屬性，就可以調用win32gui.FindWindow()獲取該窗口的句柄，進而試着操作該窗口。注意 ：雖然在Sky++中已經得到了句柄，但程序中不能直接用，因為每次運行窗口句柄是不同的。

     於是我在之前的代碼后面增加了下面這句代碼

dialog = win32gui.FindWindow("MozillaDialogClass", u"正在打開 selenium-3.11.0-py2.py3-none-any.whl")

打印了一下dialog，是有值的，就說明成功地獲取到了窗口句柄了。 

    在獲取到對話框的句柄后，就應該進一步獲取【確定】按鈕的句柄，然后點擊該按鈕，這樣就應該可以下載文件了

1. button = win32gui.FindWindowEx(dialog, None, "Button",None)
2. win32gui.PostMessage(button, win32con.WM_LBUTTONDOWN, win32con.MK_LBUTTON, 0)
3. win32gui.PostMessage(button, win32con.WM_LBUTTONUP, win32con.MK_LBUTTON, 0)

可是一直沒成功，打印button一直是0，沒有能獲取到按鈕的句柄，我懷疑是因為類名不正確的原因，但Sky++上無法看到按鈕的信息，將類名換了又換，或許是"MozillaButtonClass"，但始終沒有獲取。

    轉而我又想，既然對話框獲取到了，那對對話框鍵入ENTER呢，或許就可以了

win32gui.SendMessage(dialog,win32con.WM_KEYDOWN, win32con.VK_RETURN, 0)

嗯，真的可以了。因此用Firefox下載文件的完整代碼是

1. from selenium import webdriver
2. import win32gui, win32con
3.  
4. fp = webdriver.FirefoxProfile()
5. fp.set_preference( "browser.download.folderList", 2)
6. fp.set_preference( "browser.download.dir", os.getcwd())
7. fp.set_preference( "browser.helperApps.neverAsk.saveToDisk", "application/octet-stream")
8.  
9. driver = webdriver.Firefox(firefox_profile=fp)
10. driver.get( "http://pypi.Python.org/pypi/selenium")
11. driver.find_element_by_partial_link_text( "selenium-3.11.0-py2.py3-none-any").click()
12.  
13. dialog = win32gui.FindWindow( "MozillaDialogClass", u"正在打開 selenium-3.11.0-py2.py3-none-any.whl")
14. win32gui.SendMessage(dialog,win32con.WM_KEYDOWN, win32con.VK_RETURN, 0)

    只是有一個問題，在import win32gui和win32con時，會有錯誤標識

但運行沒問題，不知道為什么。。。。。。 

    還要注意的是，需要根據當前網絡的狀況，以及下載文件的大小，適當添加一些等待時間，不然會可能出現最后關閉瀏覽器驅動時文件還沒下完的情況。

5.      再次Firefox

    以上那些是昨天寫的，是到昨天晚上為止的認識。今兒早上來了，在搜另外一個問題的時候，發現有博文說Firefox不能下載的原因，很可能是因為"browser.helperApps.neverAsk.saveToDisk"設置的文件類型不對的緣故，我把那個例子里面的類型"binary/octet-stream"貼過來試了一下，果然是可以的。但關於browser.download.manager.showWhenStarting設置為Flase不彈出提示框，其實代碼中把文件類型設置正確了，沒有設置browser.download.manager.showWhenStarting，也並沒有彈出提示框。

1. from selenium import webdriver
2.  
3. fp = webdriver.FirefoxProfile()
4. fp.set_preference( "browser.download.folderList", 2)
5. fp.set_preference( "browser.download.dir", os.getcwd())
6. fp.set_preference( "browser.helperApps.neverAsk.saveToDisk", "binary/octet-stream")
7.  
8. driver = webdriver.Firefox(firefox_profile=fp)
9. driver.get( "http://pypi.Python.org/pypi/selenium")
10. driver.find_element_by_partial_link_text( "selenium-3.11.0-py2.py3-none-any").click()

 

    有朋友友情提示，該加個參考資料小節。以前寫東西的時候倒是每次都有注意引用這個問題，而最近寫的也只是記錄自己的一個學習Selenium+Python Web自動化測試的過程。不過好習慣還是該保持，這篇已經不太記得自己到底搜了哪些文章了，就記下幾篇還記得的吧。

 

6.      參考資料

[1]  《Selenium2自動化測試實戰：基於Python語言》蟲師編著 電子工業出版社 2016年1月

[2] 用selenium的webdriver下載文件（基於python，firefox和chrome） https://blog.csdn.net/cyjs1988/article/details/74988997

[3]   Selenium 設置瀏覽器下載  http://www.cnblogs.com/fnng/p/7700620.html