欢迎访问宙启技术站
智能推送

导出精美的代码片段:在Python中使用pygments.formatters.html的HtmlFormatter()进行代码高亮

发布时间:2023-12-14 06:59:51

下面是一个使用pygments.formatters.html的HtmlFormatter()进行代码高亮的Python代码示例:

from pygments import highlight
from pygments.lexers import PythonLexer
from pygments.formatters import HtmlFormatter

# 定义待高亮的Python代码
code = """
def fibonacci(n):
    if n <= 0:
        return []
    elif n == 1:
        return [0]
    elif n == 2:
        return [0, 1]
    else:
        fib_list = [0, 1]
        while len(fib_list) < n:
            fib_list.append(fib_list[-1] + fib_list[-2])
        return fib_list
"""

# 创建HtmlFormatter对象,并指定样式
formatter = HtmlFormatter(style='colorful')

# 使用PythonLexer将代码解析为Token流
lexer = PythonLexer()

# 使用highlight函数进行代码高亮,并使用HtmlFormatter进行格式化
highlighted_code = highlight(code, lexer, formatter)

# 将高亮代码保存到文件中
with open('highlighted_code.html', 'w') as f:
    f.write(highlighted_code)

在上面的示例中,我们首先定义了一个待高亮的Python代码字符串。然后,我们创建了一个HtmlFormatter对象,并指定了样式为'colorful'。接下来,我们使用PythonLexer将代码解析为Token流,并使用highlight函数进行代码高亮。最后,我们将高亮代码保存到了一个名为'highlighted_code.html'的文件中。

以上代码可以实现将Python代码高亮并保存为HTML文件的功能。你可以根据自己的需求,修改代码样式和保存的文件名。通过这个示例,你可以更好地理解如何在Python中使用pygments.formatters.html的HtmlFormatter()进行代码高亮。