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下来了

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")
driver = webdriver.Firefox(firefox_profile=fp)
driver.get( "http://pypi.Python.org/pypi/selenium")
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

又试着增加了一些等待时间，始终没成功，就想着先换一个浏览器试试吧。

options = webdriver.ChromeOptions()
prefs = { 'profile.default_content_settings.popups': 0, 'download.default_directory': os.getcwd()}
options.add_experimental_option( 'prefs', prefs)
driver = webdriver.Chrome(chrome_options=options)
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，是有值的，就说明成功地获取到了窗口句柄了。

在获取到对话框的句柄后，就应该进一步获取【确定】按钮的句柄，然后点击该按钮，这样就应该可以下载文件了

button = win32gui.FindWindowEx(dialog, None, "Button",None)
win32gui.PostMessage(button, win32con.WM_LBUTTONDOWN, win32con.MK_LBUTTON, 0)
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下载文件的完整代码是

from selenium import webdriver
import win32gui, win32con
fp = webdriver.FirefoxProfile()
fp.set_preference( "browser.download.folderList", 2)
fp.set_preference( "browser.download.dir", os.getcwd())
fp.set_preference( "browser.helperApps.neverAsk.saveToDisk", "application/octet-stream")
driver = webdriver.Firefox(firefox_profile=fp)
driver.get( "http://pypi.Python.org/pypi/selenium")
driver.find_element_by_partial_link_text( "selenium-3.11.0-py2.py3-none-any").click()
dialog = win32gui.FindWindow( "MozillaDialogClass", u"正在打开 selenium-3.11.0-py2.py3-none-any.whl")
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，也并没有弹出提示框。

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