prisma-api 0.3.4__tar.gz → 0.3.5__tar.gz

{prisma_api-0.3.4 → prisma_api-0.3.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prisma_api
-Version: 0.3.4
+Version: 0.3.5
 Summary: A package for interacting with PrISMa Platform APIs. PrISMa is a holistic system for numerical synthesis and assessment of Metal-Organic Frameworks and their applications, for Direct Air Capture of CO2. For more information, see: https://prisma.hw.ac.uk.
 Author-email: RCCS-CaptureTeam <team@prisma-platform.org>
 License: GNUv3

{prisma_api-0.3.4 → prisma_api-0.3.5}/prisma_api/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.3.4"
+__version__ = "0.3.5"
 
 
 from prisma_api.prisma_api import prisma_api as init        # Main prisma_api class for initialisation

{prisma_api-0.3.4 → prisma_api-0.3.5}/prisma_api/prisma_api_v2.py

@@ -140,6 +140,43 @@ class PrismaAPIv2:
         """
         return self._get(f"/flowsheets/{name}/bundle/")
 
+    def upsert_flowsheets(
+        self,
+        flowsheets: pd.DataFrame | list[dict],
+        on_exists: str = "append",
+        appendix: str = "_v4",
+    ) -> dict:
+        """
+        PUT /api/v2/flowsheets/upsert/
+
+        Modes:
+            * append (default): ``?on_exists=append&appendix=<suffix>``
+            * overwrite:        ``?on_exists=overwrite``
+
+        Args:
+            flowsheets: DataFrame or list of dict payload records.
+            on_exists:  Conflict mode; one of ``'append'`` or ``'overwrite'``.
+            appendix:   Suffix used only in append mode (default ``'_v4'``).
+
+        Returns:
+            API response dict.
+        """
+        if on_exists not in ("append", "overwrite"):
+            raise ValueError("on_exists must be 'append' or 'overwrite'")
+
+        records = (
+            flowsheets.to_dict(orient="records")
+            if isinstance(flowsheets, pd.DataFrame)
+            else flowsheets
+        )
+
+        if on_exists == "overwrite":
+            path = "/flowsheets/upsert/?on_exists=overwrite"
+        else:
+            path = f"/flowsheets/upsert/?on_exists=append&appendix={appendix}"
+
+        return self._put(path, records)
+
     # ── Catalog ───────────────────────────────────────────────────────────────
 
     def list_materials(self, name: str | None = None,

{prisma_api-0.3.4 → prisma_api-0.3.5}/prisma_api.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prisma_api
-Version: 0.3.4
+Version: 0.3.5
 Summary: A package for interacting with PrISMa Platform APIs. PrISMa is a holistic system for numerical synthesis and assessment of Metal-Organic Frameworks and their applications, for Direct Air Capture of CO2. For more information, see: https://prisma.hw.ac.uk.
 Author-email: RCCS-CaptureTeam <team@prisma-platform.org>
 License: GNUv3

{prisma_api-0.3.4 → prisma_api-0.3.5}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "prisma_api"
-version = "0.3.4"
+version = "0.3.5"
 description = "A package for interacting with PrISMa Platform APIs. PrISMa is a holistic system for numerical synthesis and assessment of Metal-Organic Frameworks and their applications, for Direct Air Capture of CO2. For more information, see: https://prisma.hw.ac.uk."
 authors = [
     {name = "RCCS-CaptureTeam", email = "team@prisma-platform.org"}

{prisma_api-0.3.4 → prisma_api-0.3.5}/tests/test_prisma_api_v2.py

@@ -1423,6 +1423,13 @@ def test_api_key_header_sent(api):
 
 # ── Flowsheets ────────────────────────────────────────────────────────────────
 
+_UPSERT_FLOWSHEET_FIXTURE = "reference_data/prisma_v2/dac_min_2026-07-01.json"
+
+
+def _load_upsert_flowsheet_payload() -> list[dict]:
+    with open(_UPSERT_FLOWSHEET_FIXTURE, "r", encoding="utf-8") as f:
+        return [json.load(f)]
+
 @resp_lib.activate
 @pytest.mark.skipif(
     os.getenv("CI", "").lower() == "true",
@@ -1478,3 +1485,48 @@ def test_get_flowsheet_bundle_custom_name_routes_path(api):
                  json={"template_id": "custom_case"}, status=200)
     result = api.get_flowsheet_bundle(name="custom_case")
     assert result["template_id"] == "custom_case"
+
+
+@resp_lib.activate
+def test_upsert_flowsheets_default_append_mode(api):
+    resp_lib.add(
+        resp_lib.PUT,
+        f"{PROD_BASE}/flowsheets/upsert/?on_exists=append&appendix=_v4",
+        json={"created": 1, "updated": 0},
+        status=200,
+    )
+    payload = _load_upsert_flowsheet_payload()
+    result = api.upsert_flowsheets(payload)
+    assert result["created"] == 1
+
+
+@resp_lib.activate
+def test_upsert_flowsheets_overwrite_mode(api):
+    resp_lib.add(
+        resp_lib.PUT,
+        f"{PROD_BASE}/flowsheets/upsert/?on_exists=overwrite",
+        json={"created": 0, "updated": 1},
+        status=200,
+    )
+    payload = _load_upsert_flowsheet_payload()
+    result = api.upsert_flowsheets(payload, on_exists="overwrite")
+    assert result["updated"] == 1
+
+
+@resp_lib.activate
+def test_upsert_flowsheets_uses_dev_mode_base_url(dev_api):
+    resp_lib.add(
+        resp_lib.PUT,
+        "http://localhost:8000/api/v2/flowsheets/upsert/?on_exists=append&appendix=_dev",
+        json={"created": 1, "updated": 0},
+        status=200,
+    )
+    payload = _load_upsert_flowsheet_payload()
+    result = dev_api.upsert_flowsheets(payload, appendix="_dev")
+    assert result["created"] == 1
+
+
+def test_upsert_flowsheets_rejects_invalid_on_exists(api):
+    payload = _load_upsert_flowsheet_payload()
+    with pytest.raises(ValueError):
+        api.upsert_flowsheets(payload, on_exists="replace")