diff options
Diffstat (limited to 'src/lib/Bcfg2/Reporting/templatetags/syntax_coloring.py')
-rw-r--r-- | src/lib/Bcfg2/Reporting/templatetags/syntax_coloring.py | 48 |
1 files changed, 48 insertions, 0 deletions
diff --git a/src/lib/Bcfg2/Reporting/templatetags/syntax_coloring.py b/src/lib/Bcfg2/Reporting/templatetags/syntax_coloring.py new file mode 100644 index 000000000..2712d6395 --- /dev/null +++ b/src/lib/Bcfg2/Reporting/templatetags/syntax_coloring.py @@ -0,0 +1,48 @@ +import sys +from django import template +from django.utils.encoding import smart_unicode +from django.utils.html import conditional_escape +from django.utils.safestring import mark_safe + +from Bcfg2.Compat import u_str + +register = template.Library() + +# pylint: disable=E0611 +try: + from pygments import highlight + from pygments.lexers import get_lexer_by_name + from pygments.formatters import HtmlFormatter + colorize = True +except: + colorize = False +# pylint: enable=E0611 + + +@register.filter +def syntaxhilight(value, arg="diff", autoescape=None): + """ + Returns a syntax-hilighted version of Code; + requires code/language arguments + """ + + if autoescape: + # Seems to cause a double escape + #value = conditional_escape(value) + arg = conditional_escape(arg) + + if colorize: + try: + output = u_str('<style type="text/css">') \ + + smart_unicode(HtmlFormatter().get_style_defs('.highlight')) \ + + u_str('</style>') + + lexer = get_lexer_by_name(arg) + output += highlight(value, lexer, HtmlFormatter()) + return mark_safe(output) + except: + return value + else: + return mark_safe(u_str('<div class="note-box">Tip: Install pygments ' + 'for highlighting</div><pre>%s</pre>') % value) +syntaxhilight.needs_autoescape = True |