setup.py: force GCC to use colors in case of stdout.isatty()

1 files changed, 4 insertions, 0 deletions

diff --git a/setup.py b/setup.py
index 74b5d5f3..90fea1ba 100755
--- a/setup.py
+++ b/setup.py
@@ -929,6 +929,10 @@ def add_ext_compiler_flags(ext, compiler, _cache={}):
             "-fvisibility=hidden",
         ]
 
+        # force GCC to use colors
+        if hasattr(sys.stdout, "isatty") and sys.stdout.isatty():
+            args.append("-fdiagnostics-color")
+
         _cache[cache_key] = filter_compiler_arguments(compiler, args)
 
     ext.extra_compile_args += _cache[cache_key]