from pygments import highlight as pygHighlight
from pygments.lexers import get_lexer_by_name
from pygments.formatters import HtmlFormatter

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

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

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