P3K: Fix build for py3k + pip.

astaric · GaelVaroquaux · commit 0f247204ba34 · 2012-11-18T16:20:35.000+01:00
Same fix as used in numpy. For more info see http://projects.scipy.org/numpy/ticket/1857.
diff --git a/setup.py b/setup.py
@@ -106,6 +106,20 @@ def configuration(parent_package='', top_path=None):
         if res != 0:
             raise Exception('2to3 failed, exiting ...')
 
+        # Ugly hack to make pip work with Python 3, see
+        # http://projects.scipy.org/numpy/ticket/1857.
+        # Explanation: pip messes with __file__ which interacts badly with the
+        # change in directory due to the 2to3 conversion.  Therefore we restore
+        # __file__ to what it would have been otherwise.
+        global __file__
+        __file__ = os.path.join(os.curdir, os.path.basename(__file__))
+        if '--egg-base' in sys.argv: 
+            # Change pip-egg-info entry to absolute path, so pip can find it 
+            # after changing directory. 
+            idx = sys.argv.index('--egg-base')
+            if sys.argv[idx + 1] == 'pip-egg-info': 
+                sys.argv[idx + 1] = os.path.join(old_path, 'pip-egg-info') 
+
     os.chdir(local_path)
     sys.path.insert(0, local_path)
 
