webdriver高级应用- 使用Chrome浏览器自动将文件下载到指定路径

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

class TestDemo(unittest.TestCase):

    def setUp(self):
        # 创建Chrome浏览器配置对象实例
        chromeOptions = webdriver.ChromeOptions()
        # 设定下载文件的保存目录为C盘的iDownload目录，
        # 如果该目录不存在，将会自动创建
        prefs = {"download.default_directory": "d:\\iDownload"}
        # 将自定义设置添加到Chrome配置对象实例中
        chromeOptions.add_experimental_option("prefs", prefs)
        # 启动带有自定义设置的Chrome浏览器
        self.driver = webdriver.Chrome(executable_path="e:\\chromedriver",\
                                       chrome_options=chromeOptions)

    def test_downloadFileByChrome(self):
        url = "http://pypi.python.org/pypi/selenium"
        # 访问将要下载文件的网址
        self.driver.get(url)
        # 找到要下载的文件链接页面元素，并点击进行下载
        self.driver.find_element_by_partial_link_text\
            ("selenium-3.8.1.tar.gz").click()
        # 等待100s，以便文件下载完成
        time.sleep(20)

    def tearDown(self):
        # 退出IE浏览器
        self.driver.quit()

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