Add a new option, --no-color, to get rid of ANSI colour escape codes in the
output. Useful if redirecting output to a file or piping to another script.
(It should really be --no-colour, but I'll accept US spelling in the name
of consistency with existing usage in the code comments...)
Signed-off-by: Andrew Donnellan <andrew.donnel...@au1.ibm.com>
---
scripts/checkkconfigsymbols.py | 11 +++++++++--
1 file changed, 9 insertions(+), 2 deletions(-)
diff --git a/scripts/checkkconfigsymbols.py b/scripts/checkkconfigsymbols.py
index df643f6..a9eba02 100755
--- a/scripts/checkkconfigsymbols.py
+++ b/scripts/checkkconfigsymbols.py
@@ -82,6 +82,10 @@ def parse_options():
default=False,
help="Reset current Git tree even when it's dirty.")
+ parser.add_option('', '--no-color', dest='no_color', action='store_true',
+ default=False,
+ help="Don't print colored output.")
+
(opts, _) = parser.parse_args()
if opts.commit and opts.diff:
@@ -116,6 +120,9 @@ def main():
"""Main function of this module."""
opts = parse_options()
+ global no_color
+ no_color = opts.no_color
+
if opts.sim and not opts.commit and not opts.diff:
sims = find_sims(opts.sim, opts.ignore)
if sims:
@@ -202,14 +209,14 @@ def yel(string):
"""
Color %string yellow.
"""
- return "\033[33m%s\033[0m" % string
+ return string if no_color else "\033[33m%s\033[0m" % string
def red(string):
"""
Color %string red.
"""
- return "\033[31m%s\033[0m" % string
+ return string if no_color else "\033[31m%s\033[0m" % string
def execute(cmd):
--
Andrew Donnellan OzLabs, ADL Canberra
andrew.donnel...@au1.ibm.com IBM Australia Limited
