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

django_bulk_hooks/queryset.py

@@ -371,20 +371,54 @@ class HookQuerySet(models.QuerySet):
             all_child_objects.append(child_obj)
         print(f"DEBUG: Created {len(all_child_objects)} child objects")
 
-        # Step 2.5: Single bulk insert into childmost table
+        # Step 2.5: Use Django's internal bulk_create infrastructure
         if all_child_objects:
-            print(f"DEBUG: Attempting to bulk_create {len(all_child_objects)} child objects for {child_model.__name__}")
-            try:
-                # Use Django's internal bulk_create to bypass MTI exception
-                child_model._base_manager.bulk_create(all_child_objects)
-                print(f"DEBUG: Successfully bulk_created child objects")
-            except ValueError as e:
-                print(f"DEBUG: bulk_create failed with error: {e}")
-                print(f"DEBUG: Falling back to individual creates for child objects")
-                # Fall back to individual creates if bulk_create fails
-                for child_obj in all_child_objects:
-                    child_obj.save(using=self.db)
-                print(f"DEBUG: Successfully created child objects individually")
+            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 _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]
+            
+            with transaction.atomic(using=self.db, savepoint=False):
+                 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")
 
         # Step 3: Update original objects with generated PKs and state
         pk_field_name = child_model._meta.pk.name

{django_bulk_hooks-0.1.132.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.132
+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.132.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=0CfSUANNj6bJ2ZyYK8CVSYCk18QqQ-ir7rbYAsPmxog,20147
+django_bulk_hooks/queryset.py,sha256=t402vyqe55YR8t_sAlYrYvaZ3Eokm8T3byWPvj4ujB8,22073
 django_bulk_hooks/registry.py,sha256=-mQBizJ06nz_tajZBinViKx_uP2Tbc1tIpTEMv7lwKA,705
-django_bulk_hooks-0.1.132.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
-django_bulk_hooks-0.1.132.dist-info/METADATA,sha256=Ll3JnV4XaiqF7_yELa2NPB60Mdz-im4H96JHyWKF450,6951
-django_bulk_hooks-0.1.132.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-django_bulk_hooks-0.1.132.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,,