django-bulk-hooks 0.1.54__py3-none-any.whl → 0.1.56__py3-none-any.whl

django_bulk_hooks/handler.py

@@ -7,7 +7,18 @@ from django_bulk_hooks.registry import get_hooks, register_hook
 
 logger = logging.getLogger(__name__)
 
-# Thread-local hook queue context
+# Thread-local hook context and trigger state
+class TriggerVars(threading.local):
+    def __init__(self):
+        self.new = None
+        self.old = None
+        self.event = None
+        self.model = None
+        self.depth = 0
+
+trigger = TriggerVars()
+
+# Hook queue per thread
 _hook_context = threading.local()
 
 def get_hook_queue():
@@ -15,6 +26,37 @@ def get_hook_queue():
         _hook_context.queue = deque()
     return _hook_context.queue
 
+class TriggerContextState:
+    @property
+    def is_before(self):
+        return trigger.event.startswith("before_") if trigger.event else False
+
+    @property
+    def is_after(self):
+        return trigger.event.startswith("after_") if trigger.event else False
+
+    @property
+    def is_create(self):
+        return "create" in trigger.event if trigger.event else False
+
+    @property
+    def is_update(self):
+        return "update" in trigger.event if trigger.event else False
+
+    @property
+    def new(self):
+        return trigger.new
+
+    @property
+    def old(self):
+        return trigger.old
+
+    @property
+    def model(self):
+        return trigger.model
+
+Trigger = TriggerContextState()
+
 class TriggerHandlerMeta(type):
     _registered = set()
 
@@ -67,6 +109,12 @@ class TriggerHandler(metaclass=TriggerHandlerMeta):
         old_records,
         **kwargs,
     ):
+        trigger.depth += 1
+        trigger.new = new_records
+        trigger.old = old_records
+        trigger.event = event
+        trigger.model = model
+
         hooks = sorted(get_hooks(model, event), key=lambda x: x[3])
         logger.debug("Processing %d hooks for %s.%s", len(hooks), model.__name__, event)
 
@@ -96,7 +144,14 @@ class TriggerHandler(metaclass=TriggerHandlerMeta):
                     logger.exception("Error in hook %s.%s", handler_cls.__name__, method_name)
 
         conn = transaction.get_connection()
-        if conn.in_atomic_block and event.startswith("after_"):
-            transaction.on_commit(_execute)
-        else:
-            _execute()
+        try:
+            if conn.in_atomic_block and event.startswith("after_"):
+                transaction.on_commit(_execute)
+            else:
+                _execute()
+        finally:
+            trigger.new = None
+            trigger.old = None
+            trigger.event = None
+            trigger.model = None
+            trigger.depth -= 1

django_bulk_hooks/queryset.py

@@ -15,13 +15,29 @@ class LifecycleQuerySet(models.QuerySet):
         if not instances:
             return 0
 
+        model_cls = self.model
+        pks = [obj.pk for obj in instances]
+
+        # Load originals for hook comparison
+        originals = list(model_cls.objects.filter(pk__in=pks))
+        originals_by_pk = {obj.pk: obj for obj in originals}
+
+        # Apply field updates to instances
         for obj in instances:
             for field, value in kwargs.items():
                 setattr(obj, field, value)
 
-        self.model.objects.bulk_update(
-            instances,
-            fields=list(kwargs.keys()),
-        )
+        # Run BEFORE_UPDATE hooks
+        from django_bulk_hooks.context import TriggerContext
+        from django_bulk_hooks import engine
+        ctx = TriggerContext(model_cls)
+        engine.run(model_cls, "before_update", instances, originals, ctx=ctx)
+
+        # Use Django's built-in update logic directly
+        update_count = super().update(**kwargs)
+
+        # Run AFTER_UPDATE hooks
+        engine.run(model_cls, "after_update", instances, originals, ctx=ctx)
+
+        return update_count
 
-        return len(instances)

{django_bulk_hooks-0.1.54.dist-info → django_bulk_hooks-0.1.56.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: django-bulk-hooks
-Version: 0.1.54
+Version: 0.1.56
 Summary: Lifecycle-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.54.dist-info → django_bulk_hooks-0.1.56.dist-info}/RECORD

@@ -5,13 +5,13 @@ django_bulk_hooks/context.py,sha256=VyjgaubjJTTXQHgyoGY4OLBbdCpBwo-RX4bvxRfZ1c4,
 django_bulk_hooks/decorators.py,sha256=wbT9lbpKZqDbG5vPvtTC6MHYWvjLaVccCwFM9U2nHkA,3159
 django_bulk_hooks/engine.py,sha256=l7BzU3lfYhZHZazks8FgHhZB4iLVKuzmcnVcyEQkGpQ,1978
 django_bulk_hooks/enums.py,sha256=Zo8_tJzuzZ2IKfVc7gZ-0tWPT8q1QhqZbAyoh9ZVJbs,381
-django_bulk_hooks/handler.py,sha256=P1HuX9pQpLmMMmrKfk30HHpfz-TaKzB-MMIOGlIYIR8,3608
+django_bulk_hooks/handler.py,sha256=qUulFSPhi__gWHISC4GogeVQ9aDo45bz0Dj421Y6skE,4968
 django_bulk_hooks/manager.py,sha256=jsTgbjmRe5SeakVJxV8wYxqH6f3fRX5zm3q0Wy-M9Nc,4271
 django_bulk_hooks/models.py,sha256=_cqIVHKhXb1EOnXxhsuJUlLnPjTRM_sCP5OKbq4q-I8,542
 django_bulk_hooks/priority.py,sha256=HG_2D35nga68lBCZmSXTcplXrjFoRgZFRDOy4ROKonY,376
-django_bulk_hooks/queryset.py,sha256=wrK--iZBsBNuf25HpaaZE9__EN9y6KlVmfXqioS8SmY,681
+django_bulk_hooks/queryset.py,sha256=ak3fDz9pbqsafL215ysBGZYLMJj13j-GTF1Rj32YuEQ,1344
 django_bulk_hooks/registry.py,sha256=yeTi0IhodL61J86ohb5OyITufE28T3ecMbt6RWvkzTs,585
-django_bulk_hooks-0.1.54.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
-django_bulk_hooks-0.1.54.dist-info/METADATA,sha256=Rg8LRCTFZmtVZFl1q0xfOV4Yq8UqFUKXTPrGO5OS8ak,3101
-django_bulk_hooks-0.1.54.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-django_bulk_hooks-0.1.54.dist-info/RECORD,,
+django_bulk_hooks-0.1.56.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
+django_bulk_hooks-0.1.56.dist-info/METADATA,sha256=gzKV5WfzjebNCVKeKZItmzwqEFBzkhwZdTn7bYWdmvA,3101
+django_bulk_hooks-0.1.56.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+django_bulk_hooks-0.1.56.dist-info/RECORD,,