GeneralManager 0.41.4__tar.gz → 0.41.5__tar.gz

{generalmanager-0.41.4/src/GeneralManager.egg-info → generalmanager-0.41.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: GeneralManager
-Version: 0.41.4
+Version: 0.41.5
 Summary: Modular Django-based data management framework with ORM, GraphQL, fine-grained permissions, rule validation, calculations and caching.
 Author-email: Tim Kleindick <tkleindick@yahoo.de>
 License: MIT License

{generalmanager-0.41.4 → generalmanager-0.41.5}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "GeneralManager"
-version = "0.41.4"
+version = "0.41.5"
 description = "Modular Django-based data management framework with ORM, GraphQL, fine-grained permissions, rule validation, calculations and caching."
 readme = "README.md"
 authors = [{ name = "Tim Kleindick", email = "tkleindick@yahoo.de" }]

{generalmanager-0.41.4 → generalmanager-0.41.5/src/GeneralManager.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: GeneralManager
-Version: 0.41.4
+Version: 0.41.5
 Summary: Modular Django-based data management framework with ORM, GraphQL, fine-grained permissions, rule validation, calculations and caching.
 Author-email: Tim Kleindick <tkleindick@yahoo.de>
 License: MIT License

{generalmanager-0.41.4 → generalmanager-0.41.5}/src/general_manager/bucket/database_bucket.py

@@ -140,6 +140,57 @@ class NonSortablePropertyError(ValueError):
         )
 
 
+class DuplicateDatabaseBucketSnapshotError(ValueError):
+    """Raised when a database bucket snapshot contains duplicate primary keys."""
+
+    def __init__(self) -> None:
+        super().__init__(
+            "DatabaseBucket snapshots cannot contain duplicate primary keys."
+        )
+
+
+def _ensure_unique_primary_keys(primary_keys: tuple[Any, ...]) -> None:
+    if len(primary_keys) != len(set(primary_keys)):
+        raise DuplicateDatabaseBucketSnapshotError()
+
+
+def _restore_database_bucket_from_primary_keys(
+    model: type[models.Model],
+    manager_class: Type[GeneralManagerType],
+    primary_keys: tuple[Any, ...],
+    filter_definitions: dict[str, list[Any]],
+    exclude_definitions: dict[str, list[Any]],
+    database_alias: str | None,
+    search_date: datetime | date | None,
+    sort_keys: tuple[str, ...] | None,
+    sort_reverse: bool,
+) -> DatabaseBucket[GeneralManagerType]:
+    _ensure_unique_primary_keys(primary_keys)
+    manager = model._default_manager
+    if database_alias is not None:
+        manager = manager.db_manager(database_alias)
+    if not primary_keys:
+        queryset = manager.none()
+    else:
+        preserved_order = models.Case(
+            *(
+                models.When(pk=primary_key, then=position)
+                for position, primary_key in enumerate(primary_keys)
+            ),
+            output_field=models.IntegerField(),
+        )
+        queryset = manager.filter(pk__in=primary_keys).order_by(preserved_order)
+    return DatabaseBucket(
+        queryset,
+        manager_class,
+        filter_definitions,
+        exclude_definitions,
+        search_date=search_date,
+        sort_keys=sort_keys,
+        sort_reverse=sort_reverse,
+    )
+
+
 class DatabaseBucket(Bucket[GeneralManagerType]):
     """Bucket implementation backed by Django ORM querysets."""
 
@@ -175,6 +226,28 @@ class DatabaseBucket(Bucket[GeneralManagerType]):
         self._sort_keys = sort_keys
         self._sort_reverse = sort_reverse
 
+    def __reduce__(self) -> str | tuple[Any, ...]:
+        """
+        Preserve a result snapshot without serializing the queryset object.
+        """
+        self._track_effective_dependencies()
+        primary_keys = tuple(self._data.values_list("pk", flat=True))
+        _ensure_unique_primary_keys(primary_keys)
+        return (
+            _restore_database_bucket_from_primary_keys,
+            (
+                self._data.model,
+                self._manager_class,
+                primary_keys,
+                self.filters,
+                self.excludes,
+                self._data.db,
+                self._search_date,
+                self._sort_keys,
+                self._sort_reverse,
+            ),
+        )
+
     def _build_manager(self, pk: Any) -> GeneralManagerType:
         if self._search_date is None:
             return self._manager_class(pk)