Add highlighting support

1 files changed, 16 insertions, 0 deletions

diff --git a/highlighting.py b/highlighting.py
new file mode 100644
index 0000000..9308b84
--- /dev/null
+++ b/highlighting.py
@@ -0,0 +1,16 @@
+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 ''