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))
