django-bulk-hooks 0.1.84__py3-none-any.whl → 0.1.86__py3-none-any.whl

django_bulk_hooks/engine.py

@@ -8,7 +8,12 @@ from django_bulk_hooks.conditions import safe_get_related_object, safe_get_relat
 logger = logging.getLogger(__name__)
 
 
+# Cache for hook handlers to avoid creating them repeatedly
+_handler_cache = {}
+
 def run(model_cls, event, new_instances, original_instances=None, ctx=None):
+    # Get hooks from cache or fetch them
+    cache_key = (model_cls, event)
     hooks = get_hooks(model_cls, event)
 
     if not hooks:
@@ -32,19 +37,32 @@ def run(model_cls, event, new_instances, original_instances=None, ctx=None):
                     logger.error("Unexpected error during validation for %s: %s", instance, e)
                     raise
 
+    # Pre-create None list for originals if needed
+    if original_instances is None:
+        original_instances = [None] * len(new_instances)
+
+    # Process all hooks
     for handler_cls, method_name, condition, priority in hooks:
-        handler_instance = handler_cls()
-        func = getattr(handler_instance, method_name)
+        # Get or create handler instance from cache
+        handler_key = (handler_cls, method_name)
+        if handler_key not in _handler_cache:
+            handler_instance = handler_cls()
+            func = getattr(handler_instance, method_name)
+            _handler_cache[handler_key] = (handler_instance, func)
+        else:
+            handler_instance, func = _handler_cache[handler_key]
+
+        # If no condition, process all instances at once
+        if not condition:
+            func(new_records=new_instances, old_records=original_instances if any(original_instances) else None)
+            continue
 
+        # For conditional hooks, filter instances first
         to_process_new = []
         to_process_old = []
 
-        for new, original in zip(
-            new_instances,
-            original_instances or [None] * len(new_instances),
-            strict=True,
-        ):
-            if not condition or condition.check(new, original):
+        for new, original in zip(new_instances, original_instances, strict=True):
+            if condition.check(new, original):
                 to_process_new.append(new)
                 to_process_old.append(original)
 

django_bulk_hooks/models.py

@@ -14,6 +14,35 @@ from django_bulk_hooks.constants import (
 from django_bulk_hooks.context import HookContext
 from django_bulk_hooks.engine import run
 from django_bulk_hooks.manager import BulkHookManager
+from django.db.models.fields.related_descriptors import ForwardManyToOneDescriptor
+from functools import wraps
+import contextlib
+
+
+@contextlib.contextmanager
+def patch_foreign_key_behavior():
+    """
+    Temporarily patches Django's foreign key descriptor to return None instead of raising
+    RelatedObjectDoesNotExist when accessing an unset foreign key field.
+    """
+    original_get = ForwardManyToOneDescriptor.__get__
+    
+    @wraps(original_get)
+    def safe_get(self, instance, cls=None):
+        if instance is None:
+            return self
+        try:
+            return original_get(self, instance, cls)
+        except self.RelatedObjectDoesNotExist:
+            return None
+    
+    # Patch the descriptor
+    ForwardManyToOneDescriptor.__get__ = safe_get
+    try:
+        yield
+    finally:
+        # Restore original behavior
+        ForwardManyToOneDescriptor.__get__ = original_get
 
 
 class HookModelMixin(models.Model):
@@ -43,64 +72,55 @@ class HookModelMixin(models.Model):
         if is_create:
             # For create operations, run VALIDATE_CREATE hooks for validation
             ctx = HookContext(self.__class__)
-            run(self.__class__, VALIDATE_CREATE, [self], ctx=ctx)
+            with patch_foreign_key_behavior():
+                run(self.__class__, VALIDATE_CREATE, [self], ctx=ctx)
         else:
             # For update operations, run VALIDATE_UPDATE hooks for validation
             try:
                 old_instance = self.__class__.objects.get(pk=self.pk)
                 ctx = HookContext(self.__class__)
-                run(self.__class__, VALIDATE_UPDATE, [self], [old_instance], ctx=ctx)
+                with patch_foreign_key_behavior():
+                    run(self.__class__, VALIDATE_UPDATE, [self], [old_instance], ctx=ctx)
             except self.__class__.DoesNotExist:
                 # If the old instance doesn't exist, treat as create
                 ctx = HookContext(self.__class__)
-                run(self.__class__, VALIDATE_CREATE, [self], ctx=ctx)
+                with patch_foreign_key_behavior():
+                    run(self.__class__, VALIDATE_CREATE, [self], ctx=ctx)
 
     def save(self, *args, **kwargs):
         is_create = self.pk is None
         ctx = HookContext(self.__class__)
 
-        if is_create:
-            # For create operations, run BEFORE hooks first
-            run(self.__class__, BEFORE_CREATE, [self], ctx=ctx)
-            
-            # Then let Django save
-            super().save(*args, **kwargs)
-            
-            # Then run AFTER hooks
-            run(self.__class__, AFTER_CREATE, [self], ctx=ctx)
-        else:
-            # For update operations, we need to get the old record
-            try:
-                old_instance = self.__class__.objects.get(pk=self.pk)
-                
-                # Run BEFORE hooks first
-                run(self.__class__, BEFORE_UPDATE, [self], [old_instance], ctx=ctx)
-                
-                # Then let Django save
-                super().save(*args, **kwargs)
-                
-                # Then run AFTER hooks
-                run(self.__class__, AFTER_UPDATE, [self], [old_instance], ctx=ctx)
-            except self.__class__.DoesNotExist:
-                # If the old instance doesn't exist, treat as create
+        # Use a single context manager for all hooks
+        with patch_foreign_key_behavior():
+            if is_create:
+                # For create operations
                 run(self.__class__, BEFORE_CREATE, [self], ctx=ctx)
-                
                 super().save(*args, **kwargs)
-                
                 run(self.__class__, AFTER_CREATE, [self], ctx=ctx)
+            else:
+                # For update operations
+                try:
+                    old_instance = self.__class__.objects.get(pk=self.pk)
+                    run(self.__class__, BEFORE_UPDATE, [self], [old_instance], ctx=ctx)
+                    super().save(*args, **kwargs)
+                    run(self.__class__, AFTER_UPDATE, [self], [old_instance], ctx=ctx)
+                except self.__class__.DoesNotExist:
+                    # If the old instance doesn't exist, treat as create
+                    run(self.__class__, BEFORE_CREATE, [self], ctx=ctx)
+                    super().save(*args, **kwargs)
+                    run(self.__class__, AFTER_CREATE, [self], ctx=ctx)
 
         return self
 
     def delete(self, *args, **kwargs):
         ctx = HookContext(self.__class__)
 
-        # Run validation hooks first
-        run(self.__class__, VALIDATE_DELETE, [self], ctx=ctx)
-
-        # Then run business logic hooks
-        run(self.__class__, BEFORE_DELETE, [self], ctx=ctx)
-
-        result = super().delete(*args, **kwargs)
-
-        run(self.__class__, AFTER_DELETE, [self], ctx=ctx)
+        # Use a single context manager for all hooks
+        with patch_foreign_key_behavior():
+            run(self.__class__, VALIDATE_DELETE, [self], ctx=ctx)
+            run(self.__class__, BEFORE_DELETE, [self], ctx=ctx)
+            result = super().delete(*args, **kwargs)
+            run(self.__class__, AFTER_DELETE, [self], ctx=ctx)
+            
         return result

{django_bulk_hooks-0.1.84.dist-info → django_bulk_hooks-0.1.86.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: django-bulk-hooks
-Version: 0.1.84
+Version: 0.1.86
 Summary: Hook-style hooks for Django bulk operations like bulk_create and bulk_update.
 License: MIT
 Keywords: django,bulk,hooks

{django_bulk_hooks-0.1.84.dist-info → django_bulk_hooks-0.1.86.dist-info}/RECORD

@@ -3,14 +3,14 @@ django_bulk_hooks/conditions.py,sha256=cif5R4C_N2HosuYiB03STc2rHb800EJV0Nby-ZPac
 django_bulk_hooks/constants.py,sha256=3x1H1fSUUNo0DZONN7GUVDuySZctTR-jtByBHmAIX5w,303
 django_bulk_hooks/context.py,sha256=HVDT73uSzvgrOR6mdXTvsBm3hLOgBU8ant_mB7VlFuM,380
 django_bulk_hooks/decorators.py,sha256=zstmb27dKcOHu3Atg7cauewCTzPvUmq03mzVKJRi56o,7230
-django_bulk_hooks/engine.py,sha256=b1AO8qyl2VMt1CHyefn1Gt1_ARaV80yBZ3mGXsJ9_eA,2021
+django_bulk_hooks/engine.py,sha256=CUcdBvXE6mq8fBhMP_Q5Z5oygJiVVAuCjNTf-X7o2j0,2826
 django_bulk_hooks/enums.py,sha256=Zo8_tJzuzZ2IKfVc7gZ-0tWPT8q1QhqZbAyoh9ZVJbs,381
 django_bulk_hooks/handler.py,sha256=Qpg_zT6SsQiTlhduvzXxPdG6uynjyR2fBjj-R6HZiXI,4861
 django_bulk_hooks/manager.py,sha256=-V128ACxPAz82ua4jQRFUkjAKtKW4MN5ppz0bHcv5s4,7138
-django_bulk_hooks/models.py,sha256=R-_OLnDUYF9gE985UQp4xZnP71on940-jvPILML2Pj8,3857
+django_bulk_hooks/models.py,sha256=9KvWkmrR0wbTHN6r7-FrSSO9ViS83NvG7iXLBw_iDZs,4793
 django_bulk_hooks/queryset.py,sha256=7lLqhZ-XOYsZ1I3Loxi4Nhz79M8HlTYE413AW8nyeDI,1330
 django_bulk_hooks/registry.py,sha256=Vh78exKYcdZhM27120kQm-iXGOjd_kf9ZUYBZ8eQ2V0,683
-django_bulk_hooks-0.1.84.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
-django_bulk_hooks-0.1.84.dist-info/METADATA,sha256=rNlGfhaSmiA--K8GQ2tMEQ29wlFz1IAGJzeEDIfNh7E,9051
-django_bulk_hooks-0.1.84.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-django_bulk_hooks-0.1.84.dist-info/RECORD,,
+django_bulk_hooks-0.1.86.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
+django_bulk_hooks-0.1.86.dist-info/METADATA,sha256=861ahXik-7mPR1mpk5nr-juB2IfhX-flzuqxbeiLq_A,9051
+django_bulk_hooks-0.1.86.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+django_bulk_hooks-0.1.86.dist-info/RECORD,,