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

小程序：扫一扫查出行
【扫一扫了解最新限行尾号】
复制小程序

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

1.      环境

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