Supress ANSI escape sequences when terminal not detected 09/107209/2
authorBartek Grzybowski <b.grzybowski@partner.samsung.com>
Wed, 6 May 2020 10:37:23 +0000 (03:37 -0700)
committerBartek Grzybowski <b.grzybowski@partner.samsung.com>
Fri, 8 May 2020 06:50:13 +0000 (06:50 +0000)
This prevents obscured output when redirecting output to
a file descriptor.

Change-Id: Ie56b6f65b1d2adffc69d3f75a9c8f98c0444faa8
Issue-ID: INT-1577
Signed-off-by: Bartek Grzybowski <b.grzybowski@partner.samsung.com>
test/mocks/mass-pnf-sim/mass-pnf-sim.py

index c4fb0de..db567ed 100755 (executable)
@@ -8,8 +8,6 @@ import logging
 from requests import get
 from requests.exceptions import MissingSchema, InvalidSchema, InvalidURL, ConnectionError, ConnectTimeout
 
-logging.basicConfig(level=logging.INFO, format='\033[92m[%(levelname)s]\033[0m %(message)s')
-
 def validate_url(url):
     '''Helper function to perform --urlves input param validation'''
     logger = logging.getLogger("urllib3")
@@ -31,6 +29,11 @@ def validate_ip(ip):
     else:
         return ip_valid
 
+if sys.stdout.isatty():
+    logging.basicConfig(level=logging.INFO, format='\033[92m[%(levelname)s]\033[0m %(message)s')
+else:
+    logging.basicConfig(level=logging.INFO, format='[%(levelname)s] %(message)s')
+
 parser = argparse.ArgumentParser()
 parser.add_argument('--bootstrap', help='Bootstrap the system', type=int, metavar='COUNT')
 parser.add_argument('--trigger', help='Trigger one single VES event from each simulator', type=int,
@@ -52,6 +55,7 @@ parser.add_argument('--verbose', help='Verbosity level', choices=['info', 'debug
                     type=str, default='debug')
 
 args = parser.parse_args()
+
 logger = logging.getLogger(__name__)
 logger.setLevel(getattr(logging, args.verbose.upper()))