summaryrefslogtreecommitdiff
path: root/highlighting.py
blob: 439ae78fa17119ba4ed16fa9a39063e54fe3dc4a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
from pygments import highlight as pygHighlight
from pygments.token import Name
from pygments.lexers import get_lexer_by_name, BashLexer
from pygments.formatters import HtmlFormatter

formatter = HtmlFormatter(encoding = "utf-8", classprefix="pygm", nowrap=True)

BashLexer.tokens["basic"].append((r'\b(emerge|layman|git(?!:))\s*\b(?!\.)', Name.Builtin))

def highlight(code, lang):
    l = get_lexer_by_name(lang, encoding = "utf-8")
    return pygHighlight(code, l, formatter)

def get_css():
    return """<style type="text/css">
%s
</style>""" % formatter.get_style_defs()