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

django_bulk_hooks/queryset.py

@@ -178,9 +178,17 @@ class HookQuerySet(models.QuerySet):
         """
         Bulk update objects in the database with MTI support.
         """
+        print(f"\n=== BULK UPDATE DEBUG ===")
+        print(f"Model: {self.model.__name__}")
+        print(f"Number of objects: {len(objs)}")
+        print(f"Fields: {fields}")
+        print(f"Bypass hooks: {bypass_hooks}")
+        print(f"Bypass validation: {bypass_validation}")
+        
         model_cls = self.model
 
         if not objs:
+            print("No objects to update")
             return []
 
         if any(not isinstance(obj, model_cls) for obj in objs):
@@ -194,6 +202,11 @@ class HookQuerySet(models.QuerySet):
             if parent._meta.concrete_model is not model_cls._meta.concrete_model:
                 is_mti = True
                 break
+        
+        print(f"Is MTI: {is_mti}")
+        print(f"Model concrete model: {model_cls._meta.concrete_model.__name__}")
+        for parent in model_cls._meta.all_parents:
+            print(f"  Parent {parent.__name__}: concrete_model = {parent._meta.concrete_model.__name__}")
 
         if not bypass_hooks:
             # Load originals for hook comparison
@@ -220,11 +233,14 @@ class HookQuerySet(models.QuerySet):
                 fields_set = set(fields)
                 fields_set.update(modified_fields)
                 fields = list(fields_set)
+                print(f"Modified fields detected: {modified_fields}")
 
         # Handle MTI models differently
         if is_mti:
+            print("Using MTI bulk update logic")
             result = self._mti_bulk_update(objs, fields, **kwargs)
         else:
