使用pexpect库SSH连接时Bash提示符异常的原因及解决方法

为什么会出现这个差异？ #

你看到的那些\x1b]0;root@centos-1:\~\x07之类的内容，其实是ANSI终端控制转义序列，本质原因是直接SSH和pexpect spawn的终端环境完全不一样：

• 直接通过SSH连接时，系统会给你分配一个完整的交互式伪终端（pty），Bash识别到这是“真实”的人机交互场景，就会输出简化的、适合人类阅读的提示符——也就是你看到的centos-1:~#。
• 而pexpect默认调用ssh时，没有强制分配伪终端，终端环境变量（比如TERM）的设置也和真实终端不同。这时候Bash会输出包含终端控制信息的内容：比如\x1b]0;...\x07是用来设置终端窗口标题的转义码，\x1b[?1034h是终端处理Unicode字符的相关设置。这些内容在真实终端里会被后台解析执行，不会显示出来，但pexpect会直接读取到原始的转义序列。

怎么解决这个问题？ #

这里有几个可行的方案，你可以根据需求选：

1. 强制SSH分配伪终端（最推荐）

在ssh命令里加上-t参数，强制分配一个伪终端，模拟真实交互式连接的环境。修改你的pexpect代码：

import pexpect
# 添加-t参数强制分配伪终端
child = pexpect.spawn("/bin/ssh", ["-t", "root@172.23.63.10"])
# 之后的操作就会得到和直接SSH一样的干净提示符了

这样Bash会识别到交互式终端环境，输出正常的centos-1:~#，不会附带转义序列。

2. 让pexpect启用伪终端

在spawn的时候指定use_pty=True参数，结合ssh的-t效果更好：

child = pexpect.spawn("/bin/ssh", ["-t", "root@172.23.63.10"], use_pty=True, encoding='utf-8')

3. 过滤输出中的转义序列

如果不想修改连接方式，也可以在读取输出后，用正则表达式去掉所有ANSI转义序列：

import re

# 读取输出后做清洗处理
raw_output = child.readline()
# 匹配并移除所有终端控制转义序列
clean_output = re.sub(r'\x1b\[[0-9;]*m|\x1b\]0;[^\x07]*\x07|\x1b\[\?1034h', '', raw_output)
print(clean_output)

这个方法适合需要保留原始连接方式，但只想得到干净的提示符内容的场景。

补充：你提供的示例对比 #

pexpect连接的原始输出（含转义序列）

>>> import pexpect
>>> child = pexpect.spawn("/bin/ssh",["root@172.23.63.10"]);
>>> child.readline(); '#############################################################\r\n'
# ... 省略中间的登录提示内容 ...
>>> child.sendline(""); 1
>>> child.readline(); '\x1b]0;root@centos-1:\~\x07\x1b[?1034hcentos-1:~# \r\n'

直接SSH连接的输出（干净提示符）

[greddy] # ssh root@172.23.63.10
#############################################################
# #
# 本系统仅限授权用户访问 #
# ... 省略中间的登录提示内容 ...
Last login: Wed Jan 30 08:51:17 2019 from 10.70.52.187
#############################################################
centos-1:~#

内容的提问来源于stack exchange，提问作者Guru Vishnu Vardhan Reddy