django-bulk-hooks 0.2.73__tar.gz → 0.2.74__tar.gz

{django_bulk_hooks-0.2.73 → django_bulk_hooks-0.2.74}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: django-bulk-hooks
-Version: 0.2.73
+Version: 0.2.74
 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.73 → django_bulk_hooks-0.2.74}/django_bulk_hooks/dispatcher.py

@@ -409,17 +409,38 @@ class HookDispatcher:
                             if hasattr(preloaded_obj, rel):
                                 setattr(obj, rel, getattr(preloaded_obj, rel))
 
-        # Handle unsaved new_records by preloading their FK targets
+        # Handle unsaved new_records by preloading their FK targets (bulk query to avoid N+1)
         if changeset.new_records:
+            # Collect FK IDs for each relationship from unsaved records
+            field_ids_map = {rel: set() for rel in relationship_list}
+
             for obj in changeset.new_records:
                 if obj.pk is None:  # Unsaved object
                     for rel in relationship_list:
                         if hasattr(obj, f'{rel}_id'):
                             rel_id = getattr(obj, f'{rel}_id')
                             if rel_id:
-                                # Load the related object
-                                rel_model = getattr(changeset.model_cls._meta.get_field(rel).remote_field, 'model')
-                                rel_obj = rel_model.objects.get(pk=rel_id)
+                                field_ids_map[rel].add(rel_id)
+
+            # Bulk load relationships for unsaved records
+            field_objects_map = {}
+            for rel, ids in field_ids_map.items():
+                if not ids:
+                    continue
+                try:
+                    rel_model = getattr(changeset.model_cls._meta.get_field(rel).remote_field, 'model')
+                    field_objects_map[rel] = rel_model.objects.in_bulk(ids)
+                except Exception:
+                    field_objects_map[rel] = {}
+
+            # Attach relationships to unsaved records
+            for obj in changeset.new_records:
+                if obj.pk is None:  # Unsaved object
+                    for rel in relationship_list:
+                        rel_id = getattr(obj, f'{rel}_id', None)
+                        if rel_id and rel in field_objects_map:
+                            rel_obj = field_objects_map[rel].get(rel_id)
+                            if rel_obj:
                                 setattr(obj, rel, rel_obj)
 
     def _preload_select_related_for_before_create(self, changeset, select_related_fields):

{django_bulk_hooks-0.2.73 → django_bulk_hooks-0.2.74}/pyproject.toml

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