某些字符坚持我在Python cmd中的彩色提示

问题描述

我正在使用Python 2's cmd module为程序创建命令行。只要我不给提示符添加颜色，一切都会正常运行。

工作代码：

from cmd import Cmd

class App(Cmd):

    def __init__(self):
        Cmd.__init__(self)
        self.prompt = "PG ["+ (str('username'), 'green') +"@"+ str('hostname') +"]: "

    def do_exit(self, line):
        '''
        '''
        return True

App().cmdloop()

当我按如下方式更改代码时，如果我输入一个长命令或尝试在命令历史记录中搜索，某些字符将停留在我的提示符上。

问题代码：

from cmd import Cmd

class App(Cmd):

    def __init__(self):
        Cmd.__init__(self)
        self.prompt = "PG ["+ self.colorize(str('username'), 'green') +"@"+ str('hostname') +"]: "

    colorcodes =    {'green':{True:'x1b[32m',False:'x1b[39m'}}


    def colorize(self, val, color):
        return self.colorcodes[color][True] + val + self.colorcodes[color][False]

    def do_exit(self, line):
        '''
        '''
        return True

App().cmdloop()

您可以在asciicasts中看到此问题。cmd2 module也存在此问题。

---

解决方案

只需向您的颜色代码添加标记：

    colorcodes =    {'green':{True:'x01x1b[32mx02',False:'x01x1b[39mx02'}}
                                  # ^^^^        ^^^^         ^^^^        ^^^^

在asciicast中，当您退出i-search模式时出现问题，提示符会重新打印。这是因为Python不知道转义字符实际上不会占用屏幕上的空间。在每个转义序列之前放置x01，在每个转义序列之后放置x02，这会告诉Python假定这些字符不占用空间，因此提示符将被正确重新打印。

这与this answer中的解决方案相同，后者在不同的上下文中有相应的问题。在Python readline文档中有open issue提到这一点，但我没有看到它已经完成。

我在Cygwin上使用Python2.7.12测试了上述colorcodes值，运行在mintty中。在提示符中，username打印为绿色，其他所有内容均打印为默认值(浅灰色)。我使用的是标准系统cmd模块，不是cmd2(您链接的)。