selenium控制文件下载位置

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