VS Code SSH连接的特殊操作与EC2上Selenium自动化问题排查

VS Code SSH与普通终端SSH的核心差异 #

VS Code建立SSH连接时会触发一些普通终端SSH不会执行的操作，导致环境产生关键差异：

• 完整加载用户环境变量：VS Code的SSH会话属于交互式登录，会自动加载~/.bashrc、~/.bash_profile等配置文件中的所有环境变量；而普通终端SSH的非交互式会话（或crontab任务）仅加载~/.profile，部分关键变量（如PATH、图形会话相关变量）可能缺失。
• 初始化图形会话上下文：即使是headless模式，Chrome内部仍依赖部分X Window系统相关变量（如DISPLAY、XAUTHORITY），VS Code SSH会自动初始化这些变量的基础上下文，普通终端SSH或crontab会话则无此操作。
• 调整会话权限与资源访问：VS Code SSH会优化临时目录权限、会话资源限制，让Chrome/chromedriver能正常访问系统资源（如/dev/shm）。

无需VS Code连接的解决方法 #

1. 补充Chrome Headless模式的必要参数 #

修改Selenium脚本，添加以下参数确保Chrome在无图形环境下稳定运行：

from selenium.webdriver import Chrome
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.chrome.options import Options

if __name__ == '__main__':
    chrome_options = Options()
    chrome_options.add_argument('--headless=new') # 推荐使用新版headless模式
    chrome_options.add_argument('--no-sandbox') # 绕过沙箱限制（非root用户运行时必要）
    chrome_options.add_argument('--disable-dev-shm-usage') # 避免/dev/shm空间不足
    chrome_options.add_argument('--disable-gpu') # 禁用GPU加速（无GPU环境下必须）

    service = Service('/usr/bin/chromedriver')
    driver = Chrome(service=service, options=chrome_options)
    driver.get("https://example.com/")
    print(driver.title)
    driver.quit()

2. 给crontab配置完整环境变量 #

执行crontab -e编辑定时任务，在任务前指定Shell和完整环境变量，或加载用户配置文件：

# 指定Shell为bash，确保环境变量正常加载
SHELL=/bin/bash
# 配置完整PATH，避免命令找不到
PATH=/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/games:/usr/local/games:/snap/bin

# 每分钟运行脚本，输出日志到指定文件（方便排查问题）
* * * * * source ~/.bashrc && python3 /usr/share/stratoverse/SCC/SCC/appScripts/selenium/proofOfWork.py >> /var/log/selenium_script.log 2>&1

3. 使用虚拟X服务器（Xvfb） #

如果上述方法无效，可通过Xvfb模拟图形环境：

• 安装Xvfb：sudo apt update && sudo apt install xvfb
• 修改crontab任务，用xvfb-run运行脚本：

* * * * * xvfb-run -a python3 /usr/share/stratoverse/SCC/SCC/appScripts/selenium/proofOfWork.py >> /var/log/selenium_script.log 2>&1

• 或在Python脚本中集成Xvfb（需先安装xvfbwrapper：pip install xvfbwrapper）：

from selenium.webdriver import Chrome
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.chrome.options import Options
from xvfbwrapper import Xvfb

if __name__ == '__main__':
    # 启动虚拟X服务器
    vdisplay = Xvfb(width=1280, height=720)
    vdisplay.start()

    chrome_options = Options()
    chrome_options.add_argument('--headless')
    # 其他必要参数...

    service = Service('/usr/bin/chromedriver')
    driver = Chrome(service=service, options=chrome_options)
    driver.get("https://example.com/")
    print(driver.title)
    driver.quit()

    # 停止虚拟X服务器
    vdisplay.stop()

4. 验证Chrome与Chromedriver版本匹配 #

确保chromedriver版本和系统安装的Chrome版本完全一致：

• 查看Chrome版本：google-chrome --version
• 查看Chromedriver版本：chromedriver --version
  版本不匹配会导致WebDriver意外退出，需下载对应版本的chromedriver替换现有文件。

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