Colorize output to terminal

supervisor_stdout.py

@@ -1,5 +1,35 @@
+import os
 import sys
 
+COLORS = tuple(range(30, 38))
+
+def get_color(key):
+    use_string = key if isinstance(key, basestring) else str(key)
+    return COLORS[use_string.__hash__() % len(COLORS)]
+
+def supports_color():
+    """
+    Taken from https://github.com/django/django/blob/master/django/core/management/color.py
+
+    Returns True if the running system's terminal supports color, and False
+    otherwise.
+    """
+
+    if os.getenv("STDOUT_ALWAYS_ENABLE_COLOR_OUTPUT"):
+        return True
+
+    if os.getenv("STDOUT_DISABLE_COLOR_OUTPUT"):
+        return False
+
+    plat = sys.platform
+    supported_platform = plat != 'Pocket PC' and (plat != 'win32' or
+                                                  'ANSICON' in os.environ)
+    # isatty is not always implemented, #6223.
+    is_a_tty = hasattr(sys.stdout, 'isatty') and sys.stdout.isatty()
+    if not supported_platform or not is_a_tty:
+        return False
+    return True
+
 def write_stdout(s):
     sys.stdout.write(s)
     sys.stdout.flush()
@@ -19,7 +49,14 @@ def main():
 def event_handler(event, response):
     line, data = response.split('\n', 1)
     headers = dict([ x.split(':') for x in line.split() ])
-    print '%s %s | %s'%(headers['processname'], headers['channel'], data),
+    if supports_color():
+        print '\033[1;%sm%s\033[1;m | \033[1;%sm%s\033[1;m | %s' % (get_color(headers['processname']),
+                                                                  headers['processname'],
+                                                                  get_color(headers['channel']),
+                                                                  headers['channel'],
+                                                                  data)
+    else:
+        print '%s | %s | %s'%(headers['processname'], headers['channel'], data)
 
 if __name__ == '__main__':
     main()