django-bulk-hooks 0.1.184__py3-none-any.whl → 0.1.185__py3-none-any.whl

django_bulk_hooks/queryset.py

@@ -537,6 +537,19 @@ class HookQuerySet(models.QuerySet):
         model_cls = self.model
         inheritance_chain = self._get_inheritance_chain()
 
+        # Remove custom hook kwargs before passing to Django internals
+        django_kwargs = {
+            k: v
+            for k, v in kwargs.items()
+            if k not in ["bypass_hooks", "bypass_validation"]
+        }
+
+        # Safety check to prevent infinite recursion
+        if len(inheritance_chain) > 10:  # Arbitrary limit to prevent infinite loops
+            raise ValueError(
+                "Inheritance chain too deep - possible infinite recursion detected"
+            )
+
         # Group fields by model in the inheritance chain
         field_groups = {}
         for field_name in fields:
@@ -549,8 +562,28 @@ class HookQuerySet(models.QuerySet):
                     field_groups[model].append(field_name)
                     break
 
-        # Update each table in the inheritance chain
+        # Process in batches
+        batch_size = django_kwargs.get("batch_size") or len(objs)
         total_updated = 0
+        
+        with transaction.atomic(using=self.db, savepoint=False):
+            for i in range(0, len(objs), batch_size):
+                batch = objs[i : i + batch_size]
+                batch_result = self._process_mti_bulk_update_batch(
+                    batch, field_groups, inheritance_chain, **django_kwargs
+                )
+                total_updated += batch_result
+
+        return total_updated
+
+    def _process_mti_bulk_update_batch(self, batch, field_groups, inheritance_chain, **kwargs):
+        """
+        Process a single batch of objects for MTI bulk update.
+        Updates each table in the inheritance chain for the batch.
+        """
+        total_updated = 0
+        
+        # Update each table in the inheritance chain
         for model, model_fields in field_groups.items():
             if not model_fields:
                 continue
@@ -560,7 +593,7 @@ class HookQuerySet(models.QuerySet):
             # Child models use the parent link to reference the root PK
             if model == inheritance_chain[0]:
                 # Root model - use primary keys directly
-                pks = [obj.pk for obj in objs]
+                pks = [obj.pk for obj in batch]
                 filter_field = 'pk'
             else:
                 # Child model - use parent link field
@@ -575,25 +608,25 @@ class HookQuerySet(models.QuerySet):
                     continue
                 
                 # Get the parent link values (these should be the same as the root PKs)
-                pks = [getattr(obj, parent_link.attname) for obj in objs]
+                pks = [getattr(obj, parent_link.attname) for obj in batch]
                 filter_field = parent_link.attname
             
             if pks:
                 base_qs = model._base_manager.using(self.db)
                 
-                # Build CASE statements for each field
+                # Build CASE statements for each field to perform a single bulk update
                 case_statements = {}
                 for field_name in model_fields:
                     field = model._meta.get_field(field_name)
                     when_statements = []
                     
-                    for pk, obj in zip(pks, objs):
+                    for pk, obj in zip(pks, batch):
                         value = getattr(obj, field_name)
                         when_statements.append(When(**{filter_field: pk}, then=Value(value, output_field=field)))
                     
                     case_statements[field_name] = Case(*when_statements, output_field=field)
                 
-                # Execute the update using the appropriate filter field
+                # Execute a single bulk update for all objects in this model
                 updated_count = base_qs.filter(**{f"{filter_field}__in": pks}).update(**case_statements)
                 total_updated += updated_count
 

{django_bulk_hooks-0.1.184.dist-info → django_bulk_hooks-0.1.185.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: django-bulk-hooks
-Version: 0.1.184
+Version: 0.1.185
 Summary: Hook-style hooks for Django bulk operations like bulk_create and bulk_update.
 License: MIT
 Keywords: django,bulk,hooks

{django_bulk_hooks-0.1.184.dist-info → django_bulk_hooks-0.1.185.dist-info}/RECORD

@@ -9,9 +9,9 @@ django_bulk_hooks/handler.py,sha256=xZt8iNdYF-ACz-MnKMY0co6scWINU5V5wC1lyDn844k,
 django_bulk_hooks/manager.py,sha256=KLEjpQRt4WlzgBAf_X3XOAPUQM8Jmc1fIt8yr62FPQc,3044
 django_bulk_hooks/models.py,sha256=7fnx5xd4HWXfLVlFhhiRzR92JRWFEuxgk6aSWLEsyJg,3996
 django_bulk_hooks/priority.py,sha256=HG_2D35nga68lBCZmSXTcplXrjFoRgZFRDOy4ROKonY,376
-django_bulk_hooks/queryset.py,sha256=gQvwa506wYi206B13I-lwkZi5Ua9nWcKctwCCgLGuaQ,25614
+django_bulk_hooks/queryset.py,sha256=9ME8FNy_DdF6As-q1mb81u7agzfMR2rNGz6NlVB5bRI,26976
 django_bulk_hooks/registry.py,sha256=-mQBizJ06nz_tajZBinViKx_uP2Tbc1tIpTEMv7lwKA,705
-django_bulk_hooks-0.1.184.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
-django_bulk_hooks-0.1.184.dist-info/METADATA,sha256=zj7hawgW0Z19ZUhnTcTMcx-pMXy_PvepfzUdstynZjU,6951
-django_bulk_hooks-0.1.184.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-django_bulk_hooks-0.1.184.dist-info/RECORD,,
+django_bulk_hooks-0.1.185.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
+django_bulk_hooks-0.1.185.dist-info/METADATA,sha256=KO4qnnw3JdSvWDFUiXipL5SIHPKm79HufxSJBNqgDTk,6951
+django_bulk_hooks-0.1.185.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+django_bulk_hooks-0.1.185.dist-info/RECORD,,