#139 Removed archive extension in as_udf_path

exasol · May 23, 2024 · 3c9520e · 3c9520e
1 parent 7be848b
commit 3c9520e
Show file tree

Hide file tree

Showing 3 changed files with 54 additions and 135 deletions.
diff --git a/exasol/bucketfs/_path.py b/exasol/bucketfs/_path.py
@@ -9,6 +9,8 @@
 from exasol.bucketfs._service import Service
 from exasol.bucketfs._error import BucketFsError
 
+ARCHIVE_SUFFIXES = [".tar", ".gz", ".tgz", ".zip", ".tar"]
+
 
 class StorageBackend(Enum):
     onprem = auto()
@@ -170,6 +172,12 @@ def __truediv__(self, other):
         """
 
 
+def _remove_archive_suffix(path: PurePath) -> PurePath:
+    while path.suffix in ARCHIVE_SUFFIXES:
+        path = path.with_suffix('')
+    return path
+
+
 class _BucketFile:
     """
     A node in a perceived file structure of a bucket.
@@ -288,7 +296,8 @@ def as_uri(self) -> str:
         return self._path.as_uri()
 
     def as_udf_path(self) -> str:
-        return str(PurePath(self._bucket_api.udf_path) / self._path)
+        return str(PurePath(self._bucket_api.udf_path) /
+                   _remove_archive_suffix(self._path))
 
     def exists(self) -> bool:
         return self._navigate() is not None