33 lines
780 B
Python
33 lines
780 B
Python
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
from prompt_toolkit.styles.pygments import pygments_token_to_classname
|
|
|
|
from .base import StyleAndTextTuples
|
|
|
|
if TYPE_CHECKING:
|
|
from pygments.token import Token
|
|
|
|
__all__ = [
|
|
"PygmentsTokens",
|
|
]
|
|
|
|
|
|
class PygmentsTokens:
|
|
"""
|
|
Turn a pygments token list into a list of prompt_toolkit text fragments
|
|
(``(style_str, text)`` tuples).
|
|
"""
|
|
|
|
def __init__(self, token_list: list[tuple[Token, str]]) -> None:
|
|
self.token_list = token_list
|
|
|
|
def __pt_formatted_text__(self) -> StyleAndTextTuples:
|
|
result: StyleAndTextTuples = []
|
|
|
|
for token, text in self.token_list:
|
|
result.append(("class:" + pygments_token_to_classname(token), text))
|
|
|
|
return result
|