codekingpro/portable-devtools

1"""2    pygments.formatters.terminal3    ~~~~~~~~~~~~~~~~~~~~~~~~~~~~4 5    Formatter for terminal output with ANSI sequences.6 7    :copyright: Copyright 2006-present by the Pygments team, see AUTHORS.8    :license: BSD, see LICENSE for details.9"""10 11from pygments.formatter import Formatter12from pygments.token import Keyword, Name, Comment, String, Error, \13    Number, Operator, Generic, Token, Whitespace14from pygments.console import ansiformat15from pygments.util import get_choice_opt16 17 18__all__ = ['TerminalFormatter']19 20 21#: Map token types to a tuple of color values for light and dark22#: backgrounds.23TERMINAL_COLORS = {24    Token:              ('',            ''),25 26    Whitespace:         ('gray',   'brightblack'),27    Comment:            ('gray',   'brightblack'),28    Comment.Preproc:    ('cyan',        'brightcyan'),29    Keyword:            ('blue',    'brightblue'),30    Keyword.Type:       ('cyan',        'brightcyan'),31    Operator.Word:      ('magenta',      'brightmagenta'),32    Name.Builtin:       ('cyan',        'brightcyan'),33    Name.Function:      ('green',   'brightgreen'),34    Name.Namespace:     ('_cyan_',      '_brightcyan_'),35    Name.Class:         ('_green_', '_brightgreen_'),36    Name.Exception:     ('cyan',        'brightcyan'),37    Name.Decorator:     ('brightblack',    'gray'),38    Name.Variable:      ('red',     'brightred'),39    Name.Constant:      ('red',     'brightred'),40    Name.Attribute:     ('cyan',        'brightcyan'),41    Name.Tag:           ('brightblue',        'brightblue'),42    String:             ('yellow',       'yellow'),43    Number:             ('blue',    'brightblue'),44 45    Generic.Deleted:    ('brightred',        'brightred'),46    Generic.Inserted:   ('green',  'brightgreen'),47    Generic.Heading:    ('**',         '**'),48    Generic.Subheading: ('*magenta*',   '*brightmagenta*'),49    Generic.Prompt:     ('**',         '**'),50    Generic.Error:      ('brightred',        'brightred'),51 52    Error:              ('_brightred_',      '_brightred_'),53}54 55 56class TerminalFormatter(Formatter):57    r"""58    Format tokens with ANSI color sequences, for output in a text console.59    Color sequences are terminated at newlines, so that paging the output60    works correctly.61 62    The `get_style_defs()` method doesn't do anything special since there is63    no support for common styles.64 65    Options accepted:66 67    `bg`68        Set to ``"light"`` or ``"dark"`` depending on the terminal's background69        (default: ``"light"``).70 71    `colorscheme`72        A dictionary mapping token types to (lightbg, darkbg) color names or73        ``None`` (default: ``None`` = use builtin colorscheme).74 75    `linenos`76        Set to ``True`` to have line numbers on the terminal output as well77        (default: ``False`` = no line numbers).78    """79    name = 'Terminal'80    aliases = ['terminal', 'console']81    filenames = []82 83    def __init__(self, **options):84        Formatter.__init__(self, **options)85        self.darkbg = get_choice_opt(options, 'bg',86                                     ['light', 'dark'], 'light') == 'dark'87        self.colorscheme = options.get('colorscheme', None) or TERMINAL_COLORS88        self.linenos = options.get('linenos', False)89        self._lineno = 090 91    def format(self, tokensource, outfile):92        return Formatter.format(self, tokensource, outfile)93 94    def _write_lineno(self, outfile):95        self._lineno += 196        outfile.write("%s%04d: " % (self._lineno != 1 and '\n' or '', self._lineno))97 98    def _get_color(self, ttype):99        # self.colorscheme is a dict containing usually generic types, so we100        # have to walk the tree of dots.  The base Token type must be a key,101        # even if it's empty string, as in the default above.102        colors = self.colorscheme.get(ttype)103        while colors is None:104            ttype = ttype.parent105            colors = self.colorscheme.get(ttype)106        return colors[self.darkbg]107 108    def format_unencoded(self, tokensource, outfile):109        if self.linenos:110            self._write_lineno(outfile)111 112        for ttype, value in tokensource:113            color = self._get_color(ttype)114 115            for line in value.splitlines(True):116                if color:117                    outfile.write(ansiformat(color, line.rstrip('\n')))118                else:119                    outfile.write(line.rstrip('\n'))120                if line.endswith('\n'):121                    if self.linenos:122                        self._write_lineno(outfile)123                    else:124                        outfile.write('\n')125 126        if self.linenos:127            outfile.write("\n")128