Pexpect: sendintr() 意外截断输出的解决方案

在使用Python的Pexpect模块时，sendintr() 函数（发送 Ctrl+C 信号）有时会意外截断程序的输出，导致结果与预期不符。 这篇文章将分析此问题并提供有效的解决方法。

问题描述：

在使用 pexpect.spawn() 连接远程主机并执行命令后，如果使用 sendintr() 中断进程，cmd.before 属性可能无法获取完整的输出信息。 部分输出内容会在 Ctrl+C 信号发送后丢失。

例如，以下代码片段：

import pexpect

host = '192.168.201.106'
password = r'123.com'
cmd = pexpect.spawn(f'ssh {host}', timeout=2, encoding='utf-8')
cmd.expect('password:')
cmd.sendline(password)
cmd.sendintr()  # 发送 Ctrl+C
print(cmd.before)

预期输出可能包含完整的 SSH 登录提示信息，但实际输出可能不完整，缺少部分字符。

解决方法：

问题根源在于 sendintr() 发送中断信号的时机。 信号发送后，目标程序的输出流可能尚未完全缓冲到 Pexpect 的读取缓冲区。 因此，直接读取 cmd.before 会丢失部分数据。

为了解决这个问题，建议在发送 sendintr() 之后，添加一个短暂的延迟，等待缓冲区刷新：

import pexpect
import time

host = '192.168.201.106'
password = r'123.com'
cmd = pexpect.spawn(f'ssh {host}', timeout=2, encoding='utf-8')
cmd.expect('password:')
cmd.sendline(password)
cmd.sendintr()
time.sleep(0.1)  # 添加短暂延迟
print(cmd.before)

通过添加 time.sleep(0.1) (或根据实际情况调整延迟时间)，我们可以确保 Pexpect 有足够的时间读取目标程序的完整输出。 这比简单的字符串拼接更可靠，因为拼接方法并不能保证捕获所有丢失的数据。

另一个更稳妥的方法是使用 cmd.read() 代替 cmd.before 来读取所有输出，这能确保读取到所有可用的数据，即使在 sendintr() 后：

import pexpect

host = '192.168.201.106'
password = r'123.com'
cmd = pexpect.spawn(f'ssh {host}', timeout=2, encoding='utf-8')
cmd.expect('password:')
cmd.sendline(password)
cmd.sendintr()
print(cmd.read())

选择哪种方法取决于你的具体需求和对性能的要求。 如果性能至关重要，可以尝试调整 time.sleep() 的时间；如果需要绝对确保获取所有输出，则 cmd.read() 是更可靠的选择。