Enhance pygemntize

chhsiao90 · Sep 13, 2016 · 717499a · 717499a
1 parent d9df437
commit 717499a
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/gviewer/util.py b/gviewer/util.py
@@ -7,9 +7,16 @@
 
 
 def pygmentize(content, lexer):
+    def trim(pygment_entry):
+        token, value = pygment_entry
+        if token is pygments.token.Token.Text and value.startswith(u"\n"):
+            return (token, value.rstrip(u" "))
+        return pygment_entry
+
     if not pygments:  # pragma: no cover
         raise ImportError("no pygments")
     pygments_list = list(pygments.lex(content, lexer))
+    pygments_list = map(trim, pygments_list)
     return _join(pygments_list, (pygments.token.Token.Text, u"\n"))