lumera 0.8.3__tar.gz → 0.9.1__tar.gz

{lumera-0.8.3 → lumera-0.9.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lumera
-Version: 0.8.3
+Version: 0.9.1
 Summary: SDK for building on Lumera platform
 Requires-Python: >=3.11
 Requires-Dist: requests

{lumera-0.8.3 → lumera-0.9.1}/lumera/pb.py

@@ -318,51 +318,84 @@ def delete(collection: str, record_id: str) -> None:
 # =============================================================================
 
 
-def bulk_delete(collection: str, record_ids: Sequence[str]) -> dict[str, Any]:
+def bulk_delete(
+    collection: str,
+    record_ids: Sequence[str],
+    *,
+    transaction: bool = False,
+) -> dict[str, Any]:
     """Delete multiple records by ID.
 
     Args:
         collection: Collection name or ID
         record_ids: List of record IDs to delete (max 1000)
+        transaction: If True, use all-or-nothing semantics (rollback on any failure).
+                     If False (default), partial success is allowed.
 
     Returns:
         Result with succeeded/failed counts and any errors:
         {
             "succeeded": 10,
             "failed": 0,
-            "errors": []
+            "errors": [],
+            "rolled_back": false  # only present if transaction=True and failed
         }
 
     Example:
         >>> result = pb.bulk_delete("deposits", ["id1", "id2", "id3"])
         >>> print(f"Deleted {result['succeeded']} records")
+
+        >>> # Use transaction mode for all-or-nothing
+        >>> result = pb.bulk_delete("deposits", ids, transaction=True)
     """
-    return _bulk_delete_records(collection, record_ids)
+    return _bulk_delete_records(collection, record_ids, transaction=transaction)
 
 
 def bulk_update(
     collection: str,
-    record_ids: Sequence[str],
-    data: dict[str, Any],
+    records: Sequence[dict[str, Any]],
+    *,
+    transaction: bool = False,
 ) -> dict[str, Any]:
-    """Update multiple records with the same data.
+    """Update multiple records with individual data per record.
+
+    Each record must have an 'id' field plus fields to update.
+    Records must exist (returns error if not found, does not create).
 
     Args:
         collection: Collection name or ID
-        record_ids: List of record IDs to update (max 1000)
-        data: Fields to update on all records
+        records: List of records to update (max 1000). Each must have 'id' field.
+        transaction: If True, use all-or-nothing semantics (rollback on any failure).
+                     If False (default), partial success is allowed.
 
     Returns:
-        Result with succeeded/failed counts
+        Result with succeeded/failed counts:
+        {
+            "succeeded": 2,
+            "failed": 0,
+            "errors": [],
+            "rolled_back": false  # only present if transaction=True and failed
+        }
 
     Example:
-        >>> result = pb.bulk_update("deposits", ["id1", "id2"], {"status": "processed"})
+        >>> result = pb.bulk_update("deposits", [
+        ...     {"id": "rec1", "status": "approved", "amount": 100},
+        ...     {"id": "rec2", "status": "rejected", "amount": 200},
+        ... ])
         >>> print(f"Updated {result['succeeded']} records")
+
+        >>> # Use transaction mode for all-or-nothing
+        >>> result = pb.bulk_update("deposits", records, transaction=True)
     """
-    return _bulk_update_records(collection, record_ids, data)
+    return _bulk_update_records(collection, records, transaction=transaction)
 
 
-def bulk_upsert(collection: str, records: Sequence[dict[str, Any]]) -> dict[str, Any]:
+def bulk_upsert(
+    collection: str,
+    records: Sequence[dict[str, Any]],
+    *,
+    transaction: bool = False,
+) -> dict[str, Any]:
     """Create or update multiple records by ID.
 
     Each record can include an "id" field. Records with matching IDs will be
@@ -371,25 +404,44 @@ def bulk_upsert(collection: str, records: Sequence[dict[str, Any]]) -> dict[str,
     Args:
         collection: Collection name or ID
         records: List of records (max 1000)
+        transaction: If True, use all-or-nothing semantics (rollback on any failure).
+                     If False (default), partial success is allowed.
 
     Returns:
-        Result with succeeded/failed counts and created record IDs
+        Result with succeeded/failed counts and created record IDs:
+        {
+            "succeeded": 2,
+            "failed": 0,
+            "errors": [],
+            "records": [{"id": "..."}, ...],
+            "rolled_back": false  # only present if transaction=True and failed
+        }
 
     Example:
         >>> result = pb.bulk_upsert("deposits", [
         ...     {"id": "existing_id", "amount": 100},
         ...     {"amount": 200},  # creates new record
         ... ])
+
+        >>> # Use transaction mode for all-or-nothing
+        >>> result = pb.bulk_upsert("deposits", records, transaction=True)
     """
-    return _bulk_upsert_records(collection, records)
+    return _bulk_upsert_records(collection, records, transaction=transaction)
 
 
-def bulk_insert(collection: str, records: Sequence[dict[str, Any]]) -> dict[str, Any]:
+def bulk_insert(
+    collection: str,
+    records: Sequence[dict[str, Any]],
+    *,
+    transaction: bool = False,
+) -> dict[str, Any]:
     """Insert multiple new records.
 
     Args:
         collection: Collection name or ID
         records: List of records to create (max 1000)
+        transaction: If True, use all-or-nothing semantics (rollback on any failure).
+                     If False (default), partial success is allowed.
 
     Returns:
         Result with succeeded/failed counts and created record IDs:
@@ -397,7 +449,8 @@ def bulk_insert(collection: str, records: Sequence[dict[str, Any]]) -> dict[str,
             "succeeded": 2,
             "failed": 0,
             "errors": [],
-            "records": [{"id": "..."}, {"id": "..."}]
+            "records": [{"id": "..."}, {"id": "..."}],
+            "rolled_back": false  # only present if transaction=True and failed
         }
 
     Example:
@@ -407,8 +460,11 @@ def bulk_insert(collection: str, records: Sequence[dict[str, Any]]) -> dict[str,
         ... ])
         >>> for rec in result["records"]:
         ...     print(f"Created: {rec['id']}")
+
+        >>> # Use transaction mode for all-or-nothing
+        >>> result = pb.bulk_insert("deposits", records, transaction=True)
     """
-    return _bulk_insert_records(collection, records)
+    return _bulk_insert_records(collection, records, transaction=transaction)
 
 
 def iter_all(

{lumera-0.8.3 → lumera-0.9.1}/lumera/sdk.py

@@ -591,12 +591,15 @@ def delete_record(collection_id_or_name: str, record_id: str) -> None:
 def bulk_delete_records(
     collection_id_or_name: str,
     record_ids: Sequence[str],
+    *,
+    transaction: bool = False,
 ) -> dict[str, Any]:
     """Bulk delete records by IDs.
 
     Args:
         collection_id_or_name: Collection name or ID
         record_ids: List of record IDs to delete (max 1000)
+        transaction: If True, use all-or-nothing semantics (rollback on any failure)
 
     Returns:
         Result with succeeded/failed counts and any errors
@@ -607,46 +610,54 @@ def bulk_delete_records(
         raise ValueError("record_ids is required")
 
     path = f"collections/{collection_id_or_name}/records/bulk/delete"
-    result = _api_request("POST", path, json_body={"ids": list(record_ids)})
+    body: dict[str, Any] = {"ids": list(record_ids)}
+    if transaction:
+        body["transaction"] = True
+    result = _api_request("POST", path, json_body=body)
     return result if isinstance(result, dict) else {}
 
 
 def bulk_update_records(
     collection_id_or_name: str,
-    record_ids: Sequence[str],
-    data: Mapping[str, Any],
+    records: Sequence[Mapping[str, Any]],
+    *,
+    transaction: bool = False,
 ) -> dict[str, Any]:
-    """Update multiple records with the same data.
+    """Update multiple records with individual data per record.
 
     Args:
         collection_id_or_name: Collection name or ID
-        record_ids: List of record IDs to update (max 1000)
-        data: Fields to update on all records
+        records: List of records to update (max 1000). Each record must have an 'id' field.
+        transaction: If True, use all-or-nothing semantics (rollback on any failure)
 
     Returns:
         Result with succeeded/failed counts
     """
     if not collection_id_or_name:
         raise ValueError("collection_id_or_name is required")
-    if not record_ids:
-        raise ValueError("record_ids is required")
-    if not data:
-        raise ValueError("data is required")
+    if not records:
+        raise ValueError("records is required")
 
     path = f"collections/{collection_id_or_name}/records/bulk/update"
-    result = _api_request("POST", path, json_body={"ids": list(record_ids), "data": dict(data)})
+    body: dict[str, Any] = {"records": [dict(r) for r in records]}
+    if transaction:
+        body["transaction"] = True
+    result = _api_request("POST", path, json_body=body)
     return result if isinstance(result, dict) else {}
 
 
 def bulk_upsert_records(
     collection_id_or_name: str,
     records: Sequence[Mapping[str, Any]],
+    *,
+    transaction: bool = False,
 ) -> dict[str, Any]:
     """Upsert multiple records (create or update by ID).
 
     Args:
         collection_id_or_name: Collection name or ID
         records: List of records (max 1000). Include 'id' field to update existing.
+        transaction: If True, use all-or-nothing semantics (rollback on any failure)
 
     Returns:
         Result with succeeded/failed counts and created record IDs
@@ -657,19 +668,25 @@ def bulk_upsert_records(
         raise ValueError("records is required")
 
     path = f"collections/{collection_id_or_name}/records/bulk/upsert"
-    result = _api_request("POST", path, json_body={"records": [dict(r) for r in records]})
+    body: dict[str, Any] = {"records": [dict(r) for r in records]}
+    if transaction:
+        body["transaction"] = True
+    result = _api_request("POST", path, json_body=body)
     return result if isinstance(result, dict) else {}
 
 
 def bulk_insert_records(
     collection_id_or_name: str,
     records: Sequence[Mapping[str, Any]],
+    *,
+    transaction: bool = False,
 ) -> dict[str, Any]:
     """Insert multiple new records.
 
     Args:
         collection_id_or_name: Collection name or ID
         records: List of records to create (max 1000)
+        transaction: If True, use all-or-nothing semantics (rollback on any failure)
 
     Returns:
         Result with succeeded/failed counts and created record IDs
@@ -680,7 +697,10 @@ def bulk_insert_records(
         raise ValueError("records is required")
 
     path = f"collections/{collection_id_or_name}/records/bulk/insert"
-    result = _api_request("POST", path, json_body={"records": [dict(r) for r in records]})
+    body: dict[str, Any] = {"records": [dict(r) for r in records]}
+    if transaction:
+        body["transaction"] = True
+    result = _api_request("POST", path, json_body=body)
     return result if isinstance(result, dict) else {}
 
 

{lumera-0.8.3 → lumera-0.9.1}/lumera.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lumera
-Version: 0.8.3
+Version: 0.9.1
 Summary: SDK for building on Lumera platform
 Requires-Python: >=3.11
 Requires-Dist: requests

{lumera-0.8.3 → lumera-0.9.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "lumera"
-version = "0.8.3"
+version = "0.9.1"
 description = "SDK for building on Lumera platform"
 requires-python = ">=3.11"
 dependencies = [

{lumera-0.8.3 → lumera-0.9.1}/tests/test_sdk.py

@@ -852,3 +852,45 @@ def test_claim_locks_custom_provenance(monkeypatch: pytest.MonkeyPatch) -> None:
     body = captured.get("json_body")
     assert isinstance(body, dict)
     assert body["provenance"] == custom_prov
+
+
+def test_bulk_update_records(monkeypatch: pytest.MonkeyPatch) -> None:
+    monkeypatch.setenv(sdk.TOKEN_ENV, "tok")
+
+    captured: dict[str, object] = {}
+
+    def fake_api(method: str, path: str, **kwargs: object) -> dict[str, object]:
+        captured["method"] = method
+        captured["path"] = path
+        captured["kwargs"] = kwargs
+        return {"succeeded": 2, "failed": 0, "errors": []}
+
+    monkeypatch.setattr(sdk, "_api_request", fake_api)
+
+    result = sdk.bulk_update_records(
+        "test_collection",
+        [
+            {"id": "rec1", "status": "approved", "amount": 100},
+            {"id": "rec2", "status": "rejected", "amount": 200},
+        ],
+    )
+
+    assert result["succeeded"] == 2
+    assert result["failed"] == 0
+    assert captured["method"] == "POST"
+    assert captured["path"] == "collections/test_collection/records/bulk/update"
+    kwargs = captured["kwargs"]
+    assert "json_body" in kwargs
+    body = kwargs["json_body"]
+    assert isinstance(body, dict)
+    assert "records" in body
+    assert len(body["records"]) == 2
+    assert body["records"][0]["id"] == "rec1"
+    assert body["records"][0]["status"] == "approved"
+    assert body["records"][1]["id"] == "rec2"
+    assert body["records"][1]["status"] == "rejected"
+
+
+def test_bulk_update_records_requires_records() -> None:
+    with pytest.raises(ValueError, match="records is required"):
+        sdk.bulk_update_records("test_collection", [])