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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: django-bulk-hooks
-Version: 0.2.47
+Version: 0.2.49
 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.49}/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.49}/django_bulk_hooks/operations/mti_handler.py

@@ -78,8 +78,7 @@ class MTIHandler:
                 chain.append(current_model)
 
             # Get concrete parent models (not abstract, not proxy)
-            parents = [parent for parent in current_model._meta.parents.keys() 
-                      if not parent._meta.proxy and not parent._meta.abstract]
+            parents = [parent for parent in current_model._meta.parents.keys() if not parent._meta.proxy and not parent._meta.abstract]
 
             current_model = parents[0] if parents else None
 
@@ -113,31 +112,33 @@ class MTIHandler:
 
     def find_model_with_unique_fields(self, unique_fields):
         """
-        Find which model in the inheritance chain contains the unique fields.
-        
-        This is critical for MTI upserts: we need to query the model that has
-        the unique constraint, not necessarily the child model.
-        
+        Find which model in the inheritance chain to query for existing records.
+
+        For MTI upsert operations, we need to determine if the parent record exists
+        to properly fire AFTER_CREATE vs AFTER_UPDATE hooks. This is critical because:
+        - If parent exists but child doesn't: creating child for existing parent → AFTER_UPDATE
+        - If neither exists: creating both parent and child → AFTER_CREATE
+
+        Therefore, we return the root parent model to check if the parent record exists,
+        regardless of where the unique fields are defined.
+
         Args:
             unique_fields: List of field names forming the unique constraint
-        
+
         Returns:
-            Model class that contains all the unique fields (closest to root)
+            Model class to query for existing records (root parent for MTI)
         """
         if not unique_fields:
             return self.model_cls
-        
+
         inheritance_chain = self.get_inheritance_chain()
-        
-        # Start from root and find the first model that has all unique fields
-        for model_cls in inheritance_chain:
-            model_field_names = {f.name for f in model_cls._meta.local_fields}
-            
-            # Check if this model has all the unique fields
-            if all(field in model_field_names for field in unique_fields):
-                return model_cls
-        
-        # Fallback to child model (shouldn't happen if unique_fields are valid)
+
+        # For MTI models with multiple levels, return the root parent model
+        # This ensures we check if the parent exists, which determines create vs update hooks
+        if len(inheritance_chain) > 1:
+            return inheritance_chain[0]  # Root parent model
+
+        # For non-MTI models (shouldn't happen, but safe fallback)
         return self.model_cls
 
     # ==================== MTI BULK CREATE PLANNING ====================

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

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "django-bulk-hooks"
-version = "0.2.47"
+version = "0.2.49"
 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"