From 7caca9a882a1c5151da15b49e6c852fa42f423d7 Mon Sep 17 00:00:00 2001 From: Alinson S Xavier Date: Thu, 27 Aug 2020 10:10:32 -0500 Subject: [PATCH] Only use color log output if stdout is a TTY --- src/python/miplearn/log.py | 21 +++++++++++++++------ 1 file changed, 15 insertions(+), 6 deletions(-) diff --git a/src/python/miplearn/log.py b/src/python/miplearn/log.py index e944e42..c41f0cc 100644 --- a/src/python/miplearn/log.py +++ b/src/python/miplearn/log.py @@ -5,13 +5,22 @@ from datetime import timedelta import logging import time +import sys -log_colors = { - "green": '\033[92m', - "yellow": '\033[93m', - "red": '\033[91m', - "reset": '\033[0m', -} +if sys.stdout.isatty(): + log_colors = { + "green": '\033[92m', + "yellow": '\033[93m', + "red": '\033[91m', + "reset": '\033[0m', + } +else: + log_colors = { + "green": "", + "yellow": "", + "red": "", + "reset": "", + } class TimeFormatter(): def __init__(self, start_time):