django-bulk-hooks 0.1.115__py3-none-any.whl → 0.1.116__py3-none-any.whl

django_bulk_hooks/manager.py

@@ -1,3 +1,5 @@
+
+
 from django.db import models, transaction
 from django.db.models import AutoField
 
@@ -20,9 +22,12 @@ from django_bulk_hooks.queryset import HookQuerySet
 class BulkHookManager(models.Manager):
     CHUNK_SIZE = 200
 
+
     def get_queryset(self):
         return HookQuerySet(self.model, using=self._db)
 
+
+
     @transaction.atomic
     def bulk_update(
         self, objs, fields, bypass_hooks=False, bypass_validation=False, **kwargs
@@ -72,6 +77,90 @@ class BulkHookManager(models.Manager):
 
         return objs
 
+
+
+    @transaction.atomic
+    def bulk_create(
+        self,
+        objs,
+        batch_size=None,
+        ignore_conflicts=False,
+        update_conflicts=False,
+        update_fields=None,
+        unique_fields=None,
+        bypass_hooks=False,
+        bypass_validation=False,
+    ):
+        """
+        Insert each of the instances into the database. Behaves like Django's bulk_create,
+        but supports multi-table inheritance (MTI) models. All arguments are supported and
+        passed through to the correct logic. For MTI, only a subset of options may be supported.
+        """
+        model_cls = self.model
+
+        if batch_size is not None and batch_size <= 0:
+            raise ValueError("Batch size must be a positive integer.")
+
+        # Check that the parents share the same concrete model with our model to detect inheritance pattern
+        for parent in model_cls._meta.all_parents:
+            if parent._meta.concrete_model is not model_cls._meta.concrete_model:
+                # We allow this for MTI, but not for single-table
+                break
+
+        if not objs:
+            return objs
+
+        if any(not isinstance(obj, model_cls) for obj in objs):
+            raise TypeError(
+                f"bulk_create expected instances of {model_cls.__name__}, but got {set(type(obj).__name__ for obj in objs)}"
+            )
+
+        # Set auto_now_add/auto_now fields before DB ops
+        self._set_auto_now_fields(objs, model_cls)
+
+        # Fire hooks before DB ops
+        if not bypass_hooks:
+            ctx = HookContext(model_cls)
+            if not bypass_validation:
+                engine.run(model_cls, VALIDATE_CREATE, objs, ctx=ctx)
+            engine.run(model_cls, BEFORE_CREATE, objs, ctx=ctx)
+
+        # MTI detection: if inheritance chain > 1, use MTI logic
+        inheritance_chain = self._get_inheritance_chain()
+        if len(inheritance_chain) <= 1:
+            # Single-table: use Django's standard bulk_create
+            # Pass through all supported arguments
+            result = super(models.Manager, self).bulk_create(
+                objs,
+                batch_size=batch_size,
+                ignore_conflicts=ignore_conflicts,
+                update_conflicts=update_conflicts,
+                update_fields=update_fields,
+                unique_fields=unique_fields,
+            )
+        else:
+            # Multi-table: use workaround (parent saves, child bulk)
+            # Only batch_size is supported for MTI; others will raise NotImplementedError
+            if ignore_conflicts or update_conflicts or update_fields or unique_fields:
+                raise NotImplementedError(
+                    "bulk_create with ignore_conflicts, update_conflicts, update_fields, or unique_fields is not supported for multi-table inheritance models."
+                )
+            result = self._mti_bulk_create(
+                objs, inheritance_chain, batch_size=batch_size
+            )
+
+        if not bypass_hooks:
+            engine.run(model_cls, AFTER_CREATE, result, ctx=ctx)
+
+        return result
+
+
+
+
+
+
+    # --- Private helper methods (moved to bottom for clarity) ---
+
     def _detect_modified_fields(self, new_instances, original_instances):
         """
         Detect fields that were modified during BEFORE_UPDATE hooks by comparing
@@ -109,47 +198,6 @@ class BulkHookManager(models.Manager):
 
         return modified_fields
 
-    @transaction.atomic
-    def bulk_create(self, objs, bypass_hooks=False, bypass_validation=False, **kwargs):
-        """
-        Enhanced bulk_create that handles multi-table inheritance (MTI) and single-table models.
-        Falls back to Django's standard bulk_create for single-table models.
-        Fires hooks as usual.
-        """
-        model_cls = self.model
-
-        if not objs:
-            return []
-
-        if any(not isinstance(obj, model_cls) for obj in objs):
-            raise TypeError(
-                f"bulk_create expected instances of {model_cls.__name__}, but got {set(type(obj).__name__ for obj in objs)}"
-            )
-
-        # Fire hooks before DB ops
-        if not bypass_hooks:
-            ctx = HookContext(model_cls)
-            if not bypass_validation:
-                engine.run(model_cls, VALIDATE_CREATE, objs, ctx=ctx)
-            engine.run(model_cls, BEFORE_CREATE, objs, ctx=ctx)
-
-        # MTI detection: if inheritance chain > 1, use MTI logic
-        inheritance_chain = self._get_inheritance_chain()
-        if len(inheritance_chain) <= 1:
-            # Single-table: use Django's standard bulk_create
-            result = []
-            for i in range(0, len(objs), self.CHUNK_SIZE):
-                chunk = objs[i : i + self.CHUNK_SIZE]
-                result.extend(super(models.Manager, self).bulk_create(chunk, **kwargs))
-        else:
-            # Multi-table: use workaround (parent saves, child bulk)
-            result = self._mti_bulk_create(objs, inheritance_chain, **kwargs)
-
-        if not bypass_hooks:
-            engine.run(model_cls, AFTER_CREATE, result, ctx=ctx)
-
-        return result
-
     def _get_inheritance_chain(self):
         """
         Get the complete inheritance chain from root parent to current model.
@@ -172,12 +220,16 @@ class BulkHookManager(models.Manager):
     def _mti_bulk_create(self, objs, inheritance_chain, **kwargs):
         """
         Implements workaround: individual saves for parents, bulk create for child.
+        Sets auto_now_add/auto_now fields for each model in the chain.
         """
         batch_size = kwargs.get("batch_size") or len(objs)
         created_objects = []
         with transaction.atomic(using=self.db, savepoint=False):
             for i in range(0, len(objs), batch_size):
                 batch = objs[i : i + batch_size]
+                # Set auto_now fields for each model in the chain
+                for model in inheritance_chain:
+                    self._set_auto_now_fields(batch, model)
                 batch_result = self._process_mti_batch(
                     batch, inheritance_chain, **kwargs
                 )
@@ -272,6 +324,19 @@ class BulkHookManager(models.Manager):
                 setattr(child_obj, parent_link.name, parent_instance)
         return child_obj
 
+    def _set_auto_now_fields(self, objs, model):
+        """
+        Set auto_now_add and auto_now fields on objects before bulk_create.
+        """
+        from django.utils import timezone
+        now = timezone.now()
+        for obj in objs:
+            for field in model._meta.local_fields:
+                if getattr(field, 'auto_now_add', False) and getattr(obj, field.name, None) is None:
+                    setattr(obj, field.name, now)
+                if getattr(field, 'auto_now', False):
+                    setattr(obj, field.name, now)
+
     @transaction.atomic
     def bulk_delete(
         self, objs, batch_size=None, bypass_hooks=False, bypass_validation=False

{django_bulk_hooks-0.1.115.dist-info → django_bulk_hooks-0.1.116.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: django-bulk-hooks
-Version: 0.1.115
+Version: 0.1.116
 Summary: Hook-style hooks for Django bulk operations like bulk_create and bulk_update.
 Home-page: https://github.com/AugendLimited/django-bulk-hooks
 License: MIT

{django_bulk_hooks-0.1.115.dist-info → django_bulk_hooks-0.1.116.dist-info}/RECORD

@@ -6,12 +6,12 @@ django_bulk_hooks/decorators.py,sha256=tckDcxtOzKCbgvS9QydgeIAWTFDEl-ch3_Q--ruEG
 django_bulk_hooks/engine.py,sha256=3HbgV12JRYIy9IlygHPxZiHnFXj7EwzLyTuJNQeVIoI,1402
 django_bulk_hooks/enums.py,sha256=Zo8_tJzuzZ2IKfVc7gZ-0tWPT8q1QhqZbAyoh9ZVJbs,381
 django_bulk_hooks/handler.py,sha256=xZt8iNdYF-ACz-MnKMY0co6scWINU5V5wC1lyDn844k,4854
-django_bulk_hooks/manager.py,sha256=sQ9xDCo3MRp2CNwsfo_a_43LzlWlSdqqoz04oUp6sx8,13351
+django_bulk_hooks/manager.py,sha256=RZbfRf1H1KHsWvvvtQhIN9yDLVN1KbLaCkU9r7WWJg8,15706
 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=iet4z-9SKhnresA4FBQbxx9rdYnoaOWbw9LUlGftlP0,1466
 django_bulk_hooks/registry.py,sha256=-mQBizJ06nz_tajZBinViKx_uP2Tbc1tIpTEMv7lwKA,705
-django_bulk_hooks-0.1.115.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
-django_bulk_hooks-0.1.115.dist-info/METADATA,sha256=uA5Qw2PzVWnyDRiGHqr6pF3AdeoQ4xhMUh_eJSoOSRs,6939
-django_bulk_hooks-0.1.115.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-django_bulk_hooks-0.1.115.dist-info/RECORD,,
+django_bulk_hooks-0.1.116.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
+django_bulk_hooks-0.1.116.dist-info/METADATA,sha256=4KODRNtf6mTPWNsVLI8IH1yjGFoaGnPL0LsaCgCaTFs,6939
+django_bulk_hooks-0.1.116.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+django_bulk_hooks-0.1.116.dist-info/RECORD,,