python,在pexpect中设置终端类型

Question

我有一个使用pexpect启动CLI程序的脚本.它的工作方式有点像shell,你可以在其中输入一些命令.

我认为,我遇到的问题是该程序使用彩色提示.

这就是我的工作

import pprint
import pexpect

1 a = pexpect.spawn('program')
2 a.expect("prompt>")
3 print "---------start------------"
4 print(a.before)
5 a.sendline("command")
6 a.expect("prompt>")
7 print "---------before------------"
8 pprint.pprint(a.before)
9 print "---------after------------"
10 pprint.pprint(a.after)

这是输出:

> python borken.py
---------start------------
A lot of text here from the enjoying programs start-up, lorem ipsum ...  
---------before------------
' \x1b[0m\x1b[8D\x1b[K\x1b[1m\x1b[34m'
---------after------------
'prompt>'

由于某种原因,第一个提示颜色编码使事情变得困难,并且在第8行之前是乱码,正常打印不起作用,即使我看到第5行的命令实际产生了大量输出.

有人知道问题可能是什么,或者是否可以在pexpect中设置终端类型以避免颜色？

我正在使用tcsh shell

Answer 1

好的,我找到了答案.csl的回答让我走上了正确的道路.

pexpect有一个"env"选项,我认为我可以使用它.像这样:

a = pexpect.spawn('program', env = {"TERM": "dumb"})

但这会产生一个对我不起作用的新shell,我们的开发环境依赖于很多环境变量:/

但是如果我在生成shell之前这样做:

import os
os.environ["TERM"] = "dumb"

我更改了当前的"TERM"和"哑"不支持颜色,这解决了我的问题.