Selenium自动化：高效排序与选择高价网页元素

import org.openqa.selenium.By;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import java.util.Comparator;
import java.util.List;
import java.util.stream.Collectors;

public class PriceSortingAndSelection {

    /**
     * 根据价格排序并选择第N个最高价的元素
     *
     * @param driver WebDriver实例
     * @param priceXPath 包含价格文本的WebElement的XPath表达式
     * @param n 要选择的第N个最高价元素 (例如，n=5表示选择第5个最高价)
     * @return 第N个最高价的WebElement，如果不存在则返回null
     */
    public static WebElement selectNthHighestPriceElement(WebDriver driver, String priceXPath, int n) {
        // 1. 查找所有价格元素
        List<WebElement> priceElements = driver.findElements(By.xpath(priceXPath));

        // 2. 使用Stream API对WebElement进行排序
        //    - sorted(): 接受一个Comparator，用于定义排序规则
        //    - Comparator: 比较两个WebElement的价格
        //    - limit(n): 限制结果集为前n个元素
        //    - collect(Collectors.toList()): 将Stream结果收集为List
        List<WebElement> topNPriceElements = priceElements.stream()
                .sorted(new Comparator<WebElement>() {
                    @Override
                    public int compare(WebElement e1, WebElement e2) {
                        float price1 = 0;
                        float price2 = 0;
                        try {
                            // 尝试从元素的文本中解析价格
                            price1 = Float.parseFloat(e1.getText().trim());
                            price2 = Float.parseFloat(e2.getText().trim());
                        } catch (NumberFormatException ex) {
                            // 处理价格无法解析的情况，例如设置为0或抛出异常
                            System.err.println("警告：无法解析价格文本 - " + ex.getMessage());
                            // 可以选择跳过此元素，或赋予默认值
                        }
                        // 降序排序（从高到低），所以 price2 - price1
                        return Float.compare(price2, price1);
                    }
                })
                .limit(n) // 获取前N个最高价的元素
                .collect(Collectors.toList());

        // 3. 检查并返回第N个元素
        if (topNPriceElements.size() >= n) {
            // 列表索引从0开始，所以第N个元素是索引 n-1
            return topNPriceElements.get(n - 1);
        } else {
            System.out.println("警告：找到的元素少于 " + n + " 个，无法获取第 " + n + " 个最高价元素。");
            return null;
        }
    }

    public static void main(String[] args) {
        // 假设您已经初始化了WebDriver实例 (例如 ChromeDriver)
        // WebDriver driver = new ChromeDriver();
        // driver.get("您的电商网站URL");

        // 示例用法：选择并点击第5个最高价的商品
        // String priceXPath = "//span[@class='a-price-whole']"; // 假设价格元素XPath
        // WebElement fifthHighestPriceElement = selectNthHighestPriceElement(driver, priceXPath, 5);

        // if (fifthHighestPriceElement != null) {
        //     System.out.println("成功找到第5个最高价的元素，其价格为：" + fifthHighestPriceElement.getText());
        //     // fifthHighestPriceElement.click(); // 执行点击操作
        // } else {
        //     System.out.println("未能找到第5个最高价的元素。");
        // }

        // driver.quit(); // 关闭浏览器
    }
}