python - 有没有办法使用 Pygments(或其他库)突出显示函数调用？

Question

我很失望地发现函数调用没有使用 Pygments 突出显示。 .



See it online (我用所有可用的样式对其进行了测试)

内置函数被突出显示，但不是我的。

我看了the tokens list但是例如没有提到“函数调用”或“对象属性”。

我已经考虑通过添加像 \w+\(.*?\) 这样的正则表达式规则来扩展词法分析器。 .但是我害怕由于我没有想到的边缘情况而使错误成倍增加。

你知道为什么这个功能没有直接在 Pygments 内部实现吗？

Answer 1

如果您有要突出显示的特定功能列表，则可以添加自定义突出显示器，例如内置 NumPyLexer :

from pygments.lexers.python import PythonLexer
from pygments.token import Keyword, Name, String

class MyFuncLexer(PythonLexer):
    name = 'MyFuncPython'
    aliases = ['myfuncpython']

    # override the mimetypes to not inherit them from python
    mimetypes = []
    filenames = []

    EXTRA_KEYWORDS = {
        'func'
    }

    def get_tokens_unprocessed(self, text):
        for index, token, value in \
                PythonLexer.get_tokens_unprocessed(self, text):
            if token is Name and value in self.EXTRA_KEYWORDS:
                yield index, Keyword.Pseudo, value
            else:
                yield index, token, value