patman: Fix indentation in terminal.py

This code came from a different project with 2-character indentation. Fix
it for U-Boot.

Signed-off-by: Simon Glass <sjg@chromium.org>
This commit is contained in:
Simon Glass 2014-08-28 09:43:34 -06:00
parent d6c1ffc7d2
commit 6ba5737ff0

View file

@ -15,66 +15,68 @@ import sys
COLOR_IF_TERMINAL, COLOR_ALWAYS, COLOR_NEVER = range(3) COLOR_IF_TERMINAL, COLOR_ALWAYS, COLOR_NEVER = range(3)
class Color(object): class Color(object):
"""Conditionally wraps text in ANSI color escape sequences.""" """Conditionally wraps text in ANSI color escape sequences."""
BLACK, RED, GREEN, YELLOW, BLUE, MAGENTA, CYAN, WHITE = range(8) BLACK, RED, GREEN, YELLOW, BLUE, MAGENTA, CYAN, WHITE = range(8)
BOLD = -1 BOLD = -1
BRIGHT_START = '\033[1;%dm' BRIGHT_START = '\033[1;%dm'
NORMAL_START = '\033[22;%dm' NORMAL_START = '\033[22;%dm'
BOLD_START = '\033[1m' BOLD_START = '\033[1m'
RESET = '\033[0m' RESET = '\033[0m'
def __init__(self, colored=COLOR_IF_TERMINAL): def __init__(self, colored=COLOR_IF_TERMINAL):
"""Create a new Color object, optionally disabling color output. """Create a new Color object, optionally disabling color output.
Args: Args:
enabled: True if color output should be enabled. If False then this enabled: True if color output should be enabled. If False then this
class will not add color codes at all. class will not add color codes at all.
""" """
self._enabled = (colored == COLOR_ALWAYS or self._enabled = (colored == COLOR_ALWAYS or
(colored == COLOR_IF_TERMINAL and os.isatty(sys.stdout.fileno()))) (colored == COLOR_IF_TERMINAL and os.isatty(sys.stdout.fileno())))
def Start(self, color, bright=True): def Start(self, color, bright=True):
"""Returns a start color code. """Returns a start color code.
Args: Args:
color: Color to use, .e.g BLACK, RED, etc. color: Color to use, .e.g BLACK, RED, etc.
Returns: Returns:
If color is enabled, returns an ANSI sequence to start the given color, If color is enabled, returns an ANSI sequence to start the given
otherwise returns empty string color, otherwise returns empty string
""" """
if self._enabled: if self._enabled:
base = self.BRIGHT_START if bright else self.NORMAL_START base = self.BRIGHT_START if bright else self.NORMAL_START
return base % (color + 30) return base % (color + 30)
return '' return ''
def Stop(self): def Stop(self):
"""Retruns a stop color code. """Retruns a stop color code.
Returns: Returns:
If color is enabled, returns an ANSI color reset sequence, otherwise If color is enabled, returns an ANSI color reset sequence,
returns empty string otherwise returns empty string
""" """
if self._enabled: if self._enabled:
return self.RESET return self.RESET
return '' return ''
def Color(self, color, text, bright=True): def Color(self, color, text, bright=True):
"""Returns text with conditionally added color escape sequences. """Returns text with conditionally added color escape sequences.
Keyword arguments: Keyword arguments:
color: Text color -- one of the color constants defined in this class. color: Text color -- one of the color constants defined in this
text: The text to color. class.
text: The text to color.
Returns: Returns:
If self._enabled is False, returns the original text. If it's True, If self._enabled is False, returns the original text. If it's True,
returns text with color escape sequences based on the value of color. returns text with color escape sequences based on the value of
""" color.
if not self._enabled: """
return text if not self._enabled:
if color == self.BOLD: return text
start = self.BOLD_START if color == self.BOLD:
else: start = self.BOLD_START
base = self.BRIGHT_START if bright else self.NORMAL_START else:
start = base % (color + 30) base = self.BRIGHT_START if bright else self.NORMAL_START
return start + text + self.RESET start = base % (color + 30)
return start + text + self.RESET