利用 Python 和 Selenium 高效启动和管理 Chrome 浏览器

在自动化测试和网页抓取的过程中,Selenium 是最常用的工具之一。其强大的功能可以与浏览器无缝集成,实现复杂的操作。然而,为了提高效率和扩展性,尤其在处理大量任务时,我们可以通过定制化的方法启动 Chrome 浏览器并与 Selenium 驱动连接。本文将详细解析两段关键代码,实现以下目标:

  1. 使用批处理(.bat)文件启动 Chrome 浏览器。
  2. 利用 Selenium WebDriver 动态连接已启动的浏览器实例。

优势:启动浏览器后可以随时获取driver,同样也可以随时driver.quit()进行其他操作后再获取driver。


代码实现

1. 使用批处理启动 Chrome

Chrome 提供了丰富的启动参数供开发者自定义浏览器行为。下面的函数通过生成 .bat 文件,以指定参数启动 Chrome:

def start_chrome():
    """
    通过bat的形式启动chrome浏览器
    """
    script = f"""
@echo off
start "" "{config.chrome_path}" --ignore-certificate-errors --remote-debugging-port=9528 --user-data-dir="C:\selenium\ChromeProfile" --download-default-directory="{config.tmp_dir}" --incognito --start-maximized
    """
    bat_path = pathlib.Path(config.tmp_dir, 'start_chrome.bat')
    bat_path.open('w').write(script)
    shell = wincl.Dispatch("WScript.Shell")
    shell.Run(str(bat_path), 1, True)
    time.sleep(3)

代码解析

  1. 批处理文件的生成

    • 利用 Python 的 pathlib 模块,动态生成批处理文件 start_chrome.bat
    • 文件内容包括多个 Chrome 的启动参数:
      • --ignore-certificate-errors:忽略 SSL 证书错误,适用于测试环境。
      • --remote-debugging-port=9528:开启调试模式,指定调试端口。
      • --user-data-dir="C:\selenium\ChromeProfile":定义浏览器用户数据目录,用于隔离会话。
      • --download-default-directory:指定文件下载路径。
      • --incognito:开启无痕模式,避免缓存和 Cookie 干扰。
      • --start-maximized:窗口启动即最大化。
  2. 执行批处理文件

    • 借助 Windows 脚本宿主(WScript.Shell)启动批处理文件,避免阻塞主线程。
    • 使用 time.sleep(3) 确保浏览器完全启动。

2. 利用 Selenium WebDriver 连接已启动的 Chrome

启动浏览器后,通过以下代码,连接到已打开的 Chrome 实例进行自动化操作:

python 复制代码
def get_driver(executable_path="chromedriver"):
    """
    获取web driver
    """
    options = webdriver.ChromeOptions()
    options.debugger_address = "127.0.0.1:9528"  # 指定调试端口
    options.add_argument('--ignore-certificate-errors')
    driver = webdriver.Chrome(options=options, executable_path=executable_path)
    driver.implicitly_wait(10)  # 设置隐式等待
    return driver

代码解析

  1. 调试模式连接

    • 使用 options.debugger_address 配置 WebDriver,通过 127.0.0.1:9528 的调试端口连接到已经启动的 Chrome 浏览器。
  2. 忽略证书错误

    • 再次添加 --ignore-certificate-errors 参数,确保与浏览器配置一致,避免潜在冲突。
  3. 隐式等待

    • 设置 driver.implicitly_wait(10),让 WebDriver 在查找元素时最多等待 10 秒,提升稳定性。

两者结合的优势

  1. 快速启动与复用

    • 通过批处理文件启动的 Chrome,可以在多次测试中复用,减少浏览器多次启动带来的资源消耗。
  2. 调试友好

    • 启用 --remote-debugging-port 后,开发者可以通过 Chrome DevTools 直接查看调试信息。
  3. 个性化配置

    • --user-data-dir 允许隔离用户数据,模拟不同用户的浏览行为。
    • 自定义下载路径,方便批量处理文件。

使用示例

完整流程如下:

  1. 启动 Chrome:

    python 复制代码
    start_chrome()
  2. 获取 WebDriver 并操作浏览器:

    python 复制代码
    driver = get_driver()
    driver.get("https://www.example.com")
    print(driver.title)

总结

通过上述方法,我们实现了批处理文件与 Selenium WebDriver 的无缝对接。这种方式特别适用于高效处理自动化测试任务,同时支持灵活的浏览器配置。如果你在实际使用中需要进一步优化,可以根据具体需求调整启动参数,比如添加代理、禁用图片加载等。

这不仅是自动化测试中的一项技巧,也是一种提升开发效率的思路。希望这篇文章能为你的 Selenium 使用提供新的启发!

相关推荐
gqkmiss1 小时前
Chrome 浏览器插件获取网页 iframe 中的 window 对象
前端·chrome·iframe·postmessage·chrome 插件
bryant_meng2 小时前
【python】OpenCV—Image Moments
开发语言·python·opencv·moments·图片矩
KevinRay_2 小时前
Python超能力:高级技巧让你的代码飞起来
网络·人工智能·python·lambda表达式·列表推导式·python高级技巧
Captain823Jack3 小时前
nlp新词发现——浅析 TF·IDF
人工智能·python·深度学习·神经网络·算法·自然语言处理
资源补给站3 小时前
大恒相机开发(2)—Python软触发调用采集图像
开发语言·python·数码相机
Captain823Jack3 小时前
w04_nlp大模型训练·中文分词
人工智能·python·深度学习·神经网络·算法·自然语言处理·中文分词
PieroPc4 小时前
Python 自动化 打开网站 填表登陆 例子
运维·python·自动化
VinciYan4 小时前
基于Jenkins+Docker的自动化部署实践——整合Git与Python脚本实现远程部署
python·ubuntu·docker·自动化·jenkins·.net·运维开发
测试老哥5 小时前
外包干了两年,技术退步明显。。。。
自动化测试·软件测试·python·功能测试·测试工具·面试·职场和发展
終不似少年遊*5 小时前
美国加州房价数据分析01
人工智能·python·机器学习·数据挖掘·数据分析·回归算法