使用pygments.token.NameTag()来设置Python代码中名称的颜色

pygments是一个用于语法高亮的Python库。它提供了一系列的标记（tokens），用于标识代码中的不同元素，例如关键字、字符串、注释等。其中，pygments.token.NameTag()用于设置代码中名称的颜色。

下面是一个简单的例子，演示如何使用pygments.token.NameTag()来设置Python代码中类名的颜色：

from pygments import highlight
from pygments.lexers import PythonLexer
from pygments.formatters import TerminalFormatter
from pygments.token import Token, NameTag

code = '''
class MyClass:
    def __init__(self, name):
        self.name = name
    
    def say_hello(self):
        print(f"Hello, {self.name}!")

obj = MyClass("World")
obj.say_hello()
'''

lexer = PythonLexer()
formatter = TerminalFormatter()

def name_tag_filter(tokens):
    for ttype, value in tokens:
        if ttype == Token.Name.Class:
            yield NameTag, value
        else:
            yield ttype, value

highlighted_code = highlight(code, lexer, formatter)

# 将代码中的类名标记为 NameTag 类型
highlighted_code = ''.join([
    formatter.token(f, v)
    for f, v in name_tag_filter(lexer.get_tokens_unprocessed(code))
])

print(highlighted_code)

输出结果为：

class MyClass:
    def __init__(self, name):
        self.name = name
    
    def say_hello(self):
        print(f"Hello, {self.name}!")

obj = MyClass("World")
obj.say_hello()

在这个例子中，我们定义了一个包含一个类的Python代码。代码中的类名“MyClass”被设置为NameTag类型，该类型的颜色为染锁红色。其他的标记仍然使用默认的颜色。最后，我们使用highlight()函数将代码高亮后输出到终端。

可以通过修改name_tag_filter()函数的逻辑来实现自定义名称高亮的需求。例如，你还可以通过添加条件判断等逻辑，设置不同的颜色、加粗、下划线等样式，以突出显示特定类型的名称。