Copy implementation of check_output into our setup.py so it runs on Python 2.6.

1 files changed, 15 insertions, 2 deletions

diff --git a/setup.py b/setup.py
index 405a903..27c0f39 100644
--- a/setup.py
+++ b/setup.py
@@ -3,8 +3,21 @@ distribute_setup.use_setuptools()
 from setuptools import setup, find_packages, Extension
 import subprocess
 
-geant4_cflags = subprocess.check_output(['geant4-config','--cflags']).split()
-geant4_libs = subprocess.check_output(['geant4-config','--libs']).split()
+def check_output(*popenargs, **kwargs):
+    if 'stdout' in kwargs:
+        raise ValueError('stdout argument not allowed, it will be overridden.')
+    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]
+        raise subprocess.CalledProcessError(retcode, cmd, output=output)
+    return output
+
+geant4_cflags = check_output(['geant4-config','--cflags']).split()
+geant4_libs = check_output(['geant4-config','--libs']).split()
 
 setup(
     name = 'Chroma',