pangopygments.py

#!/usr/bin/python
# -*- coding:utf8 -*-
"""
A pygments formatter for Pango text markup.

Written by Soheil Hasss Yeganeh, 2014.
Modified by Thiemo Leonhardt, 2021
Modified by guruthree, 2023
Not copyrighted, in public domain.
"""

import pygments
from pygments import lexers
from pygments.formatter import Formatter
from pygments.styles import get_style_by_name


class PangoFormatter(Formatter):
    """ Based on the HTML 3.2 formatter in pygments:
      http://pygments.org/docs/formatterdevelopment/ """

    def __init__(self, **options):
        Formatter.__init__(self, **options)

        self.styles = {}

        for token, style in self.style:
            start_tag = close_tag = ''

            if style['color']:
                start_tag += '<span fgcolor="#%s">' % style['color']
                close_tag = '</span>' + close_tag

            if style['bold']:
                start_tag += '<b>'
                close_tag = '</b>' + close_tag

            if style['italic']:
                start_tag += '<i>'
                close_tag = '</i>' + close_tag

            if style['underline']:
                start_tag += '<u>'
                close_tag = '</u>' + close_tag

            self.styles[token] = (start_tag, close_tag)

    def format(self, tokensource, outfile):
        lastval = ''
        lasttype = None

        for ttype, value in tokensource:
            while ttype not in self.styles:
                ttype = ttype.parent

            if ttype == lasttype:
                lastval += value
            else:
                if lastval:
                    stylebegin, styleend = self.styles[lasttype]
                    outfile.write(stylebegin + lastval + styleend)

                lastval = value
                lasttype = ttype

        if lastval:
            stylebegin, styleend = self.styles[lasttype]
            outfile.write(stylebegin + lastval + styleend)


def highlight(snippet, lang, style):
    """ snippet is the string of code snippets, and lang, the language name. """
    # The highlighter highlights (i.e., adds tags around) operators
    # (& and ;, here), so let's use a non-highlighted keyword, and escape them
    # after highlighting.
    snippet = snippet.replace("&", "__AMP__").replace("<", "__LT__")

    # Pygments messes up initial and final newlines; fix up
    begin = ''
    if snippet[:1] == '\n':
        begin = '\n'
        snippet = snippet[1:]
    end = ''
    if snippet[-1:] == '\n':
        end = '\n'
        snippet = snippet[:-1]

    if lang == 'auto':
        L = lexers.guess_lexer(snippet)
    else:
        try:
            L = lexers.get_lexer_by_name(lang)
        except pygments.util.ClassNotFound:
            print("Language %s is not supported." % lang)
            L = None

    S = get_style_by_name(style)

    if L is not None and S is not None:
        print('Using languge: %s' % L.name)
        print('Using style: %s' % style)
        snippet = pygments.highlight(snippet, L, PangoFormatter(style=S))

    if snippet[0] == '\n' and begin == '\n':
        begin = ''
    if snippet[-1] == '\n' and end == '\n':
        end = ''

    snippet = snippet.replace("__AMP__", "&amp;").replace("__LT__", "&lt;").replace("\\", "\\\\")

    return begin + snippet + end


if __name__ == '__main__':
    code = 'print "Hello World"'
    print(highlight(code, 'py'))