Adding support for colorized JSON output

awscli/formatter.py

@@ -85,15 +85,37 @@ def __call__(self, command_name, response, stream=None):
 
 class JSONFormatter(FullyBufferedFormatter):
 
+    def __init__(self, args):
+        super(JSONFormatter, self).__init__(args)
+
+        self.colorize = False
+        try:
+            if args.color == 'on':
+                self.colorize = True
+        except:
+            pass
+
     def _format_response(self, command_name, response, stream):
         # For operations that have no response body (e.g. s3 put-object)
         # the response will be an empty string.  We don't want to print
         # that out to the user but other "falsey" values like an empty
         # dictionary should be printed.
         if response != {}:
-            json.dump(response, stream, indent=4, default=json_encoder,
-                    ensure_ascii=False)
-            stream.write('\n')
+            if self.colorize:
+                from pygments import highlight, lexers, formatters
+                from StringIO import StringIO
+                io = StringIO()
+                json.dump(response, io, indent=4, default=json_encoder,
+                        ensure_ascii=False)
+                colorful_json = highlight(io.getvalue(), lexers.JsonLexer(), formatters.TerminalFormatter())
+
+                stream.write(colorful_json)
+                stream.write('\n')
+            else:
+                json.dump(response, stream, indent=4, default=json_encoder,
+                        ensure_ascii=False)
+
+                stream.write('\n')
 
 
 class TableFormatter(FullyBufferedFormatter):

requirements.txt

@@ -8,6 +8,7 @@ docutils>=0.10
 -e git://github.com/boto/jmespath.git@develop#egg=jmespath
 nose==1.3.0
 colorama>=0.2.5,<=0.3.7
+pygments==2.1.3
 mock==1.3.0
 rsa>=3.1.2,<=3.5.0
 wheel==0.24.0

setup.cfg

@@ -6,6 +6,7 @@ universal = 1
 requires-dist =
         botocore==1.4.48
         colorama>=0.2.5,<=0.3.7
+        pygments==2.1.3
         docutils>=0.10
         rsa>=3.1.2,<=3.5.0
         s3transfer>=0.1.0,<0.2.0

setup.py

@@ -8,6 +8,7 @@
 
 requires = ['botocore==1.4.48',
             'colorama>=0.2.5,<=0.3.7',
+            'pygments==2.1.3',
             'docutils>=0.10',
             'rsa>=3.1.2,<=3.5.0',
             's3transfer>=0.1.0,<0.2.0']

tests/unit/customizations/datapipeline/test_commands.py

@@ -173,6 +173,7 @@ def __init__(self, **kwargs):
         self.region = None
         self.verify_ssl = None
         self.output = None
+        self.color = None
         self.query = None
         self.__dict__.update(kwargs)