django-bulk-hooks 0.1.133__py3-none-any.whl → 0.1.134__py3-none-any.whl

django_bulk_hooks/queryset.py

@@ -371,48 +371,52 @@ class HookQuerySet(models.QuerySet):
             all_child_objects.append(child_obj)
         print(f"DEBUG: Created {len(all_child_objects)} child objects")
 
-        # Step 2.5: Use Django's internal _batched_insert method directly
+        # Step 2.5: Use Django's internal bulk_create infrastructure
         if all_child_objects:
-            print(f"DEBUG: Using Django's internal _batched_insert for {len(all_child_objects)} child objects")
+            print(f"DEBUG: Using Django's internal bulk_create infrastructure for {len(all_child_objects)} child objects")
             
             # Get the base manager's queryset
             base_qs = child_model._base_manager.using(self.db)
             
-            # Use Django's internal _batched_insert method directly
-            # This bypasses the MTI check but uses Django's optimized bulk insert
+            # Use Django's internal _prepare_for_bulk_create method
+            # This is the same method that Django's bulk_create uses internally
+            objs_with_pk, objs_without_pk = base_qs._prepare_for_bulk_create(all_child_objects)
+            
+            print(f"DEBUG: Prepared {len(objs_with_pk)} objects with PK, {len(objs_without_pk)} objects without PK")
+            
+            # Use Django's internal _batched_insert method
             opts = child_model._meta
             fields = [f for f in opts.concrete_fields if not f.generated]
             
-            # Prepare objects for bulk insert (same logic as Django's bulk_create)
-            objs_with_pk, objs_without_pk = base_qs._prepare_for_bulk_create(all_child_objects)
-            
             with transaction.atomic(using=self.db, savepoint=False):
-                if objs_with_pk:
-                    returned_columns = base_qs._batched_insert(
-                        objs_with_pk,
-                        fields,
-                        batch_size=None,
-                    )
-                    for obj_with_pk, results in zip(objs_with_pk, returned_columns):
-                        for result, field in zip(results, opts.db_returning_fields):
-                            if field != opts.pk:
-                                setattr(obj_with_pk, field.attname, result)
-                    for obj_with_pk in objs_with_pk:
-                        obj_with_pk._state.adding = False
-                        obj_with_pk._state.db = self.db
-                
-                if objs_without_pk:
-                    fields = [f for f in fields if not isinstance(f, AutoField)]
-                    returned_columns = base_qs._batched_insert(
-                        objs_without_pk,
-                        fields,
-                        batch_size=None,
-                    )
-                    for obj_without_pk, results in zip(objs_without_pk, returned_columns):
-                        for result, field in zip(results, opts.db_returning_fields):
-                            setattr(obj_without_pk, field.attname, result)
-                        obj_without_pk._state.adding = False
-                        obj_without_pk._state.db = self.db
+                 if objs_with_pk:
+                     print(f"DEBUG: Inserting {len(objs_with_pk)} objects with PK")
+                     returned_columns = base_qs._batched_insert(
+                         objs_with_pk,
+                         fields,
+                         batch_size=len(objs_with_pk),  # Use actual batch size
+                     )
+                     for obj_with_pk, results in zip(objs_with_pk, returned_columns):
+                         for result, field in zip(results, opts.db_returning_fields):
+                             if field != opts.pk:
+                                 setattr(obj_with_pk, field.attname, result)
+                     for obj_with_pk in objs_with_pk:
+                         obj_with_pk._state.adding = False
+                         obj_with_pk._state.db = self.db
+                 
+                 if objs_without_pk:
+                     print(f"DEBUG: Inserting {len(objs_without_pk)} objects without PK")
+                     fields = [f for f in fields if not isinstance(f, AutoField)]
+                     returned_columns = base_qs._batched_insert(
+                         objs_without_pk,
+                         fields,
+                         batch_size=len(objs_without_pk),  # Use actual batch size
+                     )
+                     for obj_without_pk, results in zip(objs_without_pk, returned_columns):
+                         for result, field in zip(results, opts.db_returning_fields):
+                             setattr(obj_without_pk, field.attname, result)
+                         obj_without_pk._state.adding = False
+                         obj_without_pk._state.db = self.db
             
             print(f"DEBUG: Successfully bulk created child objects using Django's internal methods")
 

{django_bulk_hooks-0.1.133.dist-info → django_bulk_hooks-0.1.134.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: django-bulk-hooks
-Version: 0.1.133
+Version: 0.1.134
 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.133.dist-info → django_bulk_hooks-0.1.134.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=0ocXCPiFDH-16RSj6glNHfuebixDuvVF36svqtjIgLA,21680
+django_bulk_hooks/queryset.py,sha256=t402vyqe55YR8t_sAlYrYvaZ3Eokm8T3byWPvj4ujB8,22073
 django_bulk_hooks/registry.py,sha256=-mQBizJ06nz_tajZBinViKx_uP2Tbc1tIpTEMv7lwKA,705
-django_bulk_hooks-0.1.133.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
-django_bulk_hooks-0.1.133.dist-info/METADATA,sha256=COVQFxmpj6XJ1v56jd3jWDeOptUdWKidTrwM1Vx9Zjs,6951
-django_bulk_hooks-0.1.133.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-django_bulk_hooks-0.1.133.dist-info/RECORD,,
+django_bulk_hooks-0.1.134.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
+django_bulk_hooks-0.1.134.dist-info/METADATA,sha256=hj79PjeJAopmRDrqSe1NXxbshsPLyQruNmC6Vnnvwf4,6951
+django_bulk_hooks-0.1.134.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+django_bulk_hooks-0.1.134.dist-info/RECORD,,