fix: get / post missing arg

jupyter-naas · Apr 2, 2024 · 18ddf68 · 18ddf68
1 parent c023bd2
commit 18ddf68
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/naas_python/domains/storage/adaptors/secondary/providers/S3StorageProviderAdaptor.py b/naas_python/domains/storage/adaptors/secondary/providers/S3StorageProviderAdaptor.py
@@ -51,6 +51,8 @@ def post_workspace_storage_object(self,
         src_file: str,
         dst_file: str,
     ) -> None:
+        if dst_file.endswith('/'):
+            dst_file = dst_file + os.path.basename(src_file)
         key = f"{workspace_id}/{storage_name}/{dst_file}"
         key = self.__clean_path(key)
 
@@ -74,7 +76,8 @@ def get_workspace_storage_object(self,
         src_file: str, 
         dst_file:str, 
     ) -> bytes :
-
+        if dst_file.endswith('/') or dst_file.endswith('.'):
+            dst_file = dst_file + os.path.basename(src_file)
         filename=dst_file
         self.__clean_path(filename)
         object_key = workspace_id + "/" + storage_name + "/" + src_file