mirror of
https://github.com/pre-commit/pre-commit.git
synced 2026-01-13 12:30:08 -06:00
51 lines
1.1 KiB
Python
51 lines
1.1 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import os
|
|
import sys
|
|
|
|
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:
|
|
pass
|
|
|
|
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())
|