merge: merge pull request #55 from DTC-Glaciers/fix-prevent-ddos

fix: prevent writing over datacube whilst already downloading

Author: gampnico

docs/source/conf.py

@@ -30,7 +30,7 @@ def setup(app):
 project = "DTCG"
 copyright = f"{date.today().year}, DTCG Contributors"
 author = "DTCG Contributors"
-release = "1.0.0"
+release = "1.0.1"
 version = os.environ.get("READTHEDOCS_VERSION", "latest")
 
 # -- General configuration ---------------------------------------------------

dtcg/api/external/call.py

@@ -101,23 +101,23 @@ def zip_datacube(self, rgi_id: str, zip_path: Path = ""):
             stream_url = self.get_url(rgi_id=rgi_id)
         try:
             zip_path = self.get_zip_path(zip_path=zip_path, rgi_id=rgi_id)
-
-            store = zarr.storage.ZipStore(zip_path, mode="w")
-            with xr.open_datatree(
-                stream_url,
-                group=None,
-                chunks={},
-                engine="zarr",
-                consolidated=True,
-                decode_cf=True,
-            ) as stream:
-                stream.compute().to_zarr(
-                    store=store,
-                    mode="w",
+            if not zip_path.is_file():
+                store = zarr.storage.ZipStore(zip_path, mode="w")
+                with xr.open_datatree(
+                    stream_url,
+                    group=None,
+                    chunks={},
+                    engine="zarr",
                     consolidated=True,
-                    zarr_format=2,
-                    # encoding=stream.encoding,
-                )
+                    decode_cf=True,
+                ) as stream:
+                    stream.compute().to_zarr(
+                        store=store,
+                        mode="w",
+                        consolidated=True,
+                        zarr_format=2,
+                        # encoding=stream.encoding,
+                    )
 
         except GroupNotFoundError as e:
             # pipe different error to frontends

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "dtcg"
-version = "1.0.0"
+version = "1.0.1"
 authors = [
     { name = "DTCG Contributors", email = "" },
 ]
@@ -20,7 +20,7 @@ keywords = [
 ]
 classifiers = [
     "License :: OSI Approved :: Apache Software License",
-    "Development Status :: 1 - Planning",
+    "Development Status :: 5 - Production/Stable",
     "Private :: Do Not Upload",
     "Intended Audience :: Science/Research",
     "Operating System :: POSIX :: Linux",