From 6f5e450bddc480fbf8f8a4bcb67bd856f3a9d88d Mon Sep 17 00:00:00 2001
From: Sylvain Corlay <sylvain.corlay@gmail.com>
Date: Thu, 7 Apr 2016 18:39:01 -0400
Subject: [PATCH] Better implementation of has_flag method

---
 setup.py | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/setup.py b/setup.py
index 9a3e7bc..296acf8 100644
--- a/setup.py
+++ b/setup.py
@@ -17,18 +17,19 @@ ext_modules = [
     ),
 ]
 
+# As of Python 3.6, CCompiler has a `has_flag` method.
+# cf http://bugs.python.org/issue26689
 def has_flag(compiler, flagname):
     """Return a boolean indicating whether a flag name is supported on
     the specified compiler.
     """
     import tempfile
-    fd, fname = tempfile.mkstemp('.cpp', 'main', text=True)
-    with os.fdopen(fd, 'w') as f:
+    with tempfile.NamedTemporaryFile('w', suffix='.cpp') as f:
         f.write('int main (int argc, char **argv) { return 0; }')
-    try:
-        compiler.compile([fname], extra_postargs=[flagname])
-    except setuptools.distutils.errors.CompileError:
-        return False
+        try:
+            compiler.compile([f.name], extra_postargs=[flagname])
+        except setuptools.distutils.errors.CompileError:
+            return False
     return True
 
 def cpp_flag(compiler):