DataSink inputs changed from Directory to Str

Author: kimsin98

nipype/interfaces/io.py

@@ -212,12 +212,12 @@ class DataSinkInputSpec(DynamicTraitedSpec, BaseInterfaceInputSpec):
     """
 
     # Init inputspec data attributes
-    base_directory = Directory(desc="Path to the base directory for storing data.")
+    base_directory = Str(desc="Path to the base directory for storing data.")
     container = Str(desc="Folder within base directory in which to store output")
     parameterization = traits.Bool(
         True, usedefault=True, desc="store output in parametrized structure"
     )
-    strip_dir = Directory(desc="path to strip out of filename")
+    strip_dir = Str(desc="path to strip out of filename")
     substitutions = InputMultiPath(
         traits.Tuple(Str, Str),
         desc=(
@@ -440,24 +440,18 @@ def _check_s3_base_dir(self):
             is not a valid S3 path, defaults to '<N/A>'
         """
 
-        # NOTE: due to pathlib.Path in traits.Directory, S3 bucket paths
-        #       follow the format 's3:/bucket_name/...'
-        s3_str = "s3:/"
+        s3_str = "s3://"
         bucket_name = "<N/A>"
         base_directory = self.inputs.base_directory
 
         if not isdefined(base_directory):
             s3_flag = False
             return s3_flag, bucket_name
 
-        # Check for 's3:/' in base dir
-        if base_directory.lower().startswith(s3_str):
-            bucket_name = base_directory[len(s3_str):].split("/")[0]
-            s3_flag = True
-        else:
-            s3_flag = False
+        s3_flag = base_directory.lower().startswith(s3_str)
+        if s3_flag:
+            bucket_name = base_directory[len(s3_str):].partition('/')[0]
 
-        # Return s3_flag
         return s3_flag, bucket_name
 
     # Function to return AWS secure environment variables
@@ -611,14 +605,12 @@ def _upload_to_s3(self, bucket, src, dst):
 
         from botocore.exceptions import ClientError
 
-        # NOTE: due to pathlib.Path in traits.Directory, S3 bucket paths
-        #       follow the format 's3:/bucket_name/...'
-        s3_str = "s3:/"
+        s3_str = "s3://"
         s3_prefix = s3_str + bucket.name
 
         # Explicitly lower-case the "s3"
-        if dst[: len(s3_str)].lower() == s3_str:
-            dst = s3_str + dst[len(s3_str) :]
+        if dst.lower().startswith(s3_str):
+            dst = s3_str + dst[len(s3_str):]
 
         # If src is a directory, collect files (this assumes dst is a dir too)
         if os.path.isdir(src):