本文详解如何使用 SeleniumBase 的 SB() 上下文管理器模式，通过内置方法（如 assert_downloaded_file 和 get_path_of_downloaded_file）精准等待、定位并验证下载文件，彻底解决 JS 重网站中异步导出触发后无法可靠捕获下载完成的问题。
本文详解如何使用 seleniumbase 的 `sb()` 上下文管理器模式，通过内置方法（如 `assert_downloaded_file` 和 `get_path_of_downloaded_file`）精准等待、定位并验证下载文件，彻底解决 js 重网站中异步导出触发后无法可靠捕获下载完成的问题。
在自动化下载场景中（尤其是统计类 Web 应用），单纯点击“导出 Excel”按钮并不等同于文件已落地——现代前端常通过 Blob URL、后台任务或 iframe 动态触发下载，浏览器实际写入磁盘存在延迟。此时若直接调用 driver.quit()，极可能因文件尚未完成写入而丢失数据。SeleniumBase 的 Driver() 类虽支持 CDP 设置下载路径，但缺乏对下载状态的主动感知能力；而更强大的 SB()（SeleniumBase Context Manager）模式则提供了开箱即用的下载断言与文件操作工具链。
✅ 推荐方案：使用 SB() + 下载断言
SB() 实例自动管理下载目录（默认为 ./downloads/）、超时策略及资源清理，并提供以下关键方法：
- sb.download_file(url)：安全下载远程资源（支持重定向和认证头）
- sb.assert_downloaded_file(filename)：阻塞式等待指定文件出现在下载目录，超时抛异常（默认 10 秒，可传 timeout=30 自定义）
- sb.get_path_of_downloaded_file(filename)：返回完整绝对路径，便于后续读取或校验
以下是适配您原始需求的重构示例（含登录、导出、验证全流程）：
from seleniumbase import SB
import os
DOWNLOADS_DIR = os.path.abspath("./downloads") # 确保路径为绝对路径
EXPORT_FILENAME = "statistics_export.xlsx" # 根据实际导出逻辑调整文件名
with SB(test=True, headless=False) as sb: # test=True 启用增强日志；headless=False 便于调试
# 1. 登录流程
sb.open("https://new.stamdata.com/app/statistics")
sb.click('button:contains("Account")')
sb.click("div#app div div:nth-of-type(2) span p")
sb.type('input[name="username"]', "xxxxx")
sb.type('input[name="password"]', "xxxxx")
sb.click('button:contains("sign in")')
# 2. 触发导出（确保元素可见且可点击）
sb.assert_element('button:contains("Excel export (max 50 000 rows)")')
sb.click('button:contains("Excel export (max 50 000 rows)")')
# 3. ✅ 关键：等待并验证下载完成（自动轮询，最多 20 秒）
sb.assert_downloaded_file(EXPORT_FILENAME, timeout=20)
# 4. （可选）进一步校验文件内容或大小
file_path = sb.get_path_of_downloaded_file(EXPORT_FILENAME)
file_size = os.path.getsize(file_path)
assert file_size > 1024, f"Downloaded file {EXPORT_FILENAME} is too small: {file_size} bytes"
# 5. 安全登出
sb.click('button:contains("menu")')
sb.click("div#app div div:nth-of-type(3) span p")
sb.click('button:contains("log out")')⚠️ 注意事项与最佳实践
- 文件名必须精确匹配：assert_downloaded_file() 依赖文件系统中的实际名称。若目标网站动态生成带时间戳的文件（如 stats_20240520_1423.xlsx），需先通过 sb.get_downloaded_files() 获取列表再正则匹配，或改用 sb.wait_for_file_present() 配合自定义逻辑。
- 下载目录权限：确保 ./downloads/ 目录存在且进程有写入权限（Windows 下避免使用反斜杠 \downloads，统一用正斜杠 /downloads 或 os.path.join）。
- 避免 Driver() 模式陷阱：Driver() 不自动管理下载上下文，execute_cdp_cmd("Page.setDownloadBehavior") 仅配置行为，不提供断言能力；混合使用 Driver() 和 SB() 方法会导致未定义行为。
- 超时设置建议：大文件或慢网络下将 timeout 设为 30–60 秒；若频繁超时，应检查是否触发了下载（如弹窗拦截、CSP 策略阻止 Blob 创建）。
-
清理策略：SB() 在退出时不会自动清空下载目录，如需每次测试干净环境，可在 with 块开头手动清空：
import shutil if os.path.exists(DOWNLOADS_DIR): shutil.rmtree(DOWNLOADS_DIR) os.makedirs(DOWNLOADS_DIR)
通过 SB() 的声明式断言机制，您不再需要手写轮询 os.path.exists() 或 time.sleep()，既提升脚本健壮性，又显著增强可维护性。将下载验证作为自动化流程的正式步骤，是构建生产级爬虫与 RPA 脚本的关键一环。
