django-bulk-hooks 0.2.47__tar.gz → 0.2.48__tar.gz

{django_bulk_hooks-0.2.47 → django_bulk_hooks-0.2.48}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: django-bulk-hooks
-Version: 0.2.47
+Version: 0.2.48
 Summary: Hook-style hooks for Django bulk operations like bulk_create and bulk_update.
 License: MIT
 Keywords: django,bulk,hooks

{django_bulk_hooks-0.2.47 → django_bulk_hooks-0.2.48}/django_bulk_hooks/operations/bulk_executor.py

@@ -107,7 +107,7 @@ class BulkExecutor:
 
             # Tag objects with upsert metadata for hook dispatching
             if update_conflicts and unique_fields:
-                self._tag_upsert_metadata(result, existing_record_ids)
+                self._tag_upsert_metadata(result, existing_record_ids, existing_pks_map)
 
             return result
 
@@ -126,8 +126,8 @@ class BulkExecutor:
         if update_conflicts and unique_fields:
             # Use pre-classified results if available, otherwise classify now
             if existing_record_ids is None:
-                existing_record_ids, _ = self.record_classifier.classify_for_upsert(objs, unique_fields)
-            self._tag_upsert_metadata(result, existing_record_ids)
+                existing_record_ids, existing_pks_map = self.record_classifier.classify_for_upsert(objs, unique_fields)
+            self._tag_upsert_metadata(result, existing_record_ids, existing_pks_map)
 
         return result
 
@@ -507,7 +507,7 @@ class BulkExecutor:
 
         return QuerySet.delete(self.queryset)
 
-    def _tag_upsert_metadata(self, result_objects, existing_record_ids):
+    def _tag_upsert_metadata(self, result_objects, existing_record_ids, existing_pks_map):
         """
         Tag objects with metadata indicating whether they were created or updated.
 
@@ -517,13 +517,18 @@ class BulkExecutor:
         Args:
             result_objects: List of objects returned from bulk operation
             existing_record_ids: Set of id() for objects that existed before the operation
+            existing_pks_map: Dict mapping id(obj) -> pk for existing records
         """
         created_count = 0
         updated_count = 0
 
+        # Create a set of PKs that existed before the operation
+        existing_pks = set(existing_pks_map.values())
+
         for obj in result_objects:
-            # Tag with metadata for hook dispatching
-            was_created = id(obj) not in existing_record_ids
+            # Use PK to determine if this record was created or updated
+            # If the PK was in the existing_pks_map, it was updated; otherwise created
+            was_created = obj.pk not in existing_pks
             obj._bulk_hooks_was_created = was_created
             obj._bulk_hooks_upsert_metadata = True
 
@@ -534,6 +539,6 @@ class BulkExecutor:
 
         logger.info(
             f"Tagged upsert metadata: {created_count} created, {updated_count} updated "
-            f"(total={len(result_objects)}, existing_ids={len(existing_record_ids)})"
+            f"(total={len(result_objects)}, existing_pks={len(existing_pks)})"
         )
 

{django_bulk_hooks-0.2.47 → django_bulk_hooks-0.2.48}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "django-bulk-hooks"
-version = "0.2.47"
+version = "0.2.48"
 description = "Hook-style hooks for Django bulk operations like bulk_create and bulk_update."
 authors = ["Konrad Beck <konrad.beck@merchantcapital.co.za>"]
 readme = "README.md"