django-bulk-hooks 0.2.54__tar.gz → 0.2.56__tar.gz

{django_bulk_hooks-0.2.54 → django_bulk_hooks-0.2.56}/PKG-INFO

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

@@ -207,6 +207,8 @@ class MTIHandler:
             update_conflicts=update_conflicts,
             unique_fields=unique_fields,
             update_fields=update_fields,
+            existing_record_ids=existing_record_ids,
+            existing_pks_map=existing_pks_map,
         )
 
         # Build child object templates (without parent links - executor adds them)
@@ -235,6 +237,8 @@ class MTIHandler:
         update_conflicts=False,
         unique_fields=None,
         update_fields=None,
+        existing_record_ids=None,
+        existing_pks_map=None,
     ):
         """
         Build parent level objects for each level in the inheritance chain.
@@ -248,6 +252,12 @@ class MTIHandler:
 
         parent_levels = []
         parent_instances_map = {}  # Maps obj id() -> {model_class: parent_instance}
+        
+        # Set defaults
+        if existing_record_ids is None:
+            existing_record_ids = set()
+        if existing_pks_map is None:
+            existing_pks_map = {}
 
         for level_idx, model_class in enumerate(inheritance_chain[:-1]):
             parent_objs_for_level = []
@@ -301,6 +311,25 @@ class MTIHandler:
                         level_update_conflicts = True
                         level_unique_fields = normalized_unique
                         level_update_fields = filtered_updates
+                else:
+                    # CRITICAL FIX: Even if this parent level doesn't have the unique constraint,
+                    # we still need update_conflicts=True during MTI upsert. Without it, parent
+                    # does plain INSERT creating a new parent record with a new PK, breaking the
+                    # MTI relationship and causing child INSERT to fail on its unique constraint.
+                    #
+                    # Solution: Use the primary key as the unique field for parent levels.
+                    # In MTI upsert with existing_pks_map, parent objects will get their PKs from
+                    # bulk_create, and we need those PKs to match existing records, not create new ones.
+                    if existing_record_ids and existing_pks_map:
+                        # Use primary key for upsert matching at this level
+                        pk_field = model_class._meta.pk
+                        level_update_conflicts = True
+                        level_unique_fields = [pk_field.name]
+                        # Use a safe update field - pick the first available non-PK field
+                        # or use the PK itself as a dummy (updating to itself is a no-op)
+                        available_fields = [f.name for f in model_class._meta.local_fields 
+                                          if not isinstance(f, AutoField) and f.name in model_fields_by_name]
+                        level_update_fields = available_fields[:1] if available_fields else [pk_field.name]
 
             # Create parent level
             parent_level = ParentLevel(
@@ -361,6 +390,13 @@ class MTIHandler:
 
         # Copy field values from source using centralized field extraction
         for field in parent_model._meta.local_fields:
+            # Skip AutoField (primary key) - let Django's bulk_create with update_conflicts
+            # handle PK assignment based on unique_fields. Setting PKs manually here causes
+            # conflicts when parent records exist but update_conflicts isn't properly configured
+            # for this level, leading to IntegrityError on primary key constraint.
+            if isinstance(field, AutoField):
+                continue
+
             if hasattr(source_obj, field.name):
                 # Use centralized field value extraction for consistent FK handling
                 value = get_field_value_for_db(source_obj, field.name, source_obj.__class__)

{django_bulk_hooks-0.2.54 → django_bulk_hooks-0.2.56}/pyproject.toml

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