使用Python Wexpect执行PowerShell命令连接AWS Linux实例遇EOF错误

问题分析 #

EOF错误说明wexpect.spawn启动的powershell子进程提前终止，没有维持到Linux会话建立。通常由以下原因导致：

• Python字符串的路径转义错误，导致headnode.ps1脚本未正确执行
• headnode.ps1脚本未启动交互式SSH会话，执行后直接退出
• 未处理SSH连接过程中的中间提示（如密码输入、主机指纹确认）

修复步骤 #

1. 修正Python字符串转义问题 #

原代码中的路径转义容易出错，推荐用原始字符串或参数列表的方式定义powershell命令，避免转义混乱：

方式一：使用原始字符串

import wexpect

child = wexpect.spawn(r'powershell -Command "Set-ExecutionPolicy -Scope Process -ExecutionPolicy Bypass -Force; H:\scripts\workstation\headnode.ps1 -myUser username"')

方式二：拆分命令为参数列表

import wexpect

child = wexpect.spawn('powershell', [
    '-Command',
    'Set-ExecutionPolicy -Scope Process -ExecutionPolicy Bypass -Force; H:\\scripts\\workstation\\headnode.ps1 -myUser username'
])

2. 确保headnode.ps1启动交互式SSH会话 #

检查你的PowerShell脚本，确认它通过SSH建立持久交互式会话，而非执行后立即退出。关键是给SSH命令添加-t参数强制分配伪终端：

# headnode.ps1中的核心SSH命令示例
ssh -t ubuntu@your-aws-instance-ip

3. 处理SSH连接的中间提示 #

如果是首次连接实例或需要密码登录，必须在代码中处理这些提示，否则会导致子进程卡住或退出：

import wexpect

child = wexpect.spawn(r'powershell -Command "Set-ExecutionPolicy -Scope Process -ExecutionPolicy Bypass -Force; H:\scripts\workstation\headnode.ps1 -myUser username"')

# 处理首次连接的主机指纹确认提示
try:
    child.expect(r"Are you sure you want to continue connecting \(yes/no/\[fingerprint\]\)\?", timeout=10)
    child.sendline("yes")
except wexpect.TIMEOUT:
    # 无指纹提示则跳过
    pass

# 处理密码输入提示（按需启用）
try:
    child.expect(r"password:", timeout=20)
    child.sendline("your-ssh-password")
except wexpect.TIMEOUT:
    pass

# 等待Linux提示符出现
child.expect([r'\$ ', r'# ', r'ubuntu@'], timeout=60)

# 执行后续命令
child.sendline('sudo su - ubuntu')
child.expect([r'# ', r'ubuntu@'], timeout=30)
child.sendline('sbatch -J job.sbatch')

4. 排查子进程输出 #

若仍出现EOF，可开启日志查看子进程的真实输出，定位问题：

# 将子进程输出写入日志文件
child.logfile = open('wexpect_output.log', 'w')

查看日志文件，确认powershell命令是否执行成功、是否有错误导致进程提前退出。

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