Only emit color messages when stdout is a tty

Fixes #213.

Author: jengelh

TarSCM/scm/git.py

@@ -6,6 +6,10 @@
 
 from TarSCM.scm.base import Scm
 
+def color_red(s):
+    if not os.isatty(1):
+        return s
+    return "\033[0;31m" + s + "\033[0m"
 
 class Git(Scm):
     scm = 'git'
@@ -200,9 +204,8 @@ def _detect_parent_tag(self, args):
 
     def _detect_version_parent_tag(self, parent_tag, versionformat):  # noqa pylint: disable=no-self-use
         if not parent_tag:
-            sys.exit("\033[31mNo parent tag present for the checked out "
-                     "revision, thus @PARENT_TAG@ cannot be expanded."
-                     "\033[0m")
+            sys.exit(color_red("No parent tag present for the checked out "
+                     "revision, thus @PARENT_TAG@ cannot be expanded."))
 
         versionformat = re.sub('@PARENT_TAG@', parent_tag,
                                versionformat)
@@ -214,15 +217,14 @@ def check_ancestry(self, parent_tag):
             self.clone_dir
         )
         if rcode:
-            sys.exit("\033[31m`git rev-parse " + parent_tag + "` failed: " +
-                     tgcommit + "\033[0m")
+            sys.exit(color_red("`git rev-parse " + parent_tag + "` failed: " +
+                     tgcommit))
         rcode, head = self.helpers.run_cmd(
             ["git", "rev-parse", "HEAD"],
             self.clone_dir
         )
         if rcode:
-            sys.exit("\033[31m`git rev-parse HEAD` failed: " +
-                     head + "\033[0m")
+            sys.exit(color_red("`git rev-parse HEAD` failed: " + head))
         rcode, mgbase = self.helpers.run_cmd(
             ["git", "merge-base", parent_tag, "HEAD"],
             self.clone_dir
@@ -231,24 +233,20 @@ def check_ancestry(self, parent_tag):
         head = head.strip()
         mgbase = mgbase.strip()
         if mgbase != tgcommit:
-            sys.exit("\033[31mparent_tag is not an ancestor of HEAD. " +
-                     "Cannot compute a (meaningful) distance.\033[0m")
+            sys.exit(color_red("parent_tag is not an ancestor of HEAD. " +
+                     "Cannot compute a (meaningful) distance."))
 
     def _detect_version_tag_offset(self, parent_tag, versionformat):
         if not parent_tag:
-            sys.exit("\033[31m@TAG_OFFSET@ cannot be expanded, "
-                     "as no parent tag was discovered.\033[0m")
-
+            sys.exit(color_red("@TAG_OFFSET@ cannot be expanded, as no parent tag was discovered."))
         self.check_ancestry(parent_tag)
         rcode, output = self.helpers.run_cmd(
             self._get_scm_cmd() + ['rev-list', '--count', parent_tag + '..HEAD'],
             self.clone_dir
         )
 
         if rcode:
-            msg = "\033[31m@TAG_OFFSET@ can not be expanded: %s\033[0m"
-            msg.format(out)
-            sys.exit(msg)
+            sys.exit(color_red("@TAG_OFFSET@ can not be expanded: " + output))
 
         tag_offset = out.strip()
         versionformat = re.sub('@TAG_OFFSET@', tag_offset,