Minor changes

1 files changed, 29 insertions, 0 deletions

diff --git a/pydrizzle/pydrizzle.ini b/pydrizzle/pydrizzle.ini
index 9d4c102..46cd116 100644
--- a/pydrizzle/pydrizzle.ini
+++ b/pydrizzle/pydrizzle.ini
@@ -21,6 +21,8 @@ build:
     astropy >=1.0
     numpy >=1.6.0
     stsci.distutils
+    stsci.tools
+    pyfits
     setuptools
     python
 
@@ -29,12 +31,39 @@ run:
     astropy-helpers
     astropy >=1.0
     numpy >=1.6.0
+    stsci.tools
+    pyfits
     python
 
 [cbc_build]
 linux:
+    function get_py_version() {
+        python --version 2>&1| awk '{sub(/-.*/,"",$$2);print $$2}' 
+    }
+    export -f get_py_version
+
+    function set_py3k() {
+        version=`get_py_version`
+        major=`echo $$version | cut -d '.' -f 1`
+        minor=`echo $$version | cut -d '.' -f 2`
+        hotfix=`echo $$version | cut -d '.' -f 3`
+        if (( major > 2 )); then
+            export PY3K=1
+        else
+            export PY3K=0
+        fi
+    }
+    export -f set_py3k
+
+    set_py3k
+
+    if [ $$PY3K -ne 0 ]; then
+        sed -i -e "s|stsci.distutils.hooks.tag_svn_revision|#removed|" setup.cfg
+    fi
+    
     echo This d2to1 hack is deadly.
     pip install --no-deps --upgrade --force d2to1 || exit 1
+
     python setup.py install || exit 1
 
 windows: