39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
from django import template
|
|
from django.template.defaultfilters import stringfilter
|
|
from django.utils.safestring import mark_safe
|
|
import logging
|
|
from pygments import highlight
|
|
from pygments.lexers import get_lexer_by_name
|
|
from pygments.formatters import HtmlFormatter
|
|
from gergelypolonkaiweb.solarized_dark import SolarizedDarkStyle
|
|
|
|
register = template.Library()
|
|
|
|
class CodeFormatter(HtmlFormatter):
|
|
def wrap(self, source, outfile):
|
|
return self._wrap_code(source)
|
|
|
|
def _wrap_code(self, source):
|
|
yield 0, "<pre class=\"" + self.cssclass + "\"><ol>"
|
|
for i, t in source:
|
|
if i == 1:
|
|
t = "<li>" + t + "</li>"
|
|
|
|
yield i, t
|
|
yield 0, "</ol></pre>"
|
|
|
|
def hilite(language, code):
|
|
lexer = get_lexer_by_name(language)
|
|
formatter = CodeFormatter(style = SolarizedDarkStyle, linenos = False, cssclass = language + " code", noclasses = True)
|
|
return highlight(code, lexer, formatter)
|
|
|
|
@register.filter(needs_autoescape=True)
|
|
@stringfilter
|
|
def syhilite(value, language, autoescape=None):
|
|
if language == "php":
|
|
value = "<?php\n" + value
|
|
|
|
html = hilite(language, value)
|
|
return mark_safe(html)
|
|
|