django-bulk-hooks 0.2.52__py3-none-any.whl → 0.2.54__py3-none-any.whl

django_bulk_hooks/operations/bulk_executor.py

@@ -261,57 +261,46 @@ class BulkExecutor:
                         parent_instances_map[orig_obj_id] = {}
                     parent_instances_map[orig_obj_id][parent_level.model_class] = parent_obj
 
-            # Step 2: Add parent links to child objects and set PKs for existing records
+            # Step 2: Add parent links to child objects and set PKs appropriately
             for child_obj, orig_obj in zip(plan.child_objects, plan.original_objects):
                 parent_instances = parent_instances_map.get(id(orig_obj), {})
 
-                # Set parent links
+                # Set parent links and PKs for all objects (since in MTI, child PK = parent PK)
                 for parent_model, parent_instance in parent_instances.items():
                     parent_link = plan.child_model._meta.get_ancestor_link(parent_model)
                     if parent_link:
-                        setattr(child_obj, parent_link.attname, parent_instance.pk)
+                        parent_pk = parent_instance.pk
+                        setattr(child_obj, parent_link.attname, parent_pk)
                         setattr(child_obj, parent_link.name, parent_instance)
-                        # IMPORTANT: Don't set the child's PK here - it should only get PK after insertion
-                        # The parent link field (hookmodel_ptr) is NOT the same as the child's PK
+                        # In MTI, the child PK IS the parent link
+                        child_obj.pk = parent_pk
+                        child_obj.id = parent_pk
                     else:
                         logger.warning(f"No parent link found for {parent_model} in {plan.child_model}")
 
-                # For existing records in upsert, ensure PK is set on child object
-                if id(orig_obj) in plan.existing_record_ids:
-                    pk_value = getattr(orig_obj, "pk", None)
-                    if pk_value:
-                        child_obj.pk = pk_value
-                        child_obj.id = pk_value
-                    else:
-                        # If no PK on original object, this is a new record, don't set PK
-                        logger.info(f"New record {orig_obj} - not setting PK on child object")
-
             # Step 3: Handle child objects
             # Note: We can't use bulk_create on child MTI models, so we use _batched_insert for new records
             # and bulk_update for existing records
             base_qs = BaseQuerySet(model=plan.child_model, using=self.queryset.db)
-            
-            # For MTI child objects, we need to handle them differently
-            # In MTI, child objects get PKs from parent links, but we need to distinguish
-            # between truly new records and existing records for upsert operations
+
+            # For MTI child objects, we need to distinguish between truly new records and existing records for upsert operations
             objs_without_pk, objs_with_pk = [], []
-            
+
             # Check which CHILD records actually exist in the child table
-            # This is separate from checking parent existence
             if plan.update_conflicts:
-                # Query the CHILD table to see which child records exist
-                parent_pks = []
+                # For upsert, check which child records exist based on the parent PKs
+                parent_pks_to_check = []
                 for child_obj in plan.child_objects:
                     child_pk = getattr(child_obj, plan.child_model._meta.pk.attname, None)
                     if child_pk:
-                        parent_pks.append(child_pk)
-                
+                        parent_pks_to_check.append(child_pk)
+
                 existing_child_pks = set()
-                if parent_pks:
+                if parent_pks_to_check:
                     existing_child_pks = set(
-                        base_qs.filter(pk__in=parent_pks).values_list('pk', flat=True)
+                        base_qs.filter(pk__in=parent_pks_to_check).values_list('pk', flat=True)
                     )
-                
+
                 # Split based on whether child record exists
                 for child_obj in plan.child_objects:
                     child_pk = getattr(child_obj, plan.child_model._meta.pk.attname, None)

django_bulk_hooks/operations/mti_handler.py

@@ -406,7 +406,12 @@ class MTIHandler:
         """
         child_obj = child_model()
 
-        # Copy field values (excluding AutoField and parent links)
+        # Copy field values (excluding AutoField, parent links, and inherited fields)
+        # In MTI, child objects should only have values for fields defined directly on the child model
+        parent_fields = set()
+        for parent_model in child_model._meta.parents.keys():
+            parent_fields.update(f.name for f in parent_model._meta.local_fields)
+
         for field in child_model._meta.local_fields:
             if isinstance(field, AutoField):
                 continue
@@ -417,6 +422,10 @@ class MTIHandler:
                 if child_model._meta.get_ancestor_link(field.related_model) == field:
                     continue
 
+            # Skip inherited fields - these belong to parent models
+            if field.name in parent_fields:
+                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.52.dist-info → django_bulk_hooks-0.2.54.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: django-bulk-hooks
-Version: 0.2.52
+Version: 0.2.54
 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.52.dist-info → django_bulk_hooks-0.2.54.dist-info}/RECORD

@@ -13,15 +13,15 @@ django_bulk_hooks/manager.py,sha256=3mFzB0ZzHHeXWdKGObZD_H0NlskHJc8uYBF69KKdAXU,
 django_bulk_hooks/models.py,sha256=4Vvi2LiGP0g4j08a5liqBROfsO8Wd_ermBoyjKwfrPU,2512
 django_bulk_hooks/operations/__init__.py,sha256=BtJYjmRhe_sScivLsniDaZmBkm0ZLvcmzXFKL7QY2Xg,550
 django_bulk_hooks/operations/analyzer.py,sha256=wAG8sAG9NwfwNqG9z81VfGR7AANDzRmMGE_o82MWji4,10689
-django_bulk_hooks/operations/bulk_executor.py,sha256=F_nXtqB2ZRuAEZtSzqtKedqH_TDkNQDPLVmVUT8jsPQ,24794
+django_bulk_hooks/operations/bulk_executor.py,sha256=tMbcwQSN8xNoKL7-KC8m-XhGyqwaAtAQ5CAm3PYHl_Q,24122
 django_bulk_hooks/operations/coordinator.py,sha256=iGavJLqe3eYRqFay8cMn6muwyRYzQo-HFGphsS5hL6g,30799
 django_bulk_hooks/operations/field_utils.py,sha256=Tvr5bcZLG8imH-r2S85oui1Cbw6hGv3VtuIMn4OvsU4,2895
-django_bulk_hooks/operations/mti_handler.py,sha256=Ro80PGrgm9O9wWuFwhiarcOqLlU95iCOtuvaDIcshQU,20648
+django_bulk_hooks/operations/mti_handler.py,sha256=r6F49-6tHbCeu2DFj6PX_7iU4lv6icbNBI8u-v1Fj9U,21092
 django_bulk_hooks/operations/mti_plans.py,sha256=7STQ2oA2ZT8cEG3-t-6xciRAdf7OeSf0gRLXR_BRG-Q,3363
 django_bulk_hooks/operations/record_classifier.py,sha256=kqML4aO11X9K3SSJ5DUlUukwI172j_Tk12Kr77ee8q8,7065
 django_bulk_hooks/queryset.py,sha256=aQitlbexcVnmeAdc0jtO3hci39p4QEu4srQPEzozy5s,5546
 django_bulk_hooks/registry.py,sha256=uum5jhGI3TPaoiXuA1MdBdu4gbE3rQGGwQ5YDjiMcjk,7949
-django_bulk_hooks-0.2.52.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
-django_bulk_hooks-0.2.52.dist-info/METADATA,sha256=VijPFOPGvpd1c6xb3W4WEcjCN6RJP4ASCi_2wJ1U660,9265
-django_bulk_hooks-0.2.52.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-django_bulk_hooks-0.2.52.dist-info/RECORD,,
+django_bulk_hooks-0.2.54.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
+django_bulk_hooks-0.2.54.dist-info/METADATA,sha256=ClVI63n_CscKcG0C_7B9rxCCgN4WnDoO5Rl-kWs0FhM,9265
+django_bulk_hooks-0.2.54.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+django_bulk_hooks-0.2.54.dist-info/RECORD,,