Python中如何自定义Pygments.formatters的输出格式

Pygments是一个高亮代码库，它提供了许多内置的输出格式（formatters），如HTML、ANSI颜色等。如果内置格式无法满足需求，可以自定义输出格式。

要自定义Pygments.formatters的输出格式，需要定义一个新的Formatter类，并实现以下方法：

- __init__(self, **options): 初始化方法，接收任意数量的关键字参数。

- format(self, tokensource, outfile): 实际执行高亮代码并输出的方法。tokensource参数是一个生成器，产生一个个token，outfile参数是一个可以写入的文件对象。

- get_style_defs(self, **kwargs): 返回包含CSS样式定义的字符串。可以使用这个方法来自定义输出样式。

下面是一个简单的例子，演示如何自定义输出格式。我们将创建一个名为CustomFormatter的自定义格式，它将在每个token前后添加HTML标签。

from pygments.formatters import HtmlFormatter

class CustomFormatter(HtmlFormatter):
    def __init__(self, **options):
        super().__init__(**options)
    
    def format(self, tokensource, outfile):
        outfile.write('<div class="custom-code">
')
        outfile.write('<pre>
')
        for ttype, value in tokensource:
            outfile.write(f'<span class="{self.cssclass(ttype)}">{value}</span>')
        outfile.write('</pre>
')
        outfile.write('</div>
')

css = '<style>' + CustomFormatter().get_style_defs() + '</style>'
code = 'print("Hello, world!")'

formatter = CustomFormatter()
highlighted_code = highlight(code, PythonLexer(), formatter)

output = css + highlighted_code
with open('output.html', 'w') as file:
    file.write(output)

这里我们定义了一个名为CustomFormatter的自定义格式，继承自HtmlFormatter类。在format方法中，我们先在输出文件中添加一些HTML标签来包围整个代码块，然后迭代tokensource生成器，将每个token都用<span>标签包围起来，并指定相应的CSS类名。

一般情况下，还需要调用get_style_defs方法来获取CSS样式定义，并将其添加到输出文件中。在这个例子中，我们使用自动生成的style字符串，组合成完整的HTML文件，并将其写入output.html文件中。

最后运行代码，会生成一个名为output.html的文件，其中包含了自定义格式的高亮代码。

需要注意的是，在实际的应用中，可以根据需要对format方法和CSS样式定义进行更复杂的自定义。这个例子只是一个简单的示例，尽量说明实现自定义输出格式的基本思路。