From: Patrick McHardy <kaber@trash.net>
Date: Wed, 25 Mar 2015 07:24:58 +0000 (+0000)
Subject: nft-test: don't use colors if output is not a tty
X-Git-Tag: v0.5~65
X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=d268580568da75f953acfad54804ba7b5ab89d9f;p=thirdparty%2Fnftables.git

nft-test: don't use colors if output is not a tty

Signed-off-by: Patrick McHardy <kaber@trash.net>
---

diff --git a/tests/regression/nft-test.py b/tests/regression/nft-test.py
index 2677127b..559ad41f 100755
--- a/tests/regression/nft-test.py
+++ b/tests/regression/nft-test.py
@@ -31,12 +31,18 @@ signal_received = 0
 
 
 class Colors:
-    HEADER = '\033[95m'
-    GREEN = '\033[92m'
-    YELLOW = '\033[93m'
-    RED = '\033[91m'
-    ENDC = '\033[0m'
-
+    if sys.stdout.isatty():
+        HEADER = '\033[95m'
+        GREEN = '\033[92m'
+        YELLOW = '\033[93m'
+        RED = '\033[91m'
+        ENDC = '\033[0m'
+    else:
+        HEADER = ''
+        GREEN = ''
+        YELLOW = ''
+        RED = ''
+        ENDC = ''
 
 def print_error(reason, filename=None, lineno=None):
     '''