Only colourise synctl output when attached to tty

This commit is contained in:
Erik Johnston 2018-10-17 13:46:08 +01:00
parent d6a7797dd1
commit df33c164de
1 changed files with 6 additions and 1 deletions

7
synctl
View File

@ -48,7 +48,12 @@ def pid_running(pid):
def write(message, colour=NORMAL, stream=sys.stdout):
if colour == NORMAL:
# Lets check if we're writing to a TTY before colouring
should_colour = False
if stream in (sys.stdout, sys.stderr):
should_colour = stream.isatty()
if not should_colour or colour == NORMAL:
stream.write(message + "\n")
else:
stream.write(colour + message + NORMAL + "\n")