django-bulk-hooks 0.2.29__tar.gz → 0.2.31__tar.gz

{django_bulk_hooks-0.2.29 → django_bulk_hooks-0.2.31}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: django-bulk-hooks
-Version: 0.2.29
+Version: 0.2.31
 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.29 → django_bulk_hooks-0.2.31}/django_bulk_hooks/operations/bulk_executor.py

@@ -77,9 +77,7 @@ class BulkExecutor:
             existing_record_ids = set()
             existing_pks_map = {}
             if update_conflicts and unique_fields:
-                existing_record_ids, existing_pks_map = (
-                    self.record_classifier.classify_for_upsert(objs, unique_fields)
-                )
+                existing_record_ids, existing_pks_map = self.record_classifier.classify_for_upsert(objs, unique_fields)
 
             # Build execution plan with classification results
             plan = self.mti_handler.build_create_plan(
@@ -173,13 +171,13 @@ class BulkExecutor:
     def _execute_mti_create_plan(self, plan):
         """
         Execute an MTI create plan.
-        
+
         This is where ALL database operations happen for MTI bulk_create.
         Handles both new records (INSERT) and existing records (UPDATE) for upsert.
-        
+
         Args:
             plan: MTICreatePlan object from MTIHandler
-            
+
         Returns:
             List of created/updated objects with PKs assigned
         """
@@ -230,13 +228,8 @@ class BulkExecutor:
                 # Update existing parents
                 if existing_parents and parent_level.update_fields:
                     # Filter update fields to only those that exist in this parent model
-                    parent_model_fields = {
-                        field.name for field in parent_level.model_class._meta.local_fields
-                    }
-                    filtered_update_fields = [
-                        field for field in parent_level.update_fields
-                        if field in parent_model_fields
-                    ]
+                    parent_model_fields = {field.name for field in parent_level.model_class._meta.local_fields}
+                    filtered_update_fields = [field for field in parent_level.update_fields if field in parent_model_fields]
 
                     if filtered_update_fields:
                         base_qs = BaseQuerySet(model=parent_level.model_class, using=self.queryset.db)
@@ -317,10 +310,7 @@ class BulkExecutor:
                             obj._state.db = self.queryset.db
 
                 if objs_without_pk:
-                    filtered_fields = [
-                        f for f in fields
-                        if not isinstance(f, AutoField) and not f.primary_key
-                    ]
+                    filtered_fields = [f for f in fields if not isinstance(f, AutoField) and not f.primary_key]
                     returned_columns = base_qs._batched_insert(
                         objs_without_pk,
                         filtered_fields,
@@ -341,13 +331,8 @@ class BulkExecutor:
             # Step 3.5: Update existing child objects
             if existing_child_objects and plan.update_fields:
                 # Filter update fields to only those that exist in the child model
-                child_model_fields = {
-                    field.name for field in plan.child_model._meta.local_fields
-                }
-                filtered_child_update_fields = [
-                    field for field in plan.update_fields
-                    if field in child_model_fields
-                ]
+                child_model_fields = {field.name for field in plan.child_model._meta.local_fields}
+                filtered_child_update_fields = [field for field in plan.update_fields if field in child_model_fields]
 
                 if filtered_child_update_fields:
                     base_qs = BaseQuerySet(model=plan.child_model, using=self.queryset.db)
@@ -393,9 +378,11 @@ class BulkExecutor:
                                 continue
 
                         # Copy auto_now_add, auto_now, and db_returning fields
-                        if (getattr(field, "auto_now_add", False) or
-                            getattr(field, "auto_now", False) or
-                            getattr(field, "db_returning", False)):
+                        if (
+                            getattr(field, "auto_now_add", False)
+                            or getattr(field, "auto_now", False)
+                            or getattr(field, "db_returning", False)
+                        ):
                             source_value = getattr(source_obj, field.name, None)
                             if source_value is not None:
                                 setattr(orig_obj, field.name, source_value)
@@ -409,12 +396,12 @@ class BulkExecutor:
     def _execute_mti_update_plan(self, plan):
         """
         Execute an MTI update plan.
-        
+
         Updates each table in the inheritance chain using CASE/WHEN for bulk updates.
-        
+
         Args:
             plan: MTIUpdatePlan object from MTIHandler
-            
+
         Returns:
             Number of objects updated
         """
@@ -457,6 +444,10 @@ class BulkExecutor:
                     field = field_group.model_class._meta.get_field(field_name)
                     when_statements = []
 
+                    # Determine the correct output field for type casting
+                    # For ForeignKey fields, use the target field to ensure correct SQL types
+                    case_output_field = field.target_field if isinstance(field, ForeignKey) else field
+
                     for pk, obj in zip(root_pks, plan.objects):
                         obj_pk = getattr(obj, "pk", None) or getattr(obj, "id", None)
                         if obj_pk is None:
@@ -464,16 +455,16 @@ class BulkExecutor:
 
                         # Get the field value - handle ForeignKey fields specially
                         value = getattr(obj, field.attname, None) if isinstance(field, ForeignKey) else getattr(obj, field_name)
-                        
+
                         when_statements.append(
                             When(
                                 **{field_group.filter_field: pk},
-                                then=Value(value, output_field=field),
+                                then=Value(value, output_field=case_output_field),
                             ),
                         )
 
                     if when_statements:
-                        case_statements[field_name] = Case(*when_statements, output_field=field)
+                        case_statements[field_name] = Case(*when_statements, output_field=case_output_field)
 
                 # Execute bulk update
                 if case_statements:

{django_bulk_hooks-0.2.29 → django_bulk_hooks-0.2.31}/pyproject.toml

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