forcingkit 0.12.1__tar.gz → 0.12.2__tar.gz

{forcingkit-0.12.1 → forcingkit-0.12.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: forcingkit
-Version: 0.12.1
+Version: 0.12.2
 Summary: Spatiotemporal forcing for computational Earth-system models: selects, regrids and serves model-ready time series with provenance.
 Project-URL: Documentation, https://forcingkit.docs.lhzn.io
 Project-URL: Issues, https://github.com/lhzn-io/forcingkit/issues

{forcingkit-0.12.1 → forcingkit-0.12.2}/docs/source/conf.py

@@ -9,7 +9,7 @@ sys.path.insert(0, os.path.abspath("../../src"))
 project = "forcingkit"
 copyright = "2026, Long Horizon Observatory"
 author = "Daniel Fry"
-release = "0.12.1"
+release = "0.12.2"
 
 extensions = [
     "sphinx.ext.autodoc",

{forcingkit-0.12.1 → forcingkit-0.12.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "forcingkit"
-version = "0.12.1"
+version = "0.12.2"
 description = "Spatiotemporal forcing for computational Earth-system models: selects, regrids and serves model-ready time series with provenance."
 readme = "README.md"
 requires-python = ">=3.11"

{forcingkit-0.12.1 → forcingkit-0.12.2}/service/forcingkit_serve/main.py

@@ -1,5 +1,6 @@
 import logging
 import os
+import re
 import sys
 import time
 import shutil
@@ -890,10 +891,17 @@ def generate_atmosphere_levels(request: AtmosphereLevelsRequest) -> Dict[str, An
     return {"status": "success", "zarr_id": zarr_id, "zarr_path": final_path}
 
 
+ATMOSPHERE_DOWNLOAD_ID = re.compile(r"^((atm|lev)_)?[0-9a-f]{12}$")
+
+
 @app.get("/api/v1/atmosphere/download/{zarr_id}")
 def download_atmosphere(zarr_id: str):
+    """The surface (`atm_`) or lower-atmosphere (`lev_`) store as a zip; a bare hash means
+    `atm_`."""
+    if not ATMOSPHERE_DOWNLOAD_ID.match(zarr_id):
+        raise HTTPException(status_code=400, detail="invalid atmosphere store id")
     cache_dir = _atmosphere_cache_dir()
-    search_id = zarr_id if zarr_id.startswith("atm_") else f"atm_{zarr_id}"
+    search_id = zarr_id if "_" in zarr_id else f"atm_{zarr_id}"
     zarr_path = os.path.join(cache_dir, f"{search_id}.zarr")
     if not os.path.isdir(zarr_path):
         raise HTTPException(status_code=404, detail="Atmosphere Zarr store not found.")

{forcingkit-0.12.1 → forcingkit-0.12.2}/tests/unit/test_jobs.py

@@ -215,6 +215,36 @@ def test_waves_out_of_reach_is_404(client, monkeypatch):
     assert response.status_code == 404 and "no NWPS cycle" in response.json()["detail"]
 
 
+@pytest.mark.parametrize("prefix", ["atm_", "lev_"])
+def test_atmosphere_download_serves_surface_and_levels(client, tmp_path, prefix):
+    import io
+    import zipfile
+
+    zarr_id = prefix + "0123456789ab"
+    _write_store(str(tmp_path / f"{zarr_id}.zarr"))
+    response = client.get(f"/api/v1/atmosphere/download/{zarr_id}")
+    assert response.status_code == 200
+    assert f'filename="{zarr_id}.zip"' in response.headers["content-disposition"]
+    assert ".zattrs" in zipfile.ZipFile(io.BytesIO(response.content)).namelist()
+    if prefix == "atm_":  # a bare hash still means the surface store
+        assert client.get("/api/v1/atmosphere/download/0123456789ab").status_code == 200
+
+
+def test_atmosphere_download_refuses_other_ids(client):
+    for bad in (
+        "obc_0123456789ab",
+        "lev_0123",
+        "..%2Fatm_0123456789ab",
+        "wav_0123456789ab",
+    ):
+        assert client.get(f"/api/v1/atmosphere/download/{bad}").status_code in (
+            400,
+            404,
+        )
+    assert client.get("/api/v1/atmosphere/download/obc_0123456789ab").status_code == 400
+    assert client.get("/api/v1/atmosphere/download/lev_000000000000").status_code == 404
+
+
 def test_store_files_are_served(client, tmp_path):
     _write_store(str(tmp_path / "obc_0123456789ab.zarr"))
     base = "/api/v1/stores/obc_0123456789ab.zarr"

{forcingkit-0.12.1 → forcingkit-0.12.2}/uv.lock

@@ -1000,7 +1000,7 @@ wheels = [
 
 [[package]]
 name = "forcingkit"
-version = "0.12.1"
+version = "0.12.2"
 source = { editable = "." }
 dependencies = [
     { name = "bottleneck" },