使用pygments.styles模块在Python中自定义代码语法高亮样式

Pygments是一个用Python编写的语法高亮工具，可以让你将代码展示得更加美观和易读。Pygments.styles模块提供了一些预定义的语法高亮样式，但你也可以通过继承pygments.style.Style类并重写其中的方法来自定义自己的样式。

下面是一个自定义代码语法高亮样式的例子。假设我们要定义一个名为"DarkThemeStyle"的暗黑主题样式：

from pygments.style import Style
from pygments.token import (
    Keyword, Name, Comment, String, Error,
    Number, Operator, Generic, Token
)
from pygments.styles.default import DefaultStyle


class DarkThemeStyle(DefaultStyle):
    # 自定义颜色
    background_color = '#1E1E1E'
    foreground_color = '#F8F8F2'
    comment_color = '#5D5D5D'
    keyword_color = '#F92672'
    number_color = '#AE81FF'
    string_color = '#E6DB74'

    # 自定义Token类型
    styles = {
        Comment: comment_color,
        Keyword: keyword_color,
        Number: number_color,
        String: string_color,
    }

在上述代码中，我们继承了DefaultStyle类，并通过重写styles属性来自定义我们的样式。我们将背景色设为"#1E1E1E"，前景色设为"#F8F8F2"，注释的颜色为"#5D5D5D"，关键字的颜色为"#F92672"，数字的颜色为"#AE81FF"，字符串的颜色为"#E6DB74"。

接下来我们可以使用这个自定义的样式来高亮代码。下面是一个使用的例子：

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

# 创建自定义样式实例
dark_theme_style = DarkThemeStyle()

# 对代码进行语法高亮
code = '''
def fibonacci(n):
    if n <= 1:
        return n
    else:
        return fibonacci(n-1) + fibonacci(n-2)
print(fibonacci(10))
'''

highlighted_code = highlight(code, PythonLexer(), TerminalFormatter(style=dark_theme_style))

# 打印高亮后的代码
print(highlighted_code)

在这个例子中，我们首先创建了自定义样式的实例dark_theme_style。然后我们使用highlight函数对给定的代码进行语法高亮，指定使用PythonLexer进行词法分析，使用TerminalFormatter输出高亮后的结果，并指定使用我们定义的自定义样式。

运行上述代码，你会在控制台中看到高亮后的代码，它使用了我们自定义的暗黑主题样式。

通过使用pygments.styles模块，我们可以方便地自定义代码语法高亮样式，使其更符合我们的个人喜好或项目需求。你可以根据自己的需要定义不同的Token类型和颜色，并使用highlight函数将其应用到代码上。