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 __init__(self, stream, color): StreamWriter.__init__(self, stream) self.code = ANSI_CODES[color] self.reset = ANSI_CODES["reset"]