django-bulk-hooks 0.1.123__py3-none-any.whl → 0.1.124__py3-none-any.whl

django_bulk_hooks/queryset.py

@@ -1,10 +1,6 @@
-from django.db import models, transaction, connections
-from django.db.models import AutoField, Q, Max
-from django.db import NotSupportedError
-from django.db.models.constants import OnConflict
-from django.db.models.expressions import DatabaseDefault
-import operator
-from functools import reduce
+
+from django.db import models, transaction
+from django.db.models import AutoField
 
 from django_bulk_hooks import engine
 from django_bulk_hooks.constants import (
@@ -303,8 +299,11 @@ class HookQuerySet(models.QuerySet):
         """
         Process a single batch of objects through the inheritance chain.
         """
-        # Step 1: Handle parent tables with individual saves (needed for PKs)
+        # Step 1: Handle parent tables with bulk operations where possible
         parent_objects_map = {}
+        
+        # Group parent objects by model class to enable bulk operations
+        parent_objects_by_model = {}
         for obj in batch:
             parent_instances = {}
             current_parent = None
@@ -312,10 +311,24 @@ class HookQuerySet(models.QuerySet):
                 parent_obj = self._create_parent_instance(
                     obj, model_class, current_parent
                 )
-                parent_obj.save()
                 parent_instances[model_class] = parent_obj
                 current_parent = parent_obj
+                
+                # Group by model class for potential bulk operations
+                if model_class not in parent_objects_by_model:
+                    parent_objects_by_model[model_class] = []
+                parent_objects_by_model[model_class].append(parent_obj)
+            
             parent_objects_map[id(obj)] = parent_instances
+        
+        # Save parent objects in bulk where possible
+        for model_class, parent_objs in parent_objects_by_model.items():
+            if len(parent_objs) > 1:
+                # Use bulk_create for multiple objects of the same model
+                model_class._base_manager.bulk_create(parent_objs)
+            else:
+                # Individual save for single objects
+                super(parent_objs[0].__class__, parent_objs[0]).save()
         # Step 2: Bulk insert for child objects
         child_model = inheritance_chain[-1]
         child_objects = []
@@ -324,22 +337,8 @@ class HookQuerySet(models.QuerySet):
                 obj, child_model, parent_objects_map.get(id(obj), {})
             )
             child_objects.append(child_obj)
-        # If the child model is still MTI, call our own logic recursively
+        # If the child model is still MTI, we need to handle it specially
         if len([p for p in child_model._meta.parents.keys() if not p._meta.proxy]) > 0:
-            # Build inheritance chain for the child model
-            child_inheritance_chain = []
-            current_model = child_model
-            while current_model:
-                if not current_model._meta.proxy:
-                    child_inheritance_chain.append(current_model)
-                parents = [
-                    parent
-                    for parent in current_model._meta.parents.keys()
-                    if not parent._meta.proxy
-                ]
-                current_model = parents[0] if parents else None
-            child_inheritance_chain.reverse()
-            
             # For nested MTI, we can't use bulk operations recursively
             # because it would create infinite recursion. Instead, we save each child individually.
             # We use super().save() to avoid triggering hooks that would cause recursion.
@@ -388,6 +387,12 @@ class HookQuerySet(models.QuerySet):
             elif hasattr(field, 'auto_now') and field.auto_now:
                 field.pre_save(parent_obj, add=True)
         
+        # Ensure auto_now_add fields are explicitly set to prevent null constraint violations
+        for field in parent_model._meta.local_fields:
+            if hasattr(field, 'auto_now_add') and field.auto_now_add:
+                if getattr(parent_obj, field.name) is None:
+                    field.pre_save(parent_obj, add=True)
+        
         return parent_obj
 
     def _create_child_instance(self, source_obj, child_model, parent_instances):
@@ -411,4 +416,10 @@ class HookQuerySet(models.QuerySet):
             elif hasattr(field, 'auto_now') and field.auto_now:
                 field.pre_save(child_obj, add=True)
         
+        # Ensure auto_now_add fields are explicitly set to prevent null constraint violations
+        for field in child_model._meta.local_fields:
+            if hasattr(field, 'auto_now_add') and field.auto_now_add:
+                if getattr(child_obj, field.name) is None:
+                    field.pre_save(child_obj, add=True)
+        
         return child_obj

{django_bulk_hooks-0.1.123.dist-info → django_bulk_hooks-0.1.124.dist-info}/METADATA

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

@@ -9,9 +9,9 @@ django_bulk_hooks/handler.py,sha256=xZt8iNdYF-ACz-MnKMY0co6scWINU5V5wC1lyDn844k,
 django_bulk_hooks/manager.py,sha256=r54ct3S6AcqME2OsX-jPF944CEKcoSIW3qiAx_NwUaw,2801
 django_bulk_hooks/models.py,sha256=7RG7GrOdHXFjGVPV4FPRZVNMIHHW-hMCi6hn9LH_hVI,3331
 django_bulk_hooks/priority.py,sha256=HG_2D35nga68lBCZmSXTcplXrjFoRgZFRDOy4ROKonY,376
-django_bulk_hooks/queryset.py,sha256=19uG70BxNAJwLqu3j3ZmT4DStX8sPRwbV9079m0yn84,16717
+django_bulk_hooks/queryset.py,sha256=RpPaATYlJR6mCoxwCKUcVPH0Wr0Rwl51yF9FDjTT4TQ,17433
 django_bulk_hooks/registry.py,sha256=-mQBizJ06nz_tajZBinViKx_uP2Tbc1tIpTEMv7lwKA,705
-django_bulk_hooks-0.1.123.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
-django_bulk_hooks-0.1.123.dist-info/METADATA,sha256=1irToq1BSI9i4dP80sqHhRDq3BhwGVcEK0B-Xf69xXo,6951
-django_bulk_hooks-0.1.123.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-django_bulk_hooks-0.1.123.dist-info/RECORD,,
+django_bulk_hooks-0.1.124.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
+django_bulk_hooks-0.1.124.dist-info/METADATA,sha256=Ul83KdLsZ_wvBSeVQJkG2Mf_HHg5Ckjrck5dVIwZ0FU,6951
+django_bulk_hooks-0.1.124.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+django_bulk_hooks-0.1.124.dist-info/RECORD,,