diff options
Diffstat (limited to 'pre_commit/color.py')
-rw-r--r-- | pre_commit/color.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/pre_commit/color.py b/pre_commit/color.py index eb906b7..4ddfdf5 100644 --- a/pre_commit/color.py +++ b/pre_commit/color.py @@ -1,3 +1,4 @@ +import argparse import os import sys @@ -95,3 +96,12 @@ def use_color(setting: str) -> bool: os.getenv('TERM') != 'dumb' ) ) + + +def add_color_option(parser: argparse.ArgumentParser) -> None: + parser.add_argument( + '--color', default=os.environ.get('PRE_COMMIT_COLOR', 'auto'), + type=use_color, + metavar='{' + ','.join(COLOR_CHOICES) + '}', + help='Whether to use color in output. Defaults to `%(default)s`.', + ) |