+            print("Using standard Django bulk_update")
             # For single-table models, use Django's built-in bulk_update
             django_kwargs = {
                 k: v
@@ -236,6 +252,7 @@ class HookQuerySet(models.QuerySet):
         if not bypass_hooks:
             engine.run(model_cls, AFTER_UPDATE, objs, originals, ctx=ctx)
 
+        print(f"Bulk update result: {result}")
         return result
 
     def _detect_modified_fields(self, new_instances, original_instances):
@@ -280,18 +297,29 @@ class HookQuerySet(models.QuerySet):
         Get the complete inheritance chain from root parent to current model.
         Returns list of model classes in order: [RootParent, Parent, Child]
         """
+        print(f"\n=== GET INHERITANCE CHAIN DEBUG ===")
+        print(f"Current model: {self.model.__name__}")
+        
         chain = []
         current_model = self.model
         while current_model:
+            print(f"Processing model: {current_model.__name__}")
             if not current_model._meta.proxy:
                 chain.append(current_model)
+                print(f"  Added to chain: {current_model.__name__}")
+            else:
+                print(f"  Skipped proxy model: {current_model.__name__}")
+            
             parents = [
                 parent
                 for parent in current_model._meta.parents.keys()
                 if not parent._meta.proxy
             ]
+            print(f"  Parents: {[p.__name__ for p in parents]}")
             current_model = parents[0] if parents else None
+        
         chain.reverse()
+        print(f"Final inheritance chain: {[m.__name__ for m in chain]}")
         return chain
 
     def _mti_bulk_create(self, objs, inheritance_chain=None, **kwargs):
@@ -321,13 +349,13 @@ class HookQuerySet(models.QuerySet):
         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_batch(
+                batch_result = self._process_mti_bulk_create_batch(
                     batch, inheritance_chain, **django_kwargs
                 )
                 created_objects.extend(batch_result)
         return created_objects
 
-    def _process_mti_batch(self, batch, inheritance_chain, **kwargs):
+    def _process_mti_bulk_create_batch(self, batch, inheritance_chain, **kwargs):
         """
         Process a single batch of objects through the inheritance chain.
         Implements Django's suggested workaround #2: O(n) normal inserts into parent
@@ -534,8 +562,27 @@ class HookQuerySet(models.QuerySet):
         Custom bulk update implementation for MTI models.
         Updates each table in the inheritance chain efficiently using Django's batch_size.
         """
+        print(f"\n=== MTI BULK UPDATE DEBUG ===")
+        print(f"Model: {self.model.__name__}")
+        print(f"Number of objects: {len(objs)}")
+        print(f"Fields to update: {fields}")
+        
         model_cls = self.model
         inheritance_chain = self._get_inheritance_chain()
+        print(f"Inheritance chain: {[m.__name__ for m in 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 = {}
@@ -547,12 +594,47 @@ class HookQuerySet(models.QuerySet):
                     if model not in field_groups:
                         field_groups[model] = []
                     field_groups[model].append(field_name)
+                    print(f"Field '{field_name}' belongs to model '{model.__name__}'")
                     break
+        
+        print(f"Field groups: {field_groups}")
 
-        # Update each table in the inheritance chain
+        # Process in batches
+        batch_size = django_kwargs.get("batch_size") or len(objs)
         total_updated = 0
+        
+        print(f"Processing in batches of size: {batch_size}")
+        
+        with transaction.atomic(using=self.db, savepoint=False):
+            for i in range(0, len(objs), batch_size):
+                batch = objs[i : i + batch_size]
+                print(f"\n--- Processing batch {i//batch_size + 1} ({len(batch)} objects) ---")
+                batch_result = self._process_mti_bulk_update_batch(
+                    batch, field_groups, inheritance_chain, **django_kwargs
+                )
+                total_updated += batch_result
+                print(f"Batch {i//batch_size + 1} updated {batch_result} rows")
+
+        print(f"\n=== TOTAL UPDATED: {total_updated} ===")
+        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
+        
+        print(f"Processing batch with {len(batch)} objects")
+        print(f"Field groups: {field_groups}")
+        
+        # Update each table in the inheritance chain
         for model, model_fields in field_groups.items():
+            print(f"\n--- Updating model: {model.__name__} ---")
+            print(f"Fields to update: {model_fields}")
+            
             if not model_fields:
+                print("No fields to update, skipping")
                 continue
 
             # For MTI, we need to handle parent links correctly
@@ -560,8 +642,9 @@ 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'
+                print(f"Root model - using PKs: {pks}")
             else:
                 # Child model - use parent link field
                 parent_link = None
@@ -571,30 +654,52 @@ class HookQuerySet(models.QuerySet):
                         break
                 
                 if parent_link is None:
-                    # This shouldn't happen in proper MTI, but handle gracefully
+                    print(f"No parent link found for {model.__name__}, skipping")
                     continue
                 
+                print(f"Parent link field: {parent_link.name} ({parent_link.attname})")
+                
                 # 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
+                print(f"Child model - using parent link values: {pks}")
             
             if pks:
                 base_qs = model._base_manager.using(self.db)
+                print(f"Filter field: {filter_field}")
+                print(f"PKs to filter by: {pks}")
+                
+                # Check if records exist
+                existing_count = base_qs.filter(**{f"{filter_field}__in": pks}).count()
+                print(f"Existing records with these PKs: {existing_count}")
                 
-                # 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):
+                    print(f"Building CASE statement for field: {field_name}")
+                    for pk, obj in zip(pks, batch):
                         value = getattr(obj, field_name)
+                        print(f"  PK {pk}: {field_name} = {value}")
                         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
-                updated_count = base_qs.filter(**{f"{filter_field}__in": pks}).update(**case_statements)
-                total_updated += updated_count
+                print(f"Case statements built: {list(case_statements.keys())}")
+                
+                # Execute a single bulk update for all objects in this model
+                try:
+                    updated_count = base_qs.filter(**{f"{filter_field}__in": pks}).update(**case_statements)
+                    print(f"UPDATE QUERY EXECUTED - Updated {updated_count} rows")
+                    total_updated += updated_count
+                except Exception as e:
+                    print(f"ERROR during update: {e}")
+                    import traceback
+                    traceback.print_exc()
+            else:
+                print("No PKs found, skipping update")
 
+        print(f"Batch total updated: {total_updated}")
         return total_updated

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: django-bulk-hooks
-Version: 0.1.184
+Version: 0.1.187
 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.187.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=DYt_oK8g7D1oHGIT9gzE--F9qdXKaMEgPJq5x-t9cSw,30664
 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.187.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
+django_bulk_hooks-0.1.187.dist-info/METADATA,sha256=TgXRIiryaRvkIg_IJ5l9xRaGCwkQqsAahFpNff_eymQ,6951
+django_bulk_hooks-0.1.187.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+django_bulk_hooks-0.1.187.dist-info/RECORD,,