Fall back to guess the lexer by content

If we can't guess the lexer by the file name, try to guess based on the
content.

This allows pygments to colorize extension-less files, usually scripts.

Signed-off-by: Alberto Bertogli <albertito@blitiri.com.ar>

1 files changed, 6 insertions, 2 deletions

diff --git a/utils.py b/utils.py
index cee4bb7..039d02b 100644
--- a/utils.py
+++ b/utils.py
@@ -50,9 +50,13 @@ def colorize_diff(s):
 
 def colorize_blob(fname, s):
     try:
-        lexer = lexers.guess_lexer_for_filename(fname, s)
+        lexer = lexers.guess_lexer_for_filename(fname, s, encoding = 'utf-8')
     except lexers.ClassNotFound:
-        lexer = lexers.TextLexer(encoding = 'utf-8')
+        try:
+            lexer = lexers.guess_lexer(s[:200], encoding = 'utf-8')
+        except lexers.ClassNotFound:
+            lexer = lexers.TextLexer(encoding = 'utf-8')
+
     formatter = HtmlFormatter(encoding = 'utf-8',
                     cssclass = 'source_code',
                     linenos = 'table')