django-bulk-hooks 0.1.169__py3-none-any.whl → 0.1.170__py3-none-any.whl

django_bulk_hooks/queryset.py

@@ -24,8 +24,9 @@ class HookQuerySet(models.QuerySet):
         objs = list(self)
         if not objs:
             return 0
-        # Call the base QuerySet implementation to avoid recursion
-        return super().bulk_delete(objs)
+        # Use our bulk_delete method to handle hooks properly
+        self.bulk_delete(objs)
+        return len(objs)
 
     @transaction.atomic
     def update(self, **kwargs):
@@ -33,33 +34,28 @@ class HookQuerySet(models.QuerySet):
         if not instances:
             return 0
 
-        model_cls = self.model
-        pks = [obj.pk for obj in instances]
-
-        # Load originals for hook comparison and ensure they match the order of instances
-        # Use the base manager to avoid recursion
-        original_map = {
-            obj.pk: obj for obj in model_cls._base_manager.filter(pk__in=pks)
-        }
-        originals = [original_map.get(obj.pk) for obj in instances]
-
         # Apply field updates to instances
         for obj in instances:
             for field, value in kwargs.items():
                 setattr(obj, field, value)
 
-        # Run BEFORE_UPDATE hooks
-        ctx = HookContext(model_cls)
-        engine.run(model_cls, BEFORE_UPDATE, instances, originals, ctx=ctx)
-
-        # Use Django's built-in update logic directly
-        # Call the base QuerySet implementation to avoid recursion
-        update_count = super().update(**kwargs)
+        # Use our bulk_update method to handle hooks properly
+        self.bulk_update(instances, list(kwargs.keys()))
+        return len(instances)
 
-        # Run AFTER_UPDATE hooks
-        engine.run(model_cls, AFTER_UPDATE, instances, originals, ctx=ctx)
-
-        return update_count
+    def save(self, obj):
+        """
+        Save a single object using the appropriate bulk operation.
+        """
+        if obj.pk:
+            # Use bulk_update for existing objects
+            self.bulk_update(
+                [obj], [field.name for field in obj._meta.fields if field.name != "id"]
+            )
+        else:
+            # Use bulk_create for new objects
+            self.bulk_create([obj])
+        return obj
 
     @transaction.atomic
     def bulk_create(
@@ -254,9 +250,23 @@ class HookQuerySet(models.QuerySet):
 
         pks = [obj.pk for obj in objs if obj.pk is not None]
 
-        # Call the base QuerySet implementation to avoid recursion
-        # The hooks have already been fired above, so we don't need them again
-        super().bulk_delete(objs, batch_size=batch_size)
+        # Use Django's base manager to perform the actual deletion
+        # This avoids recursion and uses Django's built-in delete logic
+        from django.db.models import QuerySet
+
+        base_qs = QuerySet(model_cls, using=self.db)
+
+        # Delete in batches if batch_size is specified
+        if batch_size:
+            for i in range(0, len(objs), batch_size):
+                batch = objs[i : i + batch_size]
+                batch_pks = [obj.pk for obj in batch if obj.pk is not None]
+                if batch_pks:
+                    base_qs.filter(pk__in=batch_pks).delete()
+        else:
+            # Delete all at once
+            if pks:
+                base_qs.filter(pk__in=pks).delete()
 
         if not bypass_hooks:
             engine.run(model_cls, AFTER_DELETE, objs, ctx=ctx)

{django_bulk_hooks-0.1.169.dist-info → django_bulk_hooks-0.1.170.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: django-bulk-hooks
-Version: 0.1.169
+Version: 0.1.170
 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.169.dist-info → django_bulk_hooks-0.1.170.dist-info}/RECORD

@@ -9,9 +9,9 @@ django_bulk_hooks/handler.py,sha256=xZt8iNdYF-ACz-MnKMY0co6scWINU5V5wC1lyDn844k,
 django_bulk_hooks/manager.py,sha256=OSzW8eVzknLV1WCvZcBkWMz9x_Vjq4bJM8raVXKiZvI,5085
 django_bulk_hooks/models.py,sha256=5PBjBoGlHwAE5b4yaZ5kDjt5UtHfQp6pDrwB2XPs2tk,3850
 django_bulk_hooks/priority.py,sha256=HG_2D35nga68lBCZmSXTcplXrjFoRgZFRDOy4ROKonY,376
-django_bulk_hooks/queryset.py,sha256=VpoVukIAZrJA2VEXg13DtT8TsqN3tAm4VOgQbR_jkeY,23571
+django_bulk_hooks/queryset.py,sha256=dHfZDjOPi5G4kY8VrC-M3ThjXTrh4Ik_7DN8sev-4I8,23826
 django_bulk_hooks/registry.py,sha256=-mQBizJ06nz_tajZBinViKx_uP2Tbc1tIpTEMv7lwKA,705
-django_bulk_hooks-0.1.169.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
-django_bulk_hooks-0.1.169.dist-info/METADATA,sha256=5VYbjsyR_zAeXdjH3bwrS_8FoTLAXox_6x9AMrRoRFM,6939
-django_bulk_hooks-0.1.169.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-django_bulk_hooks-0.1.169.dist-info/RECORD,,
+django_bulk_hooks-0.1.170.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
+django_bulk_hooks-0.1.170.dist-info/METADATA,sha256=uS80HmpLlXGDA3vbB3NKVGDbk0mnXUrkgpjRdpoXMeM,6939
+django_bulk_hooks-0.1.170.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+django_bulk_hooks-0.1.170.dist-info/RECORD,,