simpler, faster color code

papers/color.py

@@ -1,60 +1,24 @@
 # display
 
-BOLD = '\033[1m'
-END  = '\033[0m'
-
-COLORS = {
-    'black' : '\033[0;30m',
-    'red'   : '\033[0;31m',
-    'green' : '\033[0;32m',
-    'yellow': '\033[0;33m',
-    'blue'  : '\033[0;34m',
-    'purple': '\033[0;35m',
-    'cyan'  : '\033[0;36m',
-    'grey'  : '\032[0;37m',
-    }
-
-# Bold
-BCOLORS = {
-    'black' : '\033[1;30m',
-    'red'   : '\033[1;31m',
-    'green' : '\033[1;32m',
-    'yellow': '\033[1;33m',
-    'blue'  : '\033[1;34m',
-    'purple': '\033[1;35m',
-    'cyan'  : '\033[1;36m',
-    'grey'  : '\033[1;37m',
-    }
-
-# application specific
-ALIASES = {
-    'ok'      : 'green',
-    'error'   : 'red',
-    'normal'  : 'grey',
-    'citekey' : 'purple',
-    'filepath': 'cyan',
-    }
-
-
-def colored(s, color=None, bold=False):
-    if color in ALIASES:
-        color = ALIASES[color]
-    try:
-        if bold:
-            color_code = BCOLORS[color]
-        else:
-            color_code = COLORS[color]
-    except KeyError:
-        if bold:
-            color_code = BOLD
-        else:
-            color_code = ''
-    if color_code != '':
-        end_code = END
-    else:
-        end_code = ''
-    return color_code + s + end_code
-
-
-def not_colored(s, **kwargs):
-    return s
+bold = '\033[1m'
+end  = '\033[0m'
+
+black  = '\033[0;30m'
+red    = '\033[0;31m'
+green  = '\033[0;32m'
+yellow = '\033[0;33m'
+blue   = '\033[0;34m'
+purple = '\033[0;35m'
+cyan   = '\033[0;36m'
+grey   = '\033[0;37m'
+
+ok       = green
+error    = red
+normal   = grey
+citekey  = purple
+filepath = cyan
+
+def colored(s, color=end, bold=False):
+    if bold:
+        s = '\033[1' + s[3:]
+    return color + s + end