From 6cfa8e0963e9a518a25528a248932fcf41b4f969 Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Sun, 30 Aug 2020 13:22:53 +0200 Subject: Adding upstream version 2.7.1. Signed-off-by: Daniel Baumann --- pre_commit/color.py | 10 ++++++++++ 1 file changed, 10 insertions(+) (limited to 'pre_commit/color.py') 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`.', + ) -- cgit v1.2.3