Implement series patch download and view in django

1 files changed, 11 insertions, 0 deletions

diff --git a/patchtracker/Patch.py b/patchtracker/Patch.py
index 6533910..09f4911 100644
--- a/patchtracker/Patch.py
+++ b/patchtracker/Patch.py
@@ -2,6 +2,9 @@ import sys
 import os
 import errno
 from glob import glob
+import pygments
+import pygments.lexers
+import pygments.formatters
 import tarfile
 
 class Diffstat:
@@ -39,6 +42,14 @@ class Patch:
   def diffstat(self):
     return Diffstat(self)
 
+  def highlight(self):
+    for enc in ['utf-8', 'latin-1']:
+      try:
+        return pygments.highlight(str(self).decode(enc), pygments.lexers.DiffLexer(), 
+                                  pygments.formatters.HtmlFormatter(style='colorful', noclasses=True, encoding=enc, nobackground=True))
+      except UnicodeDecodeError:
+        pass
+
 class GenericPatchSeries (list):
   def blank(self):
     self.names = []