如何用Splinter指定文件名保存点击生成的随机URL文件

这种情况确实挺棘手的——生成文件的URL是动态随机的，还藏在JS逻辑里，没法直接通过常规的URL请求下载，只能依赖浏览器触发生成流程。不过我们可以通过配置浏览器的下载行为，绕过手动的保存对话框，自动把文件存到指定位置，之后再重命名成你想要的名字。

下面分Chrome和Firefox两种常用浏览器给你具体方案：

一、Chrome浏览器配置方案 #

Chrome允许我们通过设置prefs参数来指定下载目录、自动跳过保存对话框，步骤如下：

1. 先定义Chrome的下载偏好设置：

from splinter import Browser
import os

# 替换成你想存放文件的目标目录
download_dir = "/path/to/your/target/folder"
# 确保目录存在，不存在则创建
os.makedirs(download_dir, exist_ok=True)

chrome_prefs = {
    "download.default_directory": download_dir,
    "download.prompt_for_download": False,  # 关闭手动保存对话框
    "download.directory_upgrade": True,
    "safebrowsing.enabled": True  # 避免浏览器拦截下载
}

# 初始化浏览器时加载这些偏好
browser = Browser('chrome', options={'prefs': chrome_prefs})

2. 正常执行点击操作：

# 打开目标页面
browser.visit("your_target_website_url")

# 定位并点击生成文件的按钮（替换成你实际的定位方式，比如id、xpath）
browser.find_by_id("download_trigger_btn").click()

3. 等待下载完成后重命名文件：
   因为网站生成的文件名是随机的，我们需要找到下载目录里最新的文件，再改成自己想要的名字：

import glob
import time

# 根据文件大小调整等待时间，或者用轮询方式更可靠
time.sleep(5)

# 获取下载目录中最新的文件
list_of_files = glob.glob(os.path.join(download_dir, "*"))
latest_file = max(list_of_files, key=os.path.getctime)

# 替换成你想要的文件名（记得带对应后缀）
new_filename = "my_custom_filename.csv"
os.rename(latest_file, os.path.join(download_dir, new_filename))

二、Firefox浏览器配置方案 #

Firefox需要通过配置profile来实现自动下载，步骤如下：

1. 创建Firefox配置文件并设置下载偏好：

from splinter import Browser
from selenium.webdriver.firefox.options import Options
from selenium.webdriver.firefox.firefox_profile import FirefoxProfile
import os

download_dir = "/path/to/your/target/folder"
os.makedirs(download_dir, exist_ok=True)

firefox_profile = FirefoxProfile()
# 设置自定义下载目录
firefox_profile.set_preference("browser.download.dir", download_dir)
firefox_profile.set_preference("browser.download.folderList", 2)  # 2代表使用自定义目录
# 指定自动下载的文件MIME类型（根据你下载的文件类型调整，比如PDF是application/pdf）
firefox_profile.set_preference("browser.helperApps.neverAsk.saveToDisk", "text/csv, application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")
firefox_profile.set_preference("browser.download.manager.showWhenStarting", False)  # 关闭下载管理器弹窗

options = Options()
options.profile = firefox_profile

# 初始化浏览器
browser = Browser('firefox', options=options)

2. 点击按钮并重命名文件：
   和Chrome的步骤一致，点击生成按钮后等待下载完成，再找到最新文件重命名即可。

注意事项 #

• 等待下载的时间可以根据文件大小灵活调整，也可以用循环轮询检查文件是否停止写入（比如对比文件前后的大小），比固定time.sleep更可靠。
• 文件的MIME类型要准确对应，你可以打开浏览器开发者工具，在网络请求中查看下载文件的Content-Type字段来获取。

内容的提问来源于stack exchange，提问作者user3992979