Fix bug to ensure that swig generated .py files are installed.

pearu · pearu · commit 451fe6addbfd · 2010-08-25T11:23:07.000Z
diff --git a/numpy/distutils/command/__init__.py b/numpy/distutils/command/__init__.py
@@ -5,7 +5,7 @@
 
 __revision__ = "$Id: __init__.py,v 1.3 2005/05/16 11:08:49 pearu Exp $"
 
-distutils_all = [  'build_py',
+distutils_all = [  #'build_py',
                    'clean',
                    'install_clib',
                    'install_scripts',
@@ -20,6 +20,7 @@
            'config_compiler',
            'config',
            'build_src',
+           'build_py',
            'build_ext',
            'build_clib',
            'build_scripts',
diff --git a/numpy/distutils/command/build_py.py b/numpy/distutils/command/build_py.py
@@ -4,6 +4,12 @@
 
 class build_py(old_build_py):
 
+    def run(self):
+        build_src = self.get_finalized_command('build_src')
+        if build_src.py_modules_dict and self.packages is None:
+            self.packages = build_src.py_modules_dict.keys ()
+        old_build_py.run(self)
+
     def find_package_modules(self, package, package_dir):
         modules = old_build_py.find_package_modules(self, package, package_dir)
 
@@ -19,6 +25,7 @@ def find_modules(self):
         self.py_modules[:] = new_py_modules
         modules = old_build_py.find_modules(self)
         self.py_modules[:] = old_py_modules
+
         return modules
 
     # XXX: Fix find_source_files for item in py_modules such that item is 3-tuple
