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()