Python执行awssaml命令时，双因素认证码无法传递的问题咨询

问题 #

需要在Python中自动执行终端命令：

awssaml get-credentials --account-id **** --name **** --role **** --user-name ****

该命令会依次请求密码和双因素认证码，已将这两个值存入Python变量。尝试用subprocess.Popen传递时，密码能正常输入，但双因素认证码无法传递，程序停滞。改用pexpect的child.expect('password:')方法时，触发pexpect.exceptions.TIMEOUT错误，错误信息如下：

after: <class 'pexpect.exceptions.TIMEOUT'>
match: None
match_index: None
exitstatus: None
flag_eof: False
pid: 66502
child_fd: 10
closed: False
timeout: 30
delimiter: <class 'pexpect.exceptions.EOF'>
logfile: None
logfile_read: None
logfile_send: None
maxread: 2000
ignorecase: False
searchwindowsize: None
delaybeforesend: 0.05
delayafterclose: 0.1
delayafterterminate: 0.1
searcher: sear

问题原因 #

1. 提示符匹配不精确：awssaml的密码提示可能不是精确的password:，比如存在大小写差异（如Password:）、前后空格、换行，或者带有终端颜色的ANSI转义码，导致pexpect无法匹配到目标字符串。
2. 子进程输出缓冲：部分命令会将输出缓冲到终端，pexpect无法实时捕获到提示内容，超过超时时间后触发TIMEOUT。
3. 命令执行流程不符预期：awssaml可能先输出其他信息，或者提示顺序、提示文本和预设的不一致，导致pexpect在错误的时机等待不存在的提示符。

解决办法 #

1. 修正提示符匹配逻辑 #

• 使用正则表达式做宽松匹配，忽略大小写和无关字符：

  child.expect(r'[Pp]assword:', timeout=60)
  

• 如果提示带有ANSI颜色码，要么给awssaml添加--no-color（若支持）参数关闭颜色输出，要么用正则忽略转义码：

  child.expect(r'\x1b\[[0-9;]*m[Pp]assword:\x1b\[[0-9;]*m', timeout=60)
  

• 双因素认证码的提示同理，先确认真实提示文本（比如是MFA code:还是OTP code:），再调整匹配规则。

2. 禁用子进程输出缓冲 #

启动pexpect时指定编码并强制行缓冲，确保实时捕获输出：

import pexpect

cmd = "awssaml get-credentials --account-id **** --name **** --role **** --user-name ****"
# 启用编码并关闭回显，避免输入内容干扰匹配
child = pexpect.spawn(cmd, encoding='utf-8', echo=False)
# 匹配密码提示
child.expect(r'[Pp]assword:', timeout=60)
child.sendline(password_var)
# 匹配MFA提示，根据实际提示调整正则
child.expect(r'[Mm][Ff][Aa] [Cc][Oo][Dd][Ee]:', timeout=60)
child.sendline(mfa_code_var)
# 等待命令执行完成并获取输出
child.expect(pexpect.EOF)
print(child.before)

3. 调试查看真实输出 #

开启pexpect日志，记录子进程的所有输出，确认实际提示内容：

child.logfile_read = open('awssaml_debug.log', 'w', encoding='utf-8')

执行后查看日志文件，就能看到awssaml的真实提示文本，再针对性调整匹配规则。

4. 尝试环境变量或配置文件替代 #

检查awssaml是否支持通过环境变量传递密码和MFA码，若支持可直接设置环境变量后用subprocess执行：

import subprocess
import os

env = os.environ.copy()
env['AWS_PASSWORD'] = password_var  # 变量名以awssaml实际支持为准
env['AWS_MFA_CODE'] = mfa_code_var
result = subprocess.run(cmd, shell=True, env=env, capture_output=True, text=True)
print(result.stdout)

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