Ticket #6385: colorize.diff

django/core/servers/basehttp.py

@@ -6 +6 @@
 This is a simple server for use in testing or debugging Django apps. It hasn't
 been reviewed for security issues. Don't use it for production use.
 """
-
+from django.core.management.color import color_style
 from BaseHTTPServer import BaseHTTPRequestHandler, HTTPServer
 import mimetypes
 import os
@@ -443 +443 @@
 
     def close(self):
         try:
-            self.request_handler.log_request(self.status.split(' ',1)[0], self.bytes_sent)
+            style = color_style()
+            status_code = self.status.split(' ',1)[0]
+            if status_code.startswith('2'):
+                status_code = style.SUCCESS(status_code)
+            elif status_code.startswith('3'):
+                status_code = style.REDIRECT(status_code)
+            elif status_code.startswith('4'):
+                status_code = style.CLIENT_ERROR(status_code)
+            elif status_code.startswith('5'):
+                status_code = style.SERVER_ERROR(status_code)
+            self.request_handler.log_request(status_code, self.bytes_sent)
         finally:
             try:
                 if hasattr(self.result,'close'):

django/core/management/color.py

@@ -20 +20 @@
     style.SQL_COLTYPE = termcolors.make_style(fg='green')
     style.SQL_KEYWORD = termcolors.make_style(fg='yellow')
     style.SQL_TABLE = termcolors.make_style(opts=('bold',))
+    # HTTP STATUS CODES
+    style.SUCCESS = termcolors.make_style(fg='green', opts=('bold',)) # 2xx
+    style.REDIRECT = termcolors.make_style(fg='blue', opts=('bold',)) # 3xx
+    style.CLIENT_ERROR = termcolors.make_style(fg='yellow', opts=('bold',)) # 4xx
+    style.SERVER_ERROR = termcolors.make_style(fg='red', opts=('bold',)) # 5xx
     return style
 
 def no_style():