Separating supported extensions from #482

Author: ktsitsi

src/tiledb/cloud/bioimg/__init__.py

@@ -1,4 +1,6 @@
 from tiledb.cloud.bioimg.exportation import export
 from tiledb.cloud.bioimg.ingestion import ingest
 
-__all__ = ("ingest", "export")
+from .types import SupportedExtensions
+
+__all__ = ("ingest", "export", "SupportedExtensions")

src/tiledb/cloud/bioimg/helpers.py

@@ -4,6 +4,8 @@
 
 import tiledb
 
+from .types import SupportedExtensions
+
 
 def get_uris(
     source: Sequence[str], output_dir: str, config: Mapping[str, Any], output_ext: str
@@ -21,7 +23,7 @@ def create_output_path(input_file, output_dir) -> str:
 
     def iter_paths(sequence) -> Iterator[Tuple]:
         for uri in sequence:
-            if uri.endswith((".tiff", ".tif", ".tdb")):
+            if uri.endswith(tuple([ext.value for ext in SupportedExtensions])):
                 yield uri, create_output_path(uri, output_dir)
 
     if len(source) == 1 and vfs.is_dir(source[0]):

src/tiledb/cloud/bioimg/types.py

@@ -0,0 +1,7 @@
+import enum
+
+class SupportedExtensions(enum.Enum):
+    TIFF: str = ".tiff"
+    TIF: str = ".tif"
+    SVS: str = ".svs"
+    TDB: str = ".tdb"