You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
156 lines
4.7 KiB
156 lines
4.7 KiB
6 years ago
|
# -*- coding: utf-8 -*-
|
||
|
|
"""
|
||
|
|
pygments.style
|
||
|
|
~~~~~~~~~~~~~~
|
||
|
|
|
||
|
|
Basic style object.
|
||
|
|
|
||
|
|
:copyright: Copyright 2006-2017 by the Pygments team, see AUTHORS.
|
||
|
|
:license: BSD, see LICENSE for details.
|
||
|
|
"""
|
||
|
|
|
||
|
|
from pygments.token import Token, STANDARD_TYPES
|
||
|
|
from pygments.util import add_metaclass
|
||
|
|
|
||
|
|
# Default mapping of #ansixxx to RGB colors.
|
||
|
|
_ansimap = {
|
||
|
|
# dark
|
||
|
|
'#ansiblack': '000000',
|
||
|
|
'#ansidarkred': '7f0000',
|
||
|
|
'#ansidarkgreen': '007f00',
|
||
|
|
'#ansibrown': '7f7fe0',
|
||
|
|
'#ansidarkblue': '00007f',
|
||
|
|
'#ansipurple': '7f007f',
|
||
|
|
'#ansiteal': '007f7f',
|
||
|
|
'#ansilightgray': 'e5e5e5',
|
||
|
|
# normal
|
||
|
|
'#ansidarkgray': '555555',
|
||
|
|
'#ansired': 'ff0000',
|
||
|
|
'#ansigreen': '00ff00',
|
||
|
|
'#ansiyellow': 'ffff00',
|
||
|
|
'#ansiblue': '0000ff',
|
||
|
|
'#ansifuchsia': 'ff00ff',
|
||
|
|
'#ansiturquoise': '00ffff',
|
||
|
|
'#ansiwhite': 'ffffff',
|
||
|
|
}
|
||
|
|
ansicolors = set(_ansimap)
|
||
|
|
|
||
|
|
|
||
|
|
class StyleMeta(type):
|
||
|
|
|
||
|
|
def __new__(mcs, name, bases, dct):
|
||
|
|
obj = type.__new__(mcs, name, bases, dct)
|
||
|
|
for token in STANDARD_TYPES:
|
||
|
|
if token not in obj.styles:
|
||
|
|
obj.styles[token] = ''
|
||
|
|
|
||
|
|
def colorformat(text):
|
||
|
|
if text in ansicolors:
|
||
|
|
return text
|
||
|
|
if text[0:1] == '#':
|
||
|
|
col = text[1:]
|
||
|
|
if len(col) == 6:
|
||
|
|
return col
|
||
|
|
elif len(col) == 3:
|
||
|
|
return col[0]*2 + col[1]*2 + col[2]*2
|
||
|
|
elif text == '':
|
||
|
|
return ''
|
||
|
|
assert False, "wrong color format %r" % text
|
||
|
|
|
||
|
|
_styles = obj._styles = {}
|
||
|
|
|
||
|
|
for ttype in obj.styles:
|
||
|
|
for token in ttype.split():
|
||
|
|
if token in _styles:
|
||
|
|
continue
|
||
|
|
ndef = _styles.get(token.parent, None)
|
||
|
|
styledefs = obj.styles.get(token, '').split()
|
||
|
|
if not ndef or token is None:
|
||
|
|
ndef = ['', 0, 0, 0, '', '', 0, 0, 0]
|
||
|
|
elif 'noinherit' in styledefs and token is not Token:
|
||
|
|
ndef = _styles[Token][:]
|
||
|
|
else:
|
||
|
|
ndef = ndef[:]
|
||
|
|
_styles[token] = ndef
|
||
|
|
for styledef in obj.styles.get(token, '').split():
|
||
|
|
if styledef == 'noinherit':
|
||
|
|
pass
|
||
|
|
elif styledef == 'bold':
|
||
|
|
ndef[1] = 1
|
||
|
|
elif styledef == 'nobold':
|
||
|
|
ndef[1] = 0
|
||
|
|
elif styledef == 'italic':
|
||
|
|
ndef[2] = 1
|
||
|
|
elif styledef == 'noitalic':
|
||
|
|
ndef[2] = 0
|
||
|
|
elif styledef == 'underline':
|
||
|
|
ndef[3] = 1
|
||
|
|
elif styledef == 'nounderline':
|
||
|
|
ndef[3] = 0
|
||
|
|
elif styledef[:3] == 'bg:':
|
||
|
|
ndef[4] = colorformat(styledef[3:])
|
||
|
|
elif styledef[:7] == 'border:':
|
||
|
|
ndef[5] = colorformat(styledef[7:])
|
||
|
|
elif styledef == 'roman':
|
||
|
|
ndef[6] = 1
|
||
|
|
elif styledef == 'sans':
|
||
|
|
ndef[7] = 1
|
||
|
|
elif styledef == 'mono':
|
||
|
|
ndef[8] = 1
|
||
|
|
else:
|
||
|
|
ndef[0] = colorformat(styledef)
|
||
|
|
|
||
|
|
return obj
|
||
|
|
|
||
|
|
def style_for_token(cls, token):
|
||
|
|
t = cls._styles[token]
|
||
|
|
ansicolor = bgansicolor = None
|
||
|
|
color = t[0]
|
||
|
|
if color.startswith('#ansi'):
|
||
|
|
ansicolor = color
|
||
|
|
color = _ansimap[color]
|
||
|
|
bgcolor = t[4]
|
||
|
|
if bgcolor.startswith('#ansi'):
|
||
|
|
bgansicolor = bgcolor
|
||
|
|
bgcolor = _ansimap[bgcolor]
|
||
|
|
|
||
|
|
return {
|
||
|
|
'color': color or None,
|
||
|
|
'bold': bool(t[1]),
|
||
|
|
'italic': bool(t[2]),
|
||
|
|
'underline': bool(t[3]),
|
||
|
|
'bgcolor': bgcolor or None,
|
||
|
|
'border': t[5] or None,
|
||
|
|
'roman': bool(t[6]) or None,
|
||
|
|
'sans': bool(t[7]) or None,
|
||
|
|
'mono': bool(t[8]) or None,
|
||
|
|
'ansicolor': ansicolor,
|
||
|
|
'bgansicolor': bgansicolor,
|
||
|
|
}
|
||
|
|
|
||
|
|
def list_styles(cls):
|
||
|
|
return list(cls)
|
||
|
|
|
||
|
|
def styles_token(cls, ttype):
|
||
|
|
return ttype in cls._styles
|
||
|
|
|
||
|
|
def __iter__(cls):
|
||
|
|
for token in cls._styles:
|
||
|
|
yield token, cls.style_for_token(token)
|
||
|
|
|
||
|
|
def __len__(cls):
|
||
|
|
return len(cls._styles)
|
||
|
|
|
||
|
|
|
||
|
|
@add_metaclass(StyleMeta)
|
||
|
|
class Style(object):
|
||
|
|
|
||
|
|
#: overall background color (``None`` means transparent)
|
||
|
|
background_color = '#ffffff'
|
||
|
|
|
||
|
|
#: highlight background color
|
||
|
|
highlight_color = '#ffffcc'
|
||
|
|
|
||
|
|
#: Style definitions for individual token types.
|
||
|
|
styles = {}
|