Autogenerate setup.py for slicer

Author: chrisgorgo

nipype/interfaces/slicer/generate_classes.py

@@ -26,6 +26,7 @@ def add_class_to_package(class_codes, class_names, module_name, package_dir):
 
 
 def crawl_code_struct(code_struct, package_dir):
+    subpackages = []
     for k, v in code_struct.iteritems():
         if isinstance(v, str) or isinstance(v, unicode):
             module_name = k.lower()
@@ -36,6 +37,7 @@ def crawl_code_struct(code_struct, package_dir):
             module_name = k.lower()
             add_class_to_package(v.values(), v.keys(), module_name, package_dir)
         else:
+            subpackages.append(k.lower())
             f_i = open(os.path.join(package_dir, "__init__.py"), 'a+')
             f_i.write("from %s import *\n" % k.lower())
             f_i.close()
@@ -44,6 +46,24 @@ def crawl_code_struct(code_struct, package_dir):
                 rmtree(new_pkg_dir)
             os.mkdir(new_pkg_dir)
             crawl_code_struct(v, new_pkg_dir)
+        if subpackages:
+            f = open(os.path.join(package_dir, "setup.py"), 'w')
+            f.write("""# emacs: -*- mode: python; py-indent-offset: 4; indent-tabs-mode: nil -*-
+# vi: set ft=python sts=4 ts=4 sw=4 et:
+def configuration(parent_package='',top_path=None):
+    from numpy.distutils.misc_util import Configuration
+
+    config = Configuration('{pkg_name}', parent_package, top_path)
+
+    {sub_pks}
+
+    return config
+
+if __name__ == '__main__':
+    from numpy.distutils.core import setup
+    setup(**configuration(top_path='').todict())
+""".format(pkg_name=package_dir.split("/")[-1], sub_pks="\n    ".join(["config.add_data_dir('%s')"%sub_pkg for sub_pkg in subpackages])))
+            f.close()
 
 
 def generate_all_classes(modules_list=[], launcher=[]):

nipype/interfaces/slicer/setup.py

@@ -6,9 +6,9 @@ def configuration(parent_package='',top_path=None):
     config = Configuration('slicer', parent_package, top_path)
 
     config.add_data_dir('diffusion')
+    config.add_data_dir('segmentation')
     config.add_data_dir('filtering')
     config.add_data_dir('legacy')
-    config.add_data_dir('segmentation')
 
     return config