selenium控制文件下载位置

我们在自动化下载文件的时候势必存在一种需求: 通过chrome将文件保存到指定位置

1. google窗口实现

​ 配置'prefs'将文件下载到指定位置, 并通过判断文件的大小来判断文件是否下载完成

import os

import time

from selenium import webdriver

from selenium.webdriver.chrome.options import Options

from selenium.webdriver.common.by import By

options = Options()

prefs = {'profile.default_content_settings.popups': 0, 'download.default_directory': os.getcwd() + '\\download'}

options.add_experimental_option("prefs", prefs)

web = webdriver.Chrome(options=options)

web.get("https://pypi.org/project/selenium/#files")

time.sleep(5)

web.find_element_by_xpath('//*[@id="files"]/div/div[2]/a[1]').click()

# 获取下载文件的路径

download_dir = os.getcwd() + '\\download'

# 注意需要和 Chrome 浏览器设置的下载路径一致

# 文件名

file_name = 'selenium-4.9.0.tar.gz'

download_file_path = os.path.join(download_dir, file_name)

# 等待文件下载完成

wait = WebDriverWait(web, timeout=30)

wait.until(lambda driver: os.path.exists(download_file_path))

# 获取文件的初始大小

init_size = os.path.getsize(download_file_path)

# 等待指定时间，再次检查文件大小是否发生变化

time.sleep(2)

while True:

    cur_size = os.path.getsize(download_file_path)

    if cur_size == init_size:

        break

    else:

        init_size = cur_size

        time.sleep(2)

# 文件大小不再发生变化，认为下载完成

web.close()

2. 基于无头浏览器实现

​ 当我们需要将selenium部署到centos服务器上运行的时候, 就会存在无法打开google的情况, 此时我们必须要配置无头浏览器, 这时候想下载文件到指定位置就不是刚才配置一下'prefs', 因为我们在无头浏览器中是无法导出下载管理的可视化窗口的, 这意味着'prefs'将没有作用, 所以我们应该怎么做?

基于无头浏览器将文件下载到指定目录:

    1. 配置无头浏览器并下载文件到google指定目录(配置options)

    2. 通过定时等待判断文件是否下载完成(os模块)

    3. 关闭浏览器

下面是代码部分:

import os

import time

from selenium import webdriver

from selenium.webdriver.chrome.options import Options

from selenium.webdriver.common.by import By

# # 设置 ChromeOptions 配置

options = Options()

options.add_argument('--no-sandbox')

options.add_argument('--disable-dev-shm-usage')

options.add_argument('--headless')

options.add_argument('blink-settings=imagesEnabled=false')

web = webdriver.Chrome(options=options)

# 配置浏览器

web.command_executor._commands["send_command"] = ("POST", '/session/$sessionId/chromium/send_command')

params = {'cmd': 'Page.setDownloadBehavior',

          'params': {'behavior': 'allow', 'downloadPath': r"%s" % os.path.join(os.getcwd(), 'download')}}

web.execute("send_command", params=params)

web.get("https://pypi.org/project/selenium/#files")

web.find_element(By.XPATH, '//*[@id="files"]/div/div[2]/a[1]').click()

# 等待下载完成

# 通过文件大小变化(在上面)或者定时等待的方式均可判断

# 这里用定时等待的方式

download_file = os.path.join('C://Users//86183//Downloads', "selenium-4.9.0.tar.gz")

while not os.path.exists(download_file):

    time.sleep(2)

# # 移动文件到指定的目录

# destination = os.path.join(os.getcwd(), 'download')

# shutil.move(download_file, destination)

# 关闭浏览器

web.quit()

selenium控制文件下载位置的相关教程结束。