altcodepro-polydb-python 2.5.23__tar.gz → 2.5.24__tar.gz

{altcodepro_polydb_python-2.5.23/src/altcodepro_polydb_python.egg-info → altcodepro_polydb_python-2.5.24}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: altcodepro-polydb-python
-Version: 2.5.23
+Version: 2.5.24
 Summary: Production-ready multi-cloud database abstraction layer with connection pooling, retry logic, and thread safety
 Author: AltCodePro
 Project-URL: Homepage, https://github.com/altcodepro/polydb-python

{altcodepro_polydb_python-2.5.23 → altcodepro_polydb_python-2.5.24}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "altcodepro-polydb-python"
-version = "2.5.23"
+version = "2.5.24"
 description = "Production-ready multi-cloud database abstraction layer with connection pooling, retry logic, and thread safety"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"

{altcodepro_polydb_python-2.5.23 → altcodepro_polydb_python-2.5.24/src/altcodepro_polydb_python.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: altcodepro-polydb-python
-Version: 2.5.23
+Version: 2.5.24
 Summary: Production-ready multi-cloud database abstraction layer with connection pooling, retry logic, and thread safety
 Author: AltCodePro
 Project-URL: Homepage, https://github.com/altcodepro/polydb-python

{altcodepro_polydb_python-2.5.23 → altcodepro_polydb_python-2.5.24}/src/altcodepro_polydb_python.egg-info/SOURCES.txt

@@ -96,6 +96,7 @@ tests/test_azure_queue_receipt_handle.py
 tests/test_azure_table_empty_filter.py
 tests/test_azure_table_id_property_persisted.py
 tests/test_azure_table_id_return_value_sanitization_mismatch.py
+tests/test_azure_table_key_field_filter.py
 tests/test_azure_table_large_record_patch_roundtrip.py
 tests/test_azure_table_self_heal.py
 tests/test_azure_table_unpack_entity.py

{altcodepro_polydb_python-2.5.23 → altcodepro_polydb_python-2.5.24}/src/polydb/__init__.py

@@ -4,7 +4,7 @@ PolyDB - Enterprise Cloud-Independent Database Abstraction
 Full LINQ support, field-level audit, cache, soft delete, overflow storage
 """
 
-__version__ = "2.5.23"
+__version__ = "2.5.24"
 
 from .aio import AsyncDatabaseFactory, AsyncPolyDB
 from .audit.context import AuditContext

{altcodepro_polydb_python-2.5.23 → altcodepro_polydb_python-2.5.24}/src/polydb/adapters/AzureTableStorageAdapter.py

@@ -750,11 +750,33 @@ class AzureTableStorageAdapter(NoSQLKVAdapter):
 
             eff_filters = dict(filters or {})
             parts: List[str] = []
+            # An equality on the model's own pk/rk FIELD is a key lookup: filtering the same value as an ordinary
+            # property made Azure scan the whole table (1.2s for one project's artifacts). Only valid when the key is
+            # built straight from that field (no partition_config template), which is how _get_pk_rk derives it.
+            pk_field = rk_field = None
+            declared = getattr(model, "__polydb__", None) or {}
+            if not getattr(self, "partition_config", None):
+                # Only fields the model declares itself; the defaults (tenant_id / id) are left as property filters
+                # so rows written before a model declared its keys keep matching.
+                pk_field = declared.get("pk_field") or None
+                rk_field = declared.get("rk_field") or None
             for orig_k, orig_v in eff_filters.items():
                 if orig_v is None:
                     continue
 
-                if orig_k in ("partition_key", "PartitionKey"):
+                if (
+                    (pk_field or rk_field)
+                    and isinstance(orig_v, (str, int))
+                    and not isinstance(orig_v, bool)
+                    and orig_k in (pk_field, rk_field)
+                ):
+                    # Keys are sanitized on write, so the key filter uses the sanitized value; the original
+                    # property filter below is kept too because sanitizing can map two raw values to one key.
+                    key_name = "PartitionKey" if orig_k == pk_field else "RowKey"
+                    safe = self._sanitize_pk_rk(orig_v).replace("'", "''")
+                    parts.append(f"{key_name} eq '{safe}'")
+                    sk = self._sanitize_prop_name(orig_k)
+                elif orig_k in ("partition_key", "PartitionKey"):
                     sk = "PartitionKey"
                 elif orig_k in ("row_key", "RowKey"):
                     sk = "RowKey"

altcodepro_polydb_python-2.5.24/tests/test_azure_table_key_field_filter.py

@@ -0,0 +1,55 @@
+"""An equality on the model's pk/rk field must be sent as PartitionKey/RowKey, not as a property filter
+(a property filter scans the whole table)."""
+import pytest
+
+pytest.importorskip("azure.data.tables")
+
+from polydb.adapters.AzureTableStorageAdapter import AzureTableStorageAdapter  # noqa: E402
+
+
+class _Model:
+    __name__ = "artifacts"
+    __polydb__ = {"pk_field": "project_slug", "rk_field": "artifact_id"}
+
+
+class _Fake:
+    def __init__(self):
+        self.calls = []
+
+    def query_entities(self, query_filter=None, **kw):
+        self.calls.append(query_filter)
+        return []
+
+
+def _adapter(fake, partition_config=None):
+    a = object.__new__(AzureTableStorageAdapter)
+    a._get_table_client = lambda model: fake
+    a.partition_config = partition_config
+    return a
+
+
+def test_pk_field_equality_becomes_partition_key():
+    f = _Fake()
+    _adapter(f)._query_raw(_Model, {"project_id": "p1", "project_slug": "p1", "tenant_id": "t"}, 500)
+    flt = f.calls[0]
+    assert "PartitionKey eq 'p1'" in flt
+    assert "project_slug eq 'p1'" in flt  # exactness filter kept
+    assert "project_id eq 'p1'" in flt and "tenant_id eq 't'" in flt
+
+
+def test_rk_field_equality_becomes_row_key():
+    f = _Fake()
+    _adapter(f)._query_raw(_Model, {"artifact_id": "a1"}, 1)
+    assert f.calls[0] == "RowKey eq 'a1' and artifact_id eq 'a1'"
+
+
+def test_partition_config_keeps_property_filter():
+    f = _Fake()
+    _adapter(f, partition_config=object())._query_raw(_Model, {"project_slug": "p1"}, 1)
+    assert "PartitionKey" not in f.calls[0]
+
+
+def test_key_value_is_sanitized_like_the_write_path():
+    f = _Fake()
+    _adapter(f)._query_raw(_Model, {"project_slug": "a:b"}, 1)
+    assert "PartitionKey eq 'a_b'" in f.calls[0]

{altcodepro_polydb_python-2.5.23 → altcodepro_polydb_python-2.5.24}/tests/test_read_projection.py

@@ -101,7 +101,7 @@ def test_azure_sends_select_with_the_keys_the_row_needs():
     c = _Client([ENT])
     rows = _adp(c).query(_Model, {"tenant_id": "t1"}, fields=["name", "status"])
     (flt, kw), = c.kwargs
-    assert flt == "tenant_id eq 't1'"
+    assert flt == "PartitionKey eq 't1' and tenant_id eq 't1'"
     assert set(kw["select"]) == {"name", "status", "PartitionKey", "RowKey", "__polydb_model__", "__keymap__"}
     assert rows == [{"name": "n", "status": "ok"}]  # id / tenant_id added by unpack are trimmed back out