Fix build on Python 2.6

mdboom · mdboom · commit 7db8fe22902e · 2013-02-25T18:42:34.000-05:00
diff --git a/setupext.py b/setupext.py
@@ -12,12 +12,39 @@
 from textwrap import fill
 
 
+try:
+    from subprocess import check_output
+except ImportError:
+    # check_output is not available in Python 2.6
+    def check_output(*popenargs, **kwargs):
+        """
+        Run command with arguments and return its output as a byte
+        string.
+
+        Backported from Python 2.7 as it's implemented as pure python
+        on stdlib.
+        """
+        process = subprocess.Popen(
+            stdout=subprocess.PIPE, *popenargs, **kwargs)
+        output, unused_err = process.communicate()
+        retcode = process.poll()
+        if retcode:
+            cmd = kwargs.get("args")
+            if cmd is None:
+                cmd = popenargs[0]
+            error = subprocess.CalledProcessError(retcode, cmd)
+            error.output = output
+            raise error
+        return output
+
+
 if sys.platform != 'win32':
     if sys.version_info[0] < 3:
         from commands import getstatusoutput
     else:
         from subprocess import getstatusoutput
 
+
 if sys.version_info[0] < 3:
     import ConfigParser as configparser
 else:
@@ -249,7 +276,7 @@ def setup_extension(self, ext, package, default_include_dirs=[],
         use_defaults = True
         if self.has_pkgconfig:
             try:
-                output = subprocess.check_output(command, shell=True)
+                output = check_output(command, shell=True)
             except subprocess.CalledProcessError:
                 pass
             else:
