Assume isatty is always defined, and catch AttributeError. Also don't bother checking colour==Normal

pull/14/head
Erik Johnston 6 years ago
parent 1af16acd4c
commit a5aea15a6b
  1. 8
      synctl

@ -50,10 +50,14 @@ def pid_running(pid):
def write(message, colour=NORMAL, stream=sys.stdout):
# Lets check if we're writing to a TTY before colouring
should_colour = False
if stream in (sys.stdout, sys.stderr):
try:
should_colour = stream.isatty()
except AttributeError:
# Just in case `isatty` isn't defined on everything. The python
# docs are incredibly vague.
pass
if not should_colour or colour == NORMAL:
if not should_colour:
stream.write(message + "\n")
else:
stream.write(colour + message + NORMAL + "\n")

Loading…
Cancel
Save