PygmentsLexer：为Python代码添加自定义的语法高亮规则

Pygments是一个强大的语法高亮工具，支持多种语言，并提供了一个灵活的插件系统，可以轻松添加自定义的语法高亮规则。PygmentsLexer是Pygments中的一个类，可以通过继承它来实现自定义的语法高亮规则。

首先，我们需要创建一个新的类，继承自PygmentsLexer。在这个类中，我们需要重写两个方法：__init__和get_tokens_unprocessed。

from pygments.lexer import RegexLexer
from pygments.token import *

class MyLexer(RegexLexer):
    def __init__(self, **options):
        super().__init__(**options)
        self.tokens = {
            'root': [
                (r'#[^
]*', Comment),
                (r'\d+', Number),
                (r'"[^"]*"', String),
                (r'\b(if|else|while|for)\b', Keyword),
                (r'\b(True|False)\b', Literal),
                (r'\b([A-Z][a-z]*)\b', Name.Class),
                (r'\b[a-z_][a-z0-9_]*\b', Name),
                (r'(\+|\-|\*|\/|\=|\==)', Operator),
                (r'\(|\)|\[|\]|\{|\}', Punctuation),
                (r'\s+', Text),
            ]
        }

    def get_tokens_unprocessed(self, text):
        # 定义一个正则表达式，用来匹配每个标记，并返回标记和其对应的类型
        for index, token, value in RegexLexer.get_tokens_unprocessed(self, text):
           yield index, token, value

在__init__方法里，我们定义了不同类型的语法元素，并为每个语法元素指定一个正则表达式和对应的类型。例如，r'#[^

]*'匹配注释，r'\d+'匹配数字，r'"[^"]*"'匹配字符串等。类型可以是Comment、Number、String、Keyword等，也可以是自定义的类型。

在get_tokens_unprocessed方法里，我们通过RegexLexer.get_tokens_unprocessed方法获取到每个标记的位置、类型和值，并将它们作为生成器的返回值。这样，在使用PygmentsLexer时，就可以通过调用get_tokens方法来实现自定义的语法高亮。

下面是一个简单的使用例子：

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

# 创建一个自定义的Lexer
lexer = MyLexer()

# 从Lexer创建一个PygmentsLexer
pygments_lexer = get_lexer_by_name('python', stripall=True)
pygments_lexer.add_filter('my_lexer', lexer)

# 要高亮的代码
code = '''
# This is a comment
x = 10
while x > 0:
    print(x)
    x -= 1
'''

# 使用PygmentsLexer进行语法高亮
highlighted_code = highlight(code, pygments_lexer, TerminalFormatter())

# 输出高亮后的代码
print(highlighted_code)

在上面的例子中，我们首先创建了一个自定义的Lexer（MyLexer），然后通过get_lexer_by_name方法创建了一个PygmentsLexer，然后使用add_filter方法将自定义的Lexer添加到PygmentsLexer中。最后，调用highlight方法将要高亮的代码和PygmentsLexer作为参数传入，并使用TerminalFormatter作为输出格式，将高亮的代码打印到控制台上。

自定义语法高亮规则可以根据实际需求进行调整和扩展，例如添加更多的关键字、类型和操作符等。PygmentsLexer的强大功能可以帮助我们轻松地实现自定义的语法高亮。