django-bulk-hooks 0.2.30__py3-none-any.whl → 0.2.32__py3-none-any.whl

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,29 +444,27 @@ 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:
                             continue
 
                         # Get the field value - handle ForeignKey fields specially
-                        if isinstance(field, ForeignKey):
-                            value = getattr(obj, field.attname, None)
-                            # For ForeignKey fields, use the target field for proper type casting
-                            output_field = field.target_field
-                        else:
-                            value = getattr(obj, field_name)
-                            output_field = field
-                        
+                        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=output_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.30.dist-info → django_bulk_hooks-0.2.32.dist-info}/METADATA

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

@@ -13,14 +13,14 @@ 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=8xNdsoeOCHsIIP7ztIuFHDewBR2dDSS6auOrSU8zmqE,9007
-django_bulk_hooks/operations/bulk_executor.py,sha256=D7eteG0116fvI-waiP0PhWl2kBd_b3Iv--p_nWO1l2Y,21597
+django_bulk_hooks/operations/bulk_executor.py,sha256=Aqnnr3RqDh4KJgZz-mQBlAl8ScZE3NhMC2gqdAxcm-4,21224
 django_bulk_hooks/operations/coordinator.py,sha256=iSi-x3fq6_3UygWGeQY8Fq_gZWIqq_8N7WddPhVeygk,23526
 django_bulk_hooks/operations/mti_handler.py,sha256=G-pxkzIqHqXGshRGksqmsN1J3rlzePUZrSv4wm7D3cQ,19162
 django_bulk_hooks/operations/mti_plans.py,sha256=YP7LcV9Z8UqNS_x74OswF9_5swqruRTdAu6z-J_R6C0,3377
 django_bulk_hooks/operations/record_classifier.py,sha256=KzUoAhfoqzFVrOabNZAby9Akb54h-fAQZmb8O-fIx_0,6221
 django_bulk_hooks/queryset.py,sha256=rvJgQLwtSJztwc68nkJ6xwCsnbXQvkvS6_dbGGj8TFo,5886
 django_bulk_hooks/registry.py,sha256=QyeA2OqNdMAMaLjFU9UF0YGhKiPKbZkmFQpLgof7uNs,9038
-django_bulk_hooks-0.2.30.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
-django_bulk_hooks-0.2.30.dist-info/METADATA,sha256=A-StNulrNdNpSvtR4NJ3ERvr56mKsKiPeUHN9_FTKj8,9265
-django_bulk_hooks-0.2.30.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-django_bulk_hooks-0.2.30.dist-info/RECORD,,
+django_bulk_hooks-0.2.32.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
+django_bulk_hooks-0.2.32.dist-info/METADATA,sha256=SPBG50Vgbab8Lo8bqOI4luL0tU_JZRMsWPgUVwz2eN0,9265
+django_bulk_hooks-0.2.32.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+django_bulk_hooks-0.2.32.dist-info/RECORD,,