引自 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()
有朋友友情提示,该加个参考资料小节。以前写东西的时候倒是每次都有注意引用这个问题,而最近写的也只是记录自己的一个学习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