from django import template
from pygments import highlight
from pygments.formatters import HtmlFormatter
from pygments.lexers import get_lexer_by_name

register = template.Library()


@register.filter
def pygmentize(text):
    return highlight(
        text,
        get_lexer_by_name("diff", encoding="UTF-8"),
        HtmlFormatter(
            linenos="table", anchorlinenos=True, lineanchors="L", linespans="L"
        ),
    )