1. Miki Tebeka
  2. highlight


highlight / highlighter.py

#!/usr/bin/env python

from cherrypy import expose, quickstart, tools
from pygments import highlight
from pygments.lexers import get_lexer_by_name, guess_lexer
from pygments.formatters import HtmlFormatter

class Highlighter:
    def index(self, **kw):
        with open("index.html") as fo:
            return fo.read()

    def highlight(self, **kw):
        code = kw.get("code", "")
        if not code:
            return ""

        lang = kw.get("lang", "").lower()
        if not lang:
            lexer = guess_lexer(code)
            lexer = get_lexer_by_name(lang)

        linenos = kw.get("linenos", 0)

        formatter = HtmlFormatter(linenos=linenos, cssclass="source")
        return highlight(code, lexer, formatter)

    @tools.response_headers(headers=[("Content-Type", "text/css")])
    def css(self, **kw):
        return HtmlFormatter().get_style_defs()

    @tools.response_headers(headers=[("Content-Type", "application/json")])
    def js(self, **kw):
        with open("jquery.js") as fo:
            jquery = fo.read()
        with open("highlight.js") as fo:
            script = fo.read()

        return "%s\n%s" % (jquery, script)

    def demo(self, **kw):
        with open("demo.html") as fo:
            return fo.read()

if __name__ == "__main__":