mirror of
https://github.com/thegeeklab/ansible-later.git
synced 2024-06-29 05:30:59 +02:00
51 lines
961 B
Python
51 lines
961 B
Python
from flake8.formatting import base
|
|
from re import sub
|
|
|
|
|
|
COLORS = dict(
|
|
black=0,
|
|
red=1,
|
|
green=2,
|
|
yellow=3,
|
|
blue=4,
|
|
purple=5,
|
|
cyan=6,
|
|
white=7
|
|
)
|
|
|
|
SUFFIXES = dict(
|
|
fg='0;',
|
|
bold='1;',
|
|
underline='4;',
|
|
bg=''
|
|
)
|
|
|
|
MODS = dict(
|
|
reset='0',
|
|
bold='1',
|
|
underline='4',
|
|
)
|
|
|
|
|
|
class ColorFormatter(base.BaseFormatter):
|
|
name = 'flake8-colors'
|
|
version = '0.1.0'
|
|
|
|
@classmethod
|
|
def parse_options(cls, options):
|
|
options.format = sub(r'\$\{([\w]+)\}', cls._replace, options.format)
|
|
|
|
@classmethod
|
|
def _replace(cls, match):
|
|
return '\x1b[' + cls._parse(match.group(1)) + 'm'
|
|
|
|
@classmethod
|
|
def _parse(cls, var):
|
|
if var in MODS:
|
|
return MODS.get(var)
|
|
|
|
color, _, suffix = var.partition('_')
|
|
if not suffix:
|
|
suffix = 'fg'
|
|
return SUFFIXES.get(suffix) + str(COLORS.get(color) + (30 if suffix in ('fg', 'bold', 'underline') else 40))
|