webdriver高级应用- 无人工干预地自动下载某个文件

在网页上下载文件时，通常需要人为设定下载文件并选择保持路径，这样就无法实现完全自动的下载过程。下面实现基于firefox浏览器的全自动化文件下载操作：

#encoding=utf-8
from selenium import webdriver
import unittest, time

class TestDemo(unittest.TestCase):

    def setUp(self):
        # 创建一个FirefoxProfile实例，用于存放自定义配置
        profile = webdriver.FirefoxProfile() #chrome是另外的设置方法
        # 指定下载路径，默认只会自动创建一级目录，如果指定了
        # 多级不存在的目录，将会下载到默认路径
        profile.set_preference('browser.download.dir', 'd:\\iDownload')
        # 将browser.download.folderList设置为2，表示将文件下载到指定路径
        # 设置成2表示使用自定义下载路径；
        # 设置成0表示下载到桌面；设置成1表示下载到默认路径
        profile.set_preference('browser.download.folderList', 2)
        # browser.helperApps.alwaysAsk.force对于未知的 MIME 类型文件会弹出窗口
        # 让用户处理，默认值为true，设定为False表示不会记录打开未知 MIME 类型
        # 文件的方式
        profile.set_preference("browser.helperApps.alwaysAsk.force", False)
        # 在开始下载时是否显示下载管理器
        profile.set_preference('browser.download.manager.showWhenStarting',\
                               False)
        # 设定为 False 会把下载框进行隐藏
        profile.set_preference("browser.download.manager.useWindow", False)
        # 默认值为 true，设定为 False 表示不获取焦点
        profile.set_preference("browser.download.manager. focusWhenStarting",\
                               False)
        # 下载.exe文件弹出警告，默认值是 true，设定为False 则不会弹出警告框
        profile.set_preference("browser.download.manager.alertOnEXEOpen",\
                               False)
        # browser.helperApps.neverAsk.openFile表示直接打开下载文件，不显示确认框
        # 默认值为空字符串，下行代码行设定了多种文件的 MIME类型，
        # 例如application/exe，表示.exe类型的文件，
        # application/excel表示 Excel 类型的文件
        profile.set_preference("browser.helperApps.neverAsk.openFile", \
                               "application/pdf")
        # 对所给出文件类型不再弹出框进行询问，直接保存到本地磁盘
        profile.set_preference('browser.helperApps.neverAsk.saveToDisk', \
                               'application/zip, application/octet-stream')
        # browser.download.manager.showAlertOnComplete设定下载文件结束后是否显示下
        #载完成提示框，默认为true，设定为False表示下载完成后不显示下载完成提示框
        profile.set_preference("browser.download.manager. showAlertOnComplete",\
                               False);
        # browser.download.manager.closeWhenDone设定下载结束后是否自动
        # 关闭下载框，默认值为true，设定为False 表示不关闭下载管理器
        profile.set_preference("browser.download.manager.closeWhenDone",\
                               False)

        # self.driver = webdriver.Ie(executable_path="c:\\IEDriverServer")
        # 启动浏览器时，通过firefox_profile参数
        # 将自动以配置添加到FirefoxProfile对象中
        self.driver = webdriver.Firefox(executable_path="e:\\geckodriver",\
                                        firefox_profile = profile)

    def test_dataPicker(self):
        # 访问WebDriver驱动Firefox的驱动文件下载网址
        url1 = "https://github.com/mozilla/geckodriver/releases"
        self.driver.get(url1)
        # 选择下载zip类型文件，使用application/zip指代此类型文件
        self.driver.find_element_by_xpath\
            ('//strong[.="geckodriver-v0.19.1-arm7hf.tar.gz"]').click()
        # 等待加载下载文件
        time.sleep(10)

        # 访问Python2.7.12文件下载页面，下载扩展名为msi文件
        # 使用application/octet-stream来指明此类文件类型
        url = "https://www.python.org/downloads/release/python-2712/"
        self.driver.get(url)
        # 找到Python2.7.12下载页面中链接文字为“Windows x86-64 MSI installer”
        # 的链接页面元素，点击进行无人工干预的下载Python2.7.12解释器文件
        self.driver.find_element_by_link_text\
            ("Windows x86-64 MSI installer").click()
        # 等待文件下载完成，根据各自的网络带宽情况设定等待相应的时间
        time.sleep(100)

    def tearDown(self):
        self.driver.quit()

if __name__ == '__main__':
    unittest.main()