4

我有一个使用 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 行的 a.before 是乱码，正常打印不起作用，即使我看到第 5 行的命令实际上产生了很多输出。

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

我正在使用 tcsh 外壳

4

2 回答 2

8

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

pexpect 有一个我认为可以使用的“env”选项。像这样：

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

但这会产生一个对我不起作用的新外壳，我们的开发环境取决于很多环境变量：/

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

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

我更改了当前的 "TERM" 和 "dumb" 不支持颜色，这解决了我的问题。

于 2008-12-11T12:52:34.980 回答
2

在pexpect 文档中找不到用于设置终端的任何内容，但您可能可以使用 shell 显式启动程序，然后在此处设置终端类型：

shell_cmd = 'ls -l | grep LOG > log_list.txt'
child = pexpect.spawn('/bin/bash', ['-c', shell_cmd])
child.expect(pexpect.EOF)

你可以尝试类似的东西

child = pexpect.spawn('TERM=vt100 /bin/bash', ['-c', shell_cmd])

您也可以使用 --norc 和类似的方式启动 bash，以避免读取初始化文件。查看bash 手册页。

于 2008-12-11T09:38:59.270 回答