def __init__(self, stream, color): StreamWriter.__init__(self, stream) self.reset = self._get_color() self.background = self.reset & 0xf0 CODES = { "red" : self.FOREGROUND_RED | self.FOREGROUND_INTENSITY | self.background, "green" : self.FOREGROUND_GREEN | self.FOREGROUND_INTENSITY | self.background, "yellow" : self.FOREGROUND_GREEN | self.FOREGROUND_RED | self.FOREGROUND_INTENSITY | self.background, "blue" : self.FOREGROUND_BLUE | self.FOREGROUND_INTENSITY | self.background } self.code = CODES[color]
def write(self, s): self._set_color(self.code) StreamWriter.write(self, s) self._set_color(self.reset)
def __init__(self, stream, color): StreamWriter.__init__(self, stream) self.code = ANSI_CODES[color] self.reset = ANSI_CODES["reset"]
def write(self, s): StreamWriter.write(self, self.code) StreamWriter.write(self, s) StreamWriter.write(self, self.reset)
def __init__(self, stream): self.normal = StreamWriter(stream) self.success = writer_class(stream, "green") self.failure = writer_class(stream, "red") bgcolor = self.success.get_bgcolor() self.warning = writer_class(stream, self._get_warning_color(bgcolor))