django-bulk-hooks 0.1.83__py3-none-any.whl → 0.1.85__py3-none-any.whl

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,46 +72,60 @@ 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, let Django save first so relationships are set up
+            # For create operations, run BEFORE hooks first
+            with patch_foreign_key_behavior():
+                run(self.__class__, BEFORE_CREATE, [self], ctx=ctx)
+            
+            # Then let Django save
             super().save(*args, **kwargs)
             
-            # Now run hooks after Django has set up the object
-            run(self.__class__, BEFORE_CREATE, [self], ctx=ctx)
-            run(self.__class__, AFTER_CREATE, [self], ctx=ctx)
+            # Then run AFTER hooks
+            with patch_foreign_key_behavior():
+                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)
                 
-                # Let Django save first
+                # Run BEFORE hooks first
+                with patch_foreign_key_behavior():
+                    run(self.__class__, BEFORE_UPDATE, [self], [old_instance], ctx=ctx)
+                
+                # Then let Django save
                 super().save(*args, **kwargs)
                 
-                # Now run hooks after Django has set up the object
-                run(self.__class__, BEFORE_UPDATE, [self], [old_instance], ctx=ctx)
-                run(self.__class__, AFTER_UPDATE, [self], [old_instance], ctx=ctx)
+                # Then run AFTER hooks
+                with patch_foreign_key_behavior():
+                    run(self.__class__, AFTER_UPDATE, [self], [old_instance], ctx=ctx)
             except self.__class__.DoesNotExist:
                 # If the old instance doesn't exist, treat as create
+                with patch_foreign_key_behavior():
+                    run(self.__class__, BEFORE_CREATE, [self], ctx=ctx)
+                
                 super().save(*args, **kwargs)
                 
-                run(self.__class__, BEFORE_CREATE, [self], ctx=ctx)
-                run(self.__class__, AFTER_CREATE, [self], ctx=ctx)
+                with patch_foreign_key_behavior():
+                    run(self.__class__, AFTER_CREATE, [self], ctx=ctx)
 
         return self
 
@@ -90,12 +133,15 @@ class HookModelMixin(models.Model):
         ctx = HookContext(self.__class__)
 
         # Run validation hooks first
-        run(self.__class__, VALIDATE_DELETE, [self], ctx=ctx)
+        with patch_foreign_key_behavior():
+            run(self.__class__, VALIDATE_DELETE, [self], ctx=ctx)
 
         # Then run business logic hooks
-        run(self.__class__, BEFORE_DELETE, [self], ctx=ctx)
+        with patch_foreign_key_behavior():
+            run(self.__class__, BEFORE_DELETE, [self], ctx=ctx)
 
         result = super().delete(*args, **kwargs)
 
-        run(self.__class__, AFTER_DELETE, [self], ctx=ctx)
+        with patch_foreign_key_behavior():
+            run(self.__class__, AFTER_DELETE, [self], ctx=ctx)
         return result

{django_bulk_hooks-0.1.83.dist-info → django_bulk_hooks-0.1.85.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: django-bulk-hooks
-Version: 0.1.83
+Version: 0.1.85
 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.83.dist-info → django_bulk_hooks-0.1.85.dist-info}/RECORD

@@ -7,10 +7,10 @@ django_bulk_hooks/engine.py,sha256=b1AO8qyl2VMt1CHyefn1Gt1_ARaV80yBZ3mGXsJ9_eA,2
 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=m_dsCcKX8euGU-aDvlkY-vi5QTsRaXhyT-6EZotYuXM,3813
+django_bulk_hooks/models.py,sha256=CD-v8JAZCOnGAHsAxOrlJz5HiOOXNVhGi7p8n3m4oLk,5423
 django_bulk_hooks/queryset.py,sha256=7lLqhZ-XOYsZ1I3Loxi4Nhz79M8HlTYE413AW8nyeDI,1330
 django_bulk_hooks/registry.py,sha256=Vh78exKYcdZhM27120kQm-iXGOjd_kf9ZUYBZ8eQ2V0,683
-django_bulk_hooks-0.1.83.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
-django_bulk_hooks-0.1.83.dist-info/METADATA,sha256=G6_6NsM-MqUem0Y5fmZz0pktXtmsaIVpglFeBDIv7JM,9051
-django_bulk_hooks-0.1.83.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-django_bulk_hooks-0.1.83.dist-info/RECORD,,
+django_bulk_hooks-0.1.85.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
+django_bulk_hooks-0.1.85.dist-info/METADATA,sha256=JJ0uZaiHpf5An0kqNQySULj4hSSpn0JHEppz-xUwnMg,9051
+django_bulk_hooks-0.1.85.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+django_bulk_hooks-0.1.85.dist-info/RECORD,,