enh: selectdirs too

Author: mgxd

nipype/interfaces/io.py

@@ -1186,17 +1186,19 @@ def _list_outputs(self):
 class SelectFilesInputSpec(DynamicTraitedSpec, BaseInterfaceInputSpec):
 
     base_directory = Directory(exists=True,
-                               desc="Root path common to templates.")
+        desc="Root path common to templates.")
     sort_filelist = traits.Bool(True, usedefault=True,
-                                desc="When matching mutliple files, return them in sorted order.")
+        desc="When matching mutliple files, return them in sorted order.")
     raise_on_empty = traits.Bool(True, usedefault=True,
-                                 desc="Raise an exception if a template pattern matches no files.")
+        desc="Raise an exception if a template pattern matches no files.")
     force_lists = traits.Either(traits.Bool(), traits.List(Str()),
-                                default=False, usedefault=True,
-                                desc=("Whether to return outputs as a list even when only one file "
-                                      "matches the template. Either a boolean that applies to all "
-                                      "output fields or a list of output field names to coerce to "
-                                      " a list"))
+        default=False, usedefault=True,
+        desc=("Whether to return outputs as a list even when only one file "
+              "matches the template. Either a boolean that applies to all "
+              "output fields or a list of output field names to coerce to "
+              " a list"))
+    directory_mode = traits.Bool(False, usedefault=True,
+        desc="Return only directories.")
 
 
 class SelectFiles(IOBase):
@@ -1303,6 +1305,10 @@ def _list_outputs(self):
             else:
                 template = op.abspath(template)
 
+            if self.inputs.directory_mode:
+                # return only directories
+                template += os.sep
+
             # Fill in the template and glob for files
             filled_template = template.format(**info)
             filelist = glob.glob(filled_template)

nipype/interfaces/tests/test_auto_SelectFiles.py

@@ -5,6 +5,8 @@
 
 def test_SelectFiles_inputs():
     input_map = dict(base_directory=dict(),
+    directory_mode=dict(usedefault=True,
+    ),
     force_lists=dict(usedefault=True,
     ),
     ignore_exception=dict(nohash=True,