pre-commit/pre_commit/color.py at wip_progress_dots · pre-commit/pre-commit · GitHub

Repository navigation

History

from __future__ import unicode_literals

import os

import sys

terminal_supports_color = True

if os.name == 'nt': # pragma: no cover (windows)

from pre_commit.color_windows import enable_virtual_terminal_processing

try:

enable_virtual_terminal_processing()

except WindowsError:

terminal_supports_color = False

RED = '\033[41m'

GREEN = '\033[42m'

YELLOW = '\033[43;30m'

TURQUOISE = '\033[46;30m'

NORMAL = '\033[0m'

class InvalidColorSetting(ValueError):

pass

def format_color(text, color, use_color_setting):

"""Format text with color.

Args:

text - Text to be formatted with color if `use_color`

color - The color start string

use_color_setting - Whether or not to color

"""

if not use_color_setting:

return text

else:

return '{}{}{}'.format(color, text, NORMAL)

COLOR_CHOICES = ('auto', 'always', 'never')

def use_color(setting):

"""Choose whether to use color based on the command argument.

Args:

setting - Either `auto`, `always`, or `never`

"""

if setting not in COLOR_CHOICES:

raise InvalidColorSetting(setting)

return (

setting == 'always' or (

setting == 'auto' and

sys.stdout.isatty() and

terminal_supports_color and

os.getenv('TERM') != 'dumb'

)

pre-commit/pre_commit/color.py at wip_progress_dots · pre-commit/pre-commit · GitHub

Repository navigation

Expand file tree

Breadcrumbs

color.py

Latest commit

History

Breadcrumbs

color.py

File metadata and controls