Class BPythonFormatter

bpython/formatter.py:93–133

This is the custom formatter for bpython. Its format() method receives the tokensource and outfile params passed to it from the Pygments highlight() method and slops them into the appropriate format string as defined above, then writes to the outfile object the final formatte

class BPythonFormatter(Formatter):
    """This is the custom formatter for bpython.
    Its format() method receives the tokensource
    and outfile params passed to it from the
    Pygments highlight() method and slops
    them into the appropriate format string
    as defined above, then writes to the outfile
    object the final formatted string.

    See the Pygments source for more info; it's pretty
    straightforward."""

    def __init__(
        self, color_scheme: MutableMapping[str, str], **options: Any
    ) -> None:
        self.f_strings = {}
        for k, v in theme_map.items():
            self.f_strings[k] = f"\x01{color_scheme[v]}"
            if k is Parenthesis:
                # FIXME: Find a way to make this the inverse of the current
                # background colour
                self.f_strings[k] += "I"
        super().__init__(**options)

    def format(
        self,
        tokensource: Iterable[MutableMapping[_TokenType, str]],
        outfile: TextIO,
    ) -> None:
        o: str = ""
        for token, text in tokensource:
            if text == "\n":
                continue

            while token not in self.f_strings:
                if token.parent is None:
                    break
                else:
                    token = token.parent
            o += f"{self.f_strings[token]}\x03{text}\x04"
        outfile.write(o.rstrip())


# vim: sw=4 ts=4 sts=4 ai et