django-bulk-hooks 0.2.59__tar.gz → 0.2.60__tar.gz

{django_bulk_hooks-0.2.59 → django_bulk_hooks-0.2.60}/PKG-INFO

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

@@ -182,10 +182,35 @@ class BulkExecutor:
         if not objs:
             return 0
 
+        # Ensure auto_now fields are included and pre-saved for all models
+        # This handles both MTI and non-MTI models uniformly (SOC & DRY)
+        fields = list(fields)  # Make a copy so we can modify it
+        
+        # Get models to check - for MTI, check entire inheritance chain
+        if self.mti_handler.is_mti_model():
+            models_to_check = self.mti_handler.get_inheritance_chain()
+        else:
+            models_to_check = [self.model_cls]
+        
+        # Collect all auto_now fields and pre-save them
+        auto_now_fields = set()
+        for model in models_to_check:
+            for field in model._meta.local_fields:
+                if getattr(field, "auto_now", False) and not getattr(field, "auto_now_add", False):
+                    auto_now_fields.add(field.name)
+                    # Pre-save the field to set the value on instances
+                    for obj in objs:
+                        field.pre_save(obj, add=False)
+        
+        # Add auto_now fields to the update list if not already present
+        for auto_now_field in auto_now_fields:
+            if auto_now_field not in fields:
+                fields.append(auto_now_field)
+
         # Check if this is an MTI model and route accordingly
         if self.mti_handler.is_mti_model():
             logger.info(f"Detected MTI model {self.model_cls.__name__}, using MTI bulk update")
-            # Build execution plan
+            # Build execution plan (fields already have auto_now included)
             plan = self.mti_handler.build_update_plan(objs, fields, batch_size=batch_size)
             # Execute the plan
             return self._execute_mti_update_plan(plan)

{django_bulk_hooks-0.2.59 → django_bulk_hooks-0.2.60}/django_bulk_hooks/operations/mti_handler.py

@@ -537,7 +537,7 @@ class MTIHandler:
 
         Args:
             objs: List of model instances to update
-            fields: List of field names to update
+            fields: List of field names to update (auto_now fields already included by executor)
             batch_size: Number of objects per batch
 
         Returns:
@@ -555,28 +555,15 @@ class MTIHandler:
 
         batch_size = batch_size or len(objs)
 
-        # Handle auto_now fields
-        for obj in objs:
-            for model in inheritance_chain:
-                for field in model._meta.local_fields:
-                    if getattr(field, "auto_now", False):
-                        field.pre_save(obj, add=False)
-
-        # Add auto_now fields to update list
-        auto_now_fields = set()
-        for model in inheritance_chain:
-            for field in model._meta.local_fields:
-                if getattr(field, "auto_now", False):
-                    auto_now_fields.add(field.name)
-
-        all_fields = list(fields) + list(auto_now_fields)
+        # Note: auto_now fields are already handled by executor.bulk_update()
+        # which calls pre_save() and includes them in the fields list
 
         # Group fields by model
         field_groups = []
         for model_idx, model in enumerate(inheritance_chain):
             model_fields = []
 
-            for field_name in all_fields:
+            for field_name in fields:
                 try:
                     field = self.model_cls._meta.get_field(field_name)
                     if field in model._meta.local_fields:

{django_bulk_hooks-0.2.59 → django_bulk_hooks-0.2.60}/pyproject.toml

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