Using pygments to highlight source code to html

highlight_code.py

from pygments import highlight
from pygments.lexers import guess_lexer
from pygments.formatters.html import HtmlFormatter

def highlight_code(src, style='manni'):
    """This function highlight :param:`src` into html format using the style provided.

    :param src: The source code to highlight.
    :param style: The color style, valid style to see:

        from pygments.styles import STYLE_MAP
        valid_style = [k for k in STYLE_MAP]
        print valid_style

    :return: The highlighted code in html format.
    """
    opts = {'style': style, 'full': True}
    lexer = guess_lexer(src, **opts)
    formatter = HtmlFormatter(**opts)
    return highlight(src, lexer, formatter)