From a6f7eac41b777768e69c873777d991225678f442 Mon Sep 17 00:00:00 2001 From: Bryan Arant Date: Thu, 3 Nov 2016 17:33:48 -0700 Subject: [PATCH] only use colors if we are in a real terminal --- tools/cli-bootstrap/cli.bootstrap.py | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) diff --git a/tools/cli-bootstrap/cli.bootstrap.py b/tools/cli-bootstrap/cli.bootstrap.py index f55b8c4d..68fb3a49 100644 --- a/tools/cli-bootstrap/cli.bootstrap.py +++ b/tools/cli-bootstrap/cli.bootstrap.py @@ -22,6 +22,8 @@ from urllib import urlretrieve # ROVER BASE # class RoverMods: + PIPE_TO_STDOUT = not sys.stdout.isatty() + HEADER = '\033[95m' BLUE = '\033[94m' GREEN = '\033[92m' @@ -34,34 +36,58 @@ class RoverMods: @staticmethod def Header(line): + if(RoverMods.PIPE_TO_STDOUT): + return line + return RoverMods.HEADER + line + RoverMods.ENDC @staticmethod def Blue(line): + if(RoverMods.PIPE_TO_STDOUT): + return line + return RoverMods.BLUE + line + RoverMods.ENDC @staticmethod def Green(line): + if(RoverMods.PIPE_TO_STDOUT): + return line + return RoverMods.GREEN + line + RoverMods.ENDC @staticmethod def Yellow(line): + if(RoverMods.PIPE_TO_STDOUT): + return line + return RoverMods.YELLOW + line + RoverMods.ENDC @staticmethod def White(line): + if(RoverMods.PIPE_TO_STDOUT): + return line + return RoverMods.WHITE + line + RoverMods.ENDC @staticmethod def Red(line): + if(RoverMods.PIPE_TO_STDOUT): + return line + return RoverMods.RED + line + RoverMods.ENDC @staticmethod def Bold(line): + if(RoverMods.PIPE_TO_STDOUT): + return line + return RoverMods.BOLD + line + RoverMods.ENDC @staticmethod def Underline(line): + if(RoverMods.PIPE_TO_STDOUT): + return line + return RoverMods.UNDERLINE + line + RoverMods.ENDC def RoverPrint(line):