Assume isatty is always defined, and catch AttributeError. Also don't bother checking colour==Normal
parent
1af16acd4c
commit
a5aea15a6b
8
synctl
8
synctl
|
@ -50,10 +50,14 @@ def pid_running(pid):
|
||||||
def write(message, colour=NORMAL, stream=sys.stdout):
|
def write(message, colour=NORMAL, stream=sys.stdout):
|
||||||
# Lets check if we're writing to a TTY before colouring
|
# Lets check if we're writing to a TTY before colouring
|
||||||
should_colour = False
|
should_colour = False
|
||||||
if stream in (sys.stdout, sys.stderr):
|
try:
|
||||||
should_colour = stream.isatty()
|
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")
|
stream.write(message + "\n")
|
||||||
else:
|
else:
|
||||||
stream.write(colour + message + NORMAL + "\n")
|
stream.write(colour + message + NORMAL + "\n")
|
||||||
|
|
Loading…
Reference in New Issue