Selenium自动化下载文件并自定义文件名的实践指南

import io.github.bonigarcia.wdm.WebDriverManager;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.chrome.ChromeOptions;

import java.io.File;
import java.util.HashMap;
import java.util.Map;

public class SeleniumDownloadConfig {

    public static WebDriver setupDriverWithDownloadPath() {
        // 自动管理 ChromeDriver 版本
        WebDriverManager.chromedriver().setup();

        ChromeOptions options = new ChromeOptions();

        // 定义下载文件的目标路径
        // 这里使用项目根目录下的 "downloads" 文件夹
        String downloadFilepath = System.getProperty("user.dir") + File.separator + "downloads";
        System.out.println("Chrome 下载路径设置为: " + downloadFilepath);

        // 确保下载目录存在，如果不存在则创建
        File downloadtoFolder = new File(downloadFilepath);
        if (!downloadtoFolder.exists()) {
            downloadtoFolder.mkdir();
        }

        // 配置 Chrome 浏览器的偏好设置
        Map<String, Object> prefs = new HashMap<>();
        prefs.put("credentials_enable_service", false); // 禁用密码管理器服务
        prefs.put("profile.password_manager_enabled", false); // 禁用密码管理器
        prefs.put("profile.default_content_settings.popups", 0); // 禁用弹出窗口
        prefs.put("download.prompt_for_download", false); // 不弹出下载确认框，直接下载
        prefs.put("download.default_directory", downloadFilepath); // 设置默认下载目录
        prefs.put("profile.default_content_setting_values.notifications", 1); // 允许通知 (根据需要调整)
        prefs.put("profile.default_content_settings.cookies", 1); // 允许cookies (根据需要调整)

        // 将偏好设置应用到 ChromeOptions
        options.setExperimentalOption("prefs", prefs);

        // 初始化 ChromeDriver
        WebDriver driver = new ChromeDriver(options);
        return driver;
    }

    public static void main(String[] args) {
        WebDriver driver = setupDriverWithDownloadPath();
        // 在这里执行你的下载操作，例如导航到下载链接并点击
        // driver.get("your_download_page_url");
        // driver.findElement(By.id("downloadButton")).click();

        // 示例：等待一段时间，模拟下载完成
        try {
            Thread.sleep(5000); // 等待5秒，确保文件有时间下载
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

        // 下载完成后，调用重命名方法
        String downloadFilepath = System.getProperty("user.dir") + File.separator + "downloads";
        fileRename("my_renamed_file.pdf", downloadFilepath);

        driver.quit();
    }
}

import java.io.File;
import java.util.Arrays;
import java.util.Comparator;
import java.util.List;
import java.util.Optional;

public class FileRenamer {

    /**
     * 重命名指定目录中最新下载的文件。
     * 该方法会查找目录中最新修改的文件，并将其重命名为指定的新文件名。
     *
     * @param newFileName 目标文件名（包含扩展名，例如 "report.pdf"）
     * @param folderPath 包含下载文件的目录路径
     * @return 如果成功重命名，返回 true；否则返回 false。
     */
    public static boolean renameLatestDownloadedFile(String newFileName, String folderPath) {
        File folder = new File(folderPath);
        System.out.println("正在读取目录: " + folder.getAbsolutePath());

        if (!folder.isDirectory()) {
            System.err.println("错误: 指定路径不是一个有效的目录: " + folderPath);
            return false;
        }

        File[] files = folder.listFiles();
        if (files == null || files.length == 0) {
            System.out.println("目录中没有文件可供重命名。");
            return false;
        }

        // 查找最新修改的文件
        Optional<File> latestFileOptional = Arrays.stream(files)
                .filter(File::isFile) // 确保是文件而不是子目录
                .max(Comparator.comparingLong(File::lastModified)); // 根据最后修改时间排序

        if (latestFileOptional.isPresent()) {
            File fileToRename = latestFileOptional.get();
            String newFilePath = folderPath + File.separator + newFileName;
            File newFile = new File(newFilePath);

            // 如果新文件名已经存在，可能需要进行处理，例如删除旧文件或修改新文件名
            if (newFile.exists()) {
                System.out.println("警告: 目标文件 '" + newFileName + "' 已存在。正在覆盖。");
                newFile.delete(); // 或者选择不覆盖，抛出异常等
            }

            boolean isRenamed = fileToRename.renameTo(newFile);
            if (isRenamed) {
                System.out.println(String.format("成功将文件 '%s' 重命名为 '%s'", fileToRename.getName(), newFileName));
            } else {
                System.err.println(String.format("错误: 文件 '%s' 未能重命名为 '%s'", fileToRename.getName(), newFileName));
            }
            return isRenamed;
        } else {
            System.out.println("未找到可重命名的文件。");
            return false;
        }
    }

    // 原始提供的重命名方法，仅作参考，不推荐直接使用，因为它会重命名所有文件
    private static void fileRename(String newFileName, String folderPath) {
        File folder = new File(folderPath);
        System.out.println("Reading this " + folder.toString());
        if (folder.isDirectory()) {
            File[] files = folder.listFiles();
            List<File> filelist = Arrays.asList(files);
            filelist.forEach(f -> {
                System.out.println(f.getAbsolutePath());
                String newName = folderPath + File.separator + newFileName; // 修正路径分隔符
                System.out.println(newName);
                boolean isRenamed = f.renameTo(new File(newName));
                if (isRenamed)
                    System.out.println(String.format("Renamed this file %s to  %s", f.getName(), newName));
                else
                    System.out.println(String.format("%s file is not renamed to %s", f.getName(), newName));

            });

        }
    }
}