Question

我试图为我的curses输出添加颜色。然而，挑战在于文本是通过单个长条形式打印的，即self.all_results。有没有办法为字符串的单个部分添加颜色。

def main(self,stdscr):
    x,y = 0,0 # size of the window
    xx,yy = 50,200  # where to place window - up,across
    pad = curses.newpad(150,150) # nlines, ncols
    pad_pos = 0
    exit = False

    pad.addstr(4,0,str(self.all_results))

    while not exit:
        sleep(0.2)
        if self.timer != None:
            if time() - start > self.timer:
                self.stop = True
                break

        pad.addstr(0,0,str(self.format_results()))
        pad.refresh(pad_pos,0, x,y, xx,yy)

        cmd = stdscr.getch()
        stdscr.nodelay(1)

        if cmd != -1:
            pad.refresh(pad_pos,0, x,y, xx,yy)
            if len(self.format_results().split('\n')) > 100:
                if  cmd == curses.KEY_DOWN:
                    if pad_pos < len(self.format_results())+1:
                        pad_pos += 1
                    try:
                        pad.refresh(pad_pos,0, x,y, xx,yy)
                    except curses.error:
                        pass
                elif cmd == curses.KEY_UP:
                    if pad_pos != 0:
                        pad_pos -= 1
                    try:
                        pad.refresh(pad_pos,0, x,y, xx,yy)
                    except curses.error:
                        pass

Answer 1

我会使用re分割字符串，然后使用非x,y形式的addstr，指定每个部分的颜色。

#!/usr/bin/env python
import curses
from curses.wrapper import wrapper
import re


def addstr_colorized(win, y, x, s):
    colors = {'OK': curses.COLOR_GREEN, 'ERROR': curses.COLOR_RED}
    win.move(y, x)
    pattern = r'({0:s})'.format(
        '|'.join(r'\b{0:s}\b'.format(word) for word in colors.keys()))
    s = re.split(pattern, s)
    for s in s:
        win.addstr(s, curses.color_pair(colors.get(s, 0)))


def main(stdscr):
    curses.init_pair(curses.COLOR_RED,
                     curses.COLOR_RED,
                     curses.COLOR_BLACK)
    curses.init_pair(curses.COLOR_GREEN,
                     curses.COLOR_GREEN,
                     curses.COLOR_BLACK)

    addstr_colorized(stdscr,
                     4,
                     0,
                     "This line is OK.\nBut there is an ERROR in this line\n")
    stdscr.refresh()
    stdscr.getch()


wrapper(main)

enter image description here

Answer 2

这就是我过去常常做到的事情（不是很漂亮，但对于我正在做的事情来说这很好）。

您只需在某处定义颜色对，即：

curses.init_pair(1, curses.COLOR_BLUE, curses.COLOR_GREEN);

然后只需使用包含colourN [str]的消息调用该函数，其中N是颜色对num，str是您想要着色的字符串的一部分。

例如：

addStrColour(stdscr, 0, "This is an example message, colour1[This is coloured using colour pair 1!!!] and now we have normal text again");

这是功能：

def addstrColour(stdscr, pos, message):
  #Split messages based on colour components
  newMes = re.split("(colour\d\[.*?\])", message);

  totalOut = 0;

  for line in newMes:
    m = re.match("colour(\d{1})\[(.*)\]", line);
    if m:
      colourPairNum = int((m.groups()[0]));

      stdscr.addstr(pos, totalOut, m.groups()[1], curses.color_pair(colourPairNum));
      totalOut += len(m.groups()[1]);
    else:
      stdscr.addstr(pos, totalOut, line);
      totalOut += len(line);

诅咒中的颜色

2 个答案: