__docformat__ = "epytext"
+import codecs
import commands
import errno
import formatter
havecolor=1
dotitles=1
+codes = {}
+color_codes = {}
+
esc_seq = "\x1b["
-g_attr = {}
-g_attr["normal"] = 0
-
-g_attr["bold"] = 1
-g_attr["faint"] = 2
-g_attr["standout"] = 3
-g_attr["underline"] = 4
-g_attr["blink"] = 5
-g_attr["overline"] = 6 # Why is overline actually useful?
-g_attr["reverse"] = 7
-g_attr["invisible"] = 8
-
-g_attr["no-attr"] = 22
-g_attr["no-standout"] = 23
-g_attr["no-underline"] = 24
-g_attr["no-blink"] = 25
-g_attr["no-overline"] = 26
-g_attr["no-reverse"] = 27
-# 28 isn't defined?
-# 29 isn't defined?
-g_attr["black"] = 30
-g_attr["red"] = 31
-g_attr["green"] = 32
-g_attr["yellow"] = 33
-g_attr["blue"] = 34
-g_attr["magenta"] = 35
-g_attr["cyan"] = 36
-g_attr["white"] = 37
-# 38 isn't defined?
-g_attr["default"] = 39
-g_attr["bg_black"] = 40
-g_attr["bg_red"] = 41
-g_attr["bg_green"] = 42
-g_attr["bg_yellow"] = 43
-g_attr["bg_blue"] = 44
-g_attr["bg_magenta"] = 45
-g_attr["bg_cyan"] = 46
-g_attr["bg_white"] = 47
-g_attr["bg_default"] = 49
-
-
-# make_seq("blue", "black", "normal")
+color_codes["normal"] = esc_seq + "0m"
+color_codes["reset"] = esc_seq + "39;49;00m"
+
+color_codes["bold"] = esc_seq + "01m"
+color_codes["faint"] = esc_seq + "02m"
+color_codes["standout"] = esc_seq + "03m"
+color_codes["underline"] = esc_seq + "04m"
+color_codes["blink"] = esc_seq + "05m"
+color_codes["overline"] = esc_seq + "06m"
+color_codes["reverse"] = esc_seq + "07m"
+color_codes["invisible"] = esc_seq + "08m"
+
+color_codes["no-attr"] = esc_seq + "22m"
+color_codes["no-standout"] = esc_seq + "23m"
+color_codes["no-underline"] = esc_seq + "24m"
+color_codes["no-blink"] = esc_seq + "25m"
+color_codes["no-overline"] = esc_seq + "26m"
+color_codes["no-reverse"] = esc_seq + "27m"
+
+color_codes["bg_black"] = esc_seq + "40m"
+color_codes["bg_darkred"] = esc_seq + "41m"
+color_codes["bg_darkgreen"] = esc_seq + "42m"
+color_codes["bg_brown"] = esc_seq + "43m"
+color_codes["bg_darkblue"] = esc_seq + "44m"
+color_codes["bg_purple"] = esc_seq + "45m"
+color_codes["bg_teal"] = esc_seq + "46m"
+color_codes["bg_lightgray"] = esc_seq + "47m"
+color_codes["bg_default"] = esc_seq + "49m"
+color_codes["bg_darkyellow"] = color_codes["bg_brown"]
+
def color(fg, bg="default", attr=["normal"]):
- mystr = esc_seq[:] + "%02d" % g_attr[fg]
+ mystr = color_codes[fg]
for x in [bg]+attr:
- mystr += ";%02d" % g_attr[x]
- return mystr+"m"
-
-
+ mystr += color_codes[x]
+ return mystr
-codes={}
-codes["reset"] = esc_seq + "39;49;00m"
-
-codes["bold"] = esc_seq + "01m"
-codes["faint"] = esc_seq + "02m"
-codes["standout"] = esc_seq + "03m"
-codes["underline"] = esc_seq + "04m"
-codes["blink"] = esc_seq + "05m"
-codes["overline"] = esc_seq + "06m" # Who made this up? Seriously.
-codes["reverse"] = esc_seq + "07m"
ansi_color_codes = []
for x in xrange(30, 38):
'0xFF55FF', '0x00AAAA', '0x55FFFF', '0xAAAAAA', '0xFFFFFF']
for x in xrange(len(rgb_ansi_colors)):
- codes[rgb_ansi_colors[x]] = esc_seq + ansi_color_codes[x]
+ color_codes[rgb_ansi_colors[x]] = esc_seq + ansi_color_codes[x]
del x
-codes["black"] = codes["0x000000"]
-codes["darkgray"] = codes["0x555555"]
+color_codes["black"] = color_codes["0x000000"]
+color_codes["darkgray"] = color_codes["0x555555"]
-codes["red"] = codes["0xFF5555"]
-codes["darkred"] = codes["0xAA0000"]
+color_codes["red"] = color_codes["0xFF5555"]
+color_codes["darkred"] = color_codes["0xAA0000"]
-codes["green"] = codes["0x55FF55"]
-codes["darkgreen"] = codes["0x00AA00"]
+color_codes["green"] = color_codes["0x55FF55"]
+color_codes["darkgreen"] = color_codes["0x00AA00"]
-codes["yellow"] = codes["0xFFFF55"]
-codes["brown"] = codes["0xAA5500"]
+color_codes["yellow"] = color_codes["0xFFFF55"]
+color_codes["brown"] = color_codes["0xAA5500"]
-codes["blue"] = codes["0x5555FF"]
-codes["darkblue"] = codes["0x0000AA"]
+color_codes["blue"] = color_codes["0x5555FF"]
+color_codes["darkblue"] = color_codes["0x0000AA"]
-codes["fuchsia"] = codes["0xFF55FF"]
-codes["purple"] = codes["0xAA00AA"]
+color_codes["fuchsia"] = color_codes["0xFF55FF"]
+color_codes["purple"] = color_codes["0xAA00AA"]
-codes["turquoise"] = codes["0x55FFFF"]
-codes["teal"] = codes["0x00AAAA"]
+color_codes["turquoise"] = color_codes["0x55FFFF"]
+color_codes["teal"] = color_codes["0x00AAAA"]
-codes["white"] = codes["0xFFFFFF"]
-codes["lightgray"] = codes["0xAAAAAA"]
+color_codes["white"] = color_codes["0xFFFFFF"]
+color_codes["lightgray"] = color_codes["0xAAAAAA"]
-codes["darkteal"] = codes["turquoise"]
+color_codes["darkteal"] = color_codes["turquoise"]
# Some terminals have darkyellow instead of brown.
-codes["0xAAAA00"] = codes["brown"]
-codes["darkyellow"] = codes["0xAAAA00"]
+color_codes["0xAAAA00"] = color_codes["brown"]
+color_codes["darkyellow"] = color_codes["0xAAAA00"]
-codes["bg_black"] = esc_seq + "40m"
-codes["bg_darkred"] = esc_seq + "41m"
-codes["bg_darkgreen"] = esc_seq + "42m"
-codes["bg_brown"] = esc_seq + "43m"
-codes["bg_darkblue"] = esc_seq + "44m"
-codes["bg_purple"] = esc_seq + "45m"
-codes["bg_teal"] = esc_seq + "46m"
-codes["bg_lightgray"] = esc_seq + "47m"
-codes["bg_darkyellow"] = codes["bg_brown"]
# Colors from /etc/init.d/functions.sh
-codes["NORMAL"] = esc_seq + "0m"
-codes["GOOD"] = codes["green"]
-codes["WARN"] = codes["yellow"]
-codes["BAD"] = codes["red"]
-codes["HILITE"] = codes["teal"]
-codes["BRACKET"] = codes["blue"]
+codes["NORMAL"] = ( "normal", )
+codes["GOOD"] = ( "green", )
+codes["WARN"] = ( "yellow", )
+codes["BAD"] = ( "red", )
+codes["HILITE"] = ( "teal", )
+codes["BRACKET"] = ( "blue", )
# Portage functions
-codes["INFORM"] = codes["darkgreen"]
-codes["UNMERGE_WARN"] = codes["red"]
-codes["SECURITY_WARN"] = codes["red"]
-codes["MERGE_LIST_PROGRESS"] = codes["yellow"]
-codes["PKG_BLOCKER"] = codes["red"]
-codes["PKG_BLOCKER_SATISFIED"] = codes["darkblue"]
-codes["PKG_MERGE"] = codes["darkgreen"]
-codes["PKG_MERGE_SYSTEM"] = codes["darkgreen"]
-codes["PKG_MERGE_WORLD"] = codes["green"]
-codes["PKG_UNINSTALL"] = codes["red"]
-codes["PKG_NOMERGE"] = codes["darkblue"]
-codes["PKG_NOMERGE_SYSTEM"] = codes["darkblue"]
-codes["PKG_NOMERGE_WORLD"] = codes["blue"]
-codes["PROMPT_CHOICE_DEFAULT"] = codes["green"]
-codes["PROMPT_CHOICE_OTHER"] = codes["red"]
-
-def parse_color_map(onerror=None):
+codes["INFORM"] = ( "darkgreen", )
+codes["UNMERGE_WARN"] = ( "red", )
+codes["SECURITY_WARN"] = ( "red", )
+codes["MERGE_LIST_PROGRESS"] = ( "yellow", )
+codes["PKG_BLOCKER"] = ( "red", )
+codes["PKG_BLOCKER_SATISFIED"] = ( "darkblue", )
+codes["PKG_MERGE"] = ( "darkgreen", )
+codes["PKG_MERGE_SYSTEM"] = ( "darkgreen", )
+codes["PKG_MERGE_WORLD"] = ( "green", )
+codes["PKG_UNINSTALL"] = ( "red", )
+codes["PKG_NOMERGE"] = ( "darkblue", )
+codes["PKG_NOMERGE_SYSTEM"] = ( "darkblue", )
+codes["PKG_NOMERGE_WORLD"] = ( "blue", )
+codes["PROMPT_CHOICE_DEFAULT"] = ( "green", )
+codes["PROMPT_CHOICE_OTHER"] = ( "red", )
+
+def _parse_color_map(onerror=None):
"""
Parse /etc/portage/color.map and return a dict of error codes.
@return: a dictionary mapping color classes to color codes
"""
myfile = COLOR_MAP_FILE
- ansi_code_pattern = re.compile("^[0-9;]*m$")
- def strip_quotes(token, quotes):
+ ansi_code_pattern = re.compile("^[0-9;]*m$")
+ quotes = '\'"'
+ def strip_quotes(token):
if token[0] in quotes and token[0] == token[-1]:
token = token[1:-1]
return token
try:
- s = shlex.shlex(open(myfile))
- s.wordchars = s.wordchars + ";" # for ansi codes
- while True:
- k, o, v = s.get_token(), s.get_token(), s.get_token()
- if k is s.eof:
- break
- if o != "=":
- e = ParseError("%s%s'%s'" % (
- s.error_leader(myfile, s.lineno),
- "expected '=' operator: ", o))
+ lineno=0
+ for line in codecs.open( myfile, mode = 'r', errors = 'replace' ):
+ lineno += 1
+
+ commenter_pos = line.find("#")
+ line = line[:commenter_pos].strip()
+
+ if len(line) == 0:
+ continue
+
+ split_line = line.split("=")
+ if len(split_line) != 2:
+ e = ParseError("'%s', line %s: %s" % (
+ myfile, lineno,
+ "expected exactly one occurence of '=' operator"))
+ raise e
if onerror:
onerror(e)
else:
raise e
continue
- k = strip_quotes(k, s.quotes)
- v = strip_quotes(v, s.quotes)
- if not k in codes:
- e = ParseError("%s%s'%s'" % (
- s.error_leader(myfile, s.lineno),
+
+ k = strip_quotes(split_line[0].strip())
+ v = strip_quotes(split_line[1].strip())
+ if not k in codes and not k in color_codes:
+ e = ParseError("'%s', line %s: %s'%s'" % (
+ myfile, lineno,
"Unknown variable: ", k))
if onerror:
onerror(e)
raise e
continue
if ansi_code_pattern.match(v):
- codes[k] = esc_seq + v
+ if k in codes:
+ codes[k] = ( esc_seq + v, )
+ elif k in color_codes:
+ color_codes[k] = esc_seq + v
else:
code_list = []
- for x in v.split(" "):
- if x in codes:
- code_list.append(codes[x])
+ for x in v.split():
+ if x in color_codes:
+ if k in codes:
+ code_list.append(x)
+ elif k in color_codes:
+ code_list.append(color_codes[x])
else:
- e = ParseError("%s%s'%s'" % (
- s.error_leader(myfile, s.lineno),
+ e = ParseError("'%s', line %s: %s'%s'" % (
+ myfile, lineno,
"Undefined: ", x))
if onerror:
onerror(e)
else:
raise e
- codes[k] = "".join(code_list)
+ if k in codes:
+ codes[k] = tuple(code_list)
+ elif k in color_codes:
+ color_codes[k] = "".join(code_list)
except (IOError, OSError), e:
if e.errno == errno.ENOENT:
raise FileNotFound(myfile)
raise PermissionDenied(myfile)
raise
-try:
- parse_color_map(onerror=lambda e: writemsg("%s\n" % str(e), noiselevel=-1))
-except FileNotFound:
- pass
-except PermissionDenied, e:
- writemsg("Permission denied: '%s'\n" % str(e), noiselevel=-1)
- del e
-except PortageException, e:
- writemsg("%s\n" % str(e), noiselevel=-1)
- del e
-
def nc_len(mystr):
tmp = re.sub(esc_seq + "^m]+m", "", mystr);
return len(tmp)
havecolor=0
def resetColor():
- return codes["reset"]
+ return color_codes["reset"]
+
+def map_code_to_color_code(code):
+ ret = ""
+ for color_code in codes[code]:
+ # allow stuff that has found it's way through ansi_code_pattern
+ ret += color_codes.get(color_code, color_code)
+ return ret
def colorize(color_key, text):
global havecolor
if havecolor:
- return codes[color_key] + text + codes["reset"]
+ if color_key in color_codes:
+ return color_codes[color_key] + text + color_codes["reset"]
+ elif color_key in codes:
+ return map_code_to_color_code(color_key) + text + color_codes["reset"]
+ else:
+ return text
else:
return text
global havecolor
if havecolor and self._styles:
for style in self._styles:
- self._file.write(codes[style])
+ self._file.write(color_codes[style])
self._file.write(s)
- self._file.write(codes["reset"])
+ self._file.write(color_codes["reset"])
else:
self._file.write(s)
if self.write_listener:
image = image + "[" + (bar_width * "=") + \
">" + ((max_bar_width - bar_width) * " ") + "]"
return image
+
+try:
+ _parse_color_map(onerror=lambda e: writemsg("%s\n" % str(e), noiselevel=-1))
+except FileNotFound:
+ pass
+except PermissionDenied, e:
+ writemsg("Permission denied: '%s'\n" % str(e), noiselevel=-1)
+ del e
+except PortageException, e:
+ writemsg("%s\n" % str(e), noiselevel=-1)
+ del e
projects / portage.git / commitdiff