django-bulk-hooks 0.1.54__py3-none-any.whl → 0.1.55__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-0.1.54.dist-info → django_bulk_hooks-0.1.55.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: django-bulk-hooks
-Version: 0.1.54
+Version: 0.1.55
 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.55.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/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.55.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
+django_bulk_hooks-0.1.55.dist-info/METADATA,sha256=jge-9Tq4hx6ISQV49qKbu6kUrl1DrQfHxZ8-7PwKT5g,3101
+django_bulk_hooks-0.1.55.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+django_bulk_hooks-0.1.55.dist-info/RECORD,,