Add support for NO_COLOR

Christoffer Rehn · Christoffer Rehn · commit 88cb015f4ae0 · 2018-06-06T17:03:57.000+09:00
diff --git a/kitten/__init__.py b/kitten/__init__.py
@@ -14,7 +14,7 @@
 import fabric
 from six.moves import range, queue
 
-__version__ = "0.2.13"
+__version__ = "0.2.14"
 
 CHUNK_SIZE = 100
 DEFAULT = {"threads": 10, "timeout": 10}
@@ -50,9 +50,9 @@ def ansi(x):
 
 
 def colored(s, code=0, bold=False):
-    if sys.stdout.isatty():
-        return "{}{}{}{}".format(ansi(code), ansi(1) if bold else "", s, ansi(0))
-    return s
+    if not sys.stdout.isatty() or "NO_COLOR" in os.environ:
+        return s
+    return "{}{}{}{}".format(ansi(code), ansi(1) if bold else "", s, ansi(0))
 
 
 def red(s):
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="kitten",
-    version="0.2.13",
+    version="0.2.14",
     author="Chris Rehn",
     author_email="chris@rehn.me",
     description="Tiny multi-server automation tool.",
