Add colors to git cl comments.

git_cl.py

 #!/usr/bin/env python
 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 # Copyright (C) 2008 Evan Martin <martine@danga.com>
 
 """A git-command for integrating reviews on Rietveld."""
 
 import json
 import logging
 import optparse
 import os
 import re
 import stat
 import sys
 import textwrap
 import urlparse
 import urllib2
 
 try:
   import readline  # pylint: disable=F0401,W0611
 except ImportError:
   pass
 
 
+from third_party import colorama
 from third_party import upload
 import breakpad  # pylint: disable=W0611
 import fix_encoding
 import gclient_utils
 import presubmit_support
 import rietveld
 import scm
 import subprocess2
 import watchlists
 
 
 DEFAULT_SERVER = 'https://codereview.appspot.com'
 POSTUPSTREAM_HOOK_PATTERN = '.git/hooks/post-cl-%s'
 DESCRIPTION_BACKUP_FILE = '~/.git_cl_description_backup'
 GIT_INSTRUCTIONS_URL = 'http://code.google.com/p/chromium/wiki/UsingNewGit'
 CHANGE_ID = 'Change-Id:'
 
+# Shortcut since it quickly becomes redundant.
+Fore = colorama.Fore
 
 # Initialized in main()
 settings = None
 
 
 def DieWithError(message):
   print >> sys.stderr, message
   sys.exit(1)
 
 
@@ skipping 1040 matching lines @@
 def CMDcomments(parser, args):
   """show review comments of the current changelist"""
   (_, args) = parser.parse_args(args)
   if args:
     parser.error('Unsupported argument: %s' % args)
 
   cl = Changelist()
   if cl.GetIssue():
     data = cl.RpcServer().get_issue_properties(cl.GetIssue(), True)
     for message in sorted(data['messages'], key=lambda x: x['date']):
-      print '\n%s  %s' % (message['date'].split('.', 1)[0], message['sender'])
+      if message['disapproval']:
+        color = Fore.RED
+      elif message['approval']:
+        color = Fore.GREEN
+      elif message['sender'] == data['owner_email']:
+        color = Fore.MAGENTA
+      else:
+        color = Fore.BLUE
+      print '\n%s%s  %s%s' % (
+          color, message['date'].split('.', 1)[0], message['sender'],
+          Fore.RESET)
       if message['text'].strip():
         print '\n'.join('  ' + l for l in message['text'].splitlines())
   return 0
 
 
 def CreateDescriptionFromLog(args):
   """Pulls out the commit log to use as a base for the CL description."""
   log_args = []
   if len(args) == 1 and not args[0].endswith('.'):
     log_args = [args[0] + '..']
@@ skipping 872 matching lines @@
 
 
 def main(argv):
   """Doesn't parse the arguments here, just find the right subcommand to
   execute."""
   if sys.hexversion < 0x02060000:
     print >> sys.stderr, (
         '\nYour python version %s is unsupported, please upgrade.\n' %
         sys.version.split(' ', 1)[0])
     return 2
+
   # Reload settings.
   global settings
   settings = Settings()
 
   # Do it late so all commands are listed.
   CMDhelp.usage_more = ('\n\nCommands are:\n' + '\n'.join([
       '  %-10s %s' % (fn[3:], Command(fn[3:]).__doc__.split('\n')[0].strip())
       for fn in dir(sys.modules[__name__]) if fn.startswith('CMD')]))
 
   # Create the option parse and add --verbose support.
@@ skipping 26 matching lines @@
         DieWithError(
             ('AppEngine is misbehaving and returned HTTP %d, again. Keep faith '
              'and retry or visit go/isgaeup.\n%s') % (e.code, str(e)))
 
   # Not a known command. Default to help.
   GenUsage(parser, 'help')
   return CMDhelp(parser, argv)
 
 
 if __name__ == '__main__':
+  # These affect sys.stdout so do it outside of main() to simplify mocks in
+  # unit testing.
   fix_encoding.fix_encoding()
+  colorama.init()
   sys.exit(main(sys.argv[1:]))