django-bulk-hooks 0.2.56__py3-none-any.whl → 0.2.57__py3-none-any.whl

django_bulk_hooks/operations/bulk_executor.py

@@ -8,6 +8,7 @@ import logging
 
 from django.db import transaction
 from django.db.models import AutoField, ForeignKey, Case, When, Value
+from django.db.models.constants import OnConflict
 from django.db.models.functions import Cast
 
 from django_bulk_hooks.operations.field_utils import get_field_value_for_db
@@ -333,15 +334,64 @@ class BulkExecutor:
             if objs_without_pk:
                 base_qs._prepare_for_bulk_create(objs_without_pk)
                 opts = plan.child_model._meta
-                
+
                 # Include all local fields except auto-generated ones
                 # For MTI, we need to include the parent link (which is the PK)
                 filtered_fields = [f for f in opts.local_fields if not f.generated]
-                
+
+                # FIX: Pass conflict resolution parameters to _batched_insert for MTI child tables
+                # Previously, _batched_insert was called without on_conflict/unique_fields/update_fields,
+                # causing IntegrityError when child tables have unique constraints during upsert operations.
+                # See: https://github.com/user/repo/issues/XXX
+                # Prepare conflict resolution parameters for upsert
+                on_conflict = None
+                batched_unique_fields = None
+                batched_update_fields = None
+
+                if plan.update_conflicts:
+                    # Filter unique_fields and update_fields to only those on child model
+                    # Django's _batched_insert expects field objects, not field names
+                    child_model_fields_dict = {field.name: field for field in plan.child_model._meta.local_fields}
+
+                    # Unique fields may be on parent or child - filter to child only for child table insert
+                    # Convert field names to field objects
+                    if plan.unique_fields:
+                        batched_unique_fields = [
+                            child_model_fields_dict[fname]
+                            for fname in plan.unique_fields
+                            if fname in child_model_fields_dict
+                        ]
+
+                    # Update fields - filter to child only
+                    # Keep as strings - Django's _batched_insert accepts field name strings for update_fields
+                    if plan.update_fields:
+                        batched_update_fields = [
+                            fname
+                            for fname in plan.update_fields
+                            if fname in child_model_fields_dict
+                        ]
+
+                    # Only set on_conflict if we have unique fields for this table
+                    # Note: If unique_fields are all on parent, batched_unique_fields will be empty,
+                    # meaning no conflict resolution needed for child table
+                    if batched_unique_fields:
+                        if batched_update_fields:
+                            # We have both unique fields and update fields on child - use UPDATE
+                            on_conflict = OnConflict.UPDATE
+                        else:
+                            # We have unique fields on child but no update fields - use IGNORE
+                            # This handles the case where all update fields are on parent tables
+                            on_conflict = OnConflict.IGNORE
+                            # Clear batched_update_fields to avoid issues
+                            batched_update_fields = None
+
                 returned_columns = base_qs._batched_insert(
                     objs_without_pk,
                     filtered_fields,
                     batch_size=len(objs_without_pk),
+                    on_conflict=on_conflict,
+                    update_fields=batched_update_fields,
+                    unique_fields=batched_unique_fields,
                 )
                 if returned_columns:
                     for obj, results in zip(objs_without_pk, returned_columns):

django_bulk_hooks/operations/mti_handler.py

@@ -252,7 +252,7 @@ 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()
@@ -327,8 +327,11 @@ class MTIHandler:
                         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]
+                        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

{django_bulk_hooks-0.2.56.dist-info → django_bulk_hooks-0.2.57.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: django-bulk-hooks
-Version: 0.2.56
+Version: 0.2.57
 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.56.dist-info → django_bulk_hooks-0.2.57.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=tMbcwQSN8xNoKL7-KC8m-XhGyqwaAtAQ5CAm3PYHl_Q,24122
+django_bulk_hooks/operations/bulk_executor.py,sha256=Y-wkvuV_X-SZmI965JVrrtwbzPZVggUfy8mR1pzP9d0,27048
 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=l1BPeqK1eTjloMxBPZPVAaWyBK14j1jUFVELdl-GF1E,23441
+django_bulk_hooks/operations/mti_handler.py,sha256=MMDxuE6jzwkhMwVE-6qJUdp4MNKgAIFPMjwdN_umrqs,23503
 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.56.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
-django_bulk_hooks-0.2.56.dist-info/METADATA,sha256=0r_GZFFTjdK1IJBe3iSsuSgSky38xbyMLzbA4CAy150,9265
-django_bulk_hooks-0.2.56.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-django_bulk_hooks-0.2.56.dist-info/RECORD,,
+django_bulk_hooks-0.2.57.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
+django_bulk_hooks-0.2.57.dist-info/METADATA,sha256=SWZOMPOgiBDCLuSxHX9xTeJMqcd3CgvDZP-JuoY4wRs,9265
+django_bulk_hooks-0.2.57.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+django_bulk_hooks-0.2.57.dist-info/RECORD,,