django-bulk-hooks 0.1.58__py3-none-any.whl → 0.1.61__py3-none-any.whl

django_bulk_hooks/context.py

@@ -1,16 +1,16 @@
-import threading
-from collections import deque
-
-_hook_context = threading.local()
-
-
-def get_hook_queue():
-    if not hasattr(_hook_context, "queue"):
-        _hook_context.queue = deque()
-    return _hook_context.queue
-
-
-class TriggerContext:
-    def __init__(self, model_cls, metadata=None):
-        self.model_cls = model_cls
-        self.metadata = metadata or {}
+import threading
+from collections import deque
+
+_hook_context = threading.local()
+
+
+def get_hook_queue():
+    if not hasattr(_hook_context, "queue"):
+        _hook_context.queue = deque()
+    return _hook_context.queue
+
+
+class TriggerContext:
+    def __init__(self, model_cls, metadata=None):
+        self.model_cls = model_cls
+        self.metadata = metadata or {}

django_bulk_hooks/decorators.py

@@ -3,6 +3,8 @@ from functools import wraps
 
 from django.core.exceptions import FieldDoesNotExist
 from django_bulk_hooks.enums import DEFAULT_PRIORITY
+from django_bulk_hooks.constants import BEFORE_UPDATE
+from django_bulk_hooks.registry import register_hook
 
 
 def hook(event, *, model, condition=None, priority=DEFAULT_PRIORITY):
@@ -52,19 +54,29 @@ def select_related(*related_fields):
             if not new_records:
                 return func(*args, **kwargs)
 
-            # In-place preload
+            # Determine which instances actually need preloading
             model_cls = new_records[0].__class__
-            ids = [obj.pk for obj in new_records if obj.pk is not None]
-            if not ids:
-                return func(*args, **kwargs)
+            ids_to_fetch = []
+            for obj in new_records:
+                if obj.pk is None:
+                    continue
+                # if any related field is not already cached on the instance,
+                # mark it for fetching
+                if any(field not in obj._state.fields_cache for field in related_fields):
+                    ids_to_fetch.append(obj.pk)
 
-            fetched = model_cls.objects.select_related(*related_fields).in_bulk(ids)
+            fetched = {}
+            if ids_to_fetch:
+                fetched = model_cls.objects.select_related(*related_fields).in_bulk(ids_to_fetch)
 
             for obj in new_records:
                 preloaded = fetched.get(obj.pk)
                 if not preloaded:
                     continue
                 for field in related_fields:
+                    if field in obj._state.fields_cache:
+                        # don't override values that were explicitly set or already loaded
+                        continue
                     if "." in field:
                         raise ValueError(
                             f"@preload_related does not support nested fields like '{field}'"
@@ -91,3 +103,35 @@ def select_related(*related_fields):
         return wrapper
 
     return decorator
+
+
+def bulk_hook(model_cls, event, when=None, priority=None):
+    """
+    Decorator to register a bulk hook for a model.
+    
+    Args:
+        model_cls: The model class to hook into
+        event: The event to hook into (e.g., BEFORE_UPDATE, AFTER_UPDATE)
+        when: Optional condition for when the hook should run
+        priority: Optional priority for hook execution order
+    """
+    def decorator(func):
+        # Create a simple handler class for the function
+        class FunctionHandler:
+            def __init__(self):
+                self.func = func
+            
+            def handle(self, new_instances, original_instances):
+                return self.func(new_instances, original_instances)
+        
+        # Register the hook using the registry
+        register_hook(
+            model=model_cls,
+            event=event,
+            handler_cls=FunctionHandler,
+            method_name='handle',
+            condition=when,
+            priority=priority or DEFAULT_PRIORITY,
+        )
+        return func
+    return decorator

django_bulk_hooks/manager.py

@@ -10,6 +10,9 @@ from django_bulk_hooks.constants import (
 )
 from django_bulk_hooks.context import TriggerContext
 from django_bulk_hooks.queryset import LifecycleQuerySet
+import logging
+
+logger = logging.getLogger(__name__)
 
 
 class BulkLifecycleManager(models.Manager):
@@ -34,6 +37,18 @@ class BulkLifecycleManager(models.Manager):
             originals = list(model_cls.objects.filter(pk__in=[obj.pk for obj in objs]))
             ctx = TriggerContext(model_cls)
             engine.run(model_cls, BEFORE_UPDATE, objs, originals, ctx=ctx)
+            
+            # Automatically detect fields that were modified during BEFORE_UPDATE hooks
+            modified_fields = self._detect_modified_fields(objs, originals)
+            if modified_fields:
+                # Convert to set for efficient union operation
+                fields_set = set(fields)
+                fields_set.update(modified_fields)
+                fields = list(fields_set)
+                logger.info(
+                    "Automatically including modified fields in bulk_update: %s",
+                    modified_fields
+                )
 
         for i in range(0, len(objs), self.CHUNK_SIZE):
             chunk = objs[i : i + self.CHUNK_SIZE]
@@ -47,6 +62,49 @@ class BulkLifecycleManager(models.Manager):
 
         return objs
 
+    def _detect_modified_fields(self, new_instances, original_instances):
+        """
+        Detect fields that were modified during BEFORE_UPDATE hooks by comparing
+        new instances with their original values.
+        """
+        if not original_instances:
+            return set()
+        
+        # Create a mapping of pk to original instance for efficient lookup
+        original_map = {obj.pk: obj for obj in original_instances if obj.pk is not None}
+        
+        modified_fields = set()
+        
+        for new_instance in new_instances:
+            if new_instance.pk is None:
+                continue
+                
+            original = original_map.get(new_instance.pk)
+            if not original:
+                continue
+            
+            # Compare all fields to detect changes
+            for field in new_instance._meta.fields:
+                if field.name == 'id':
+                    continue
+                    
+                new_value = getattr(new_instance, field.name)
+                original_value = getattr(original, field.name)
+                
+                # Handle different field types appropriately
+                if field.is_relation:
+                    # For foreign keys, compare the pk values
+                    new_pk = new_value.pk if new_value else None
+                    original_pk = original_value.pk if original_value else None
+                    if new_pk != original_pk:
+                        modified_fields.add(field.name)
+                else:
+                    # For regular fields, use direct comparison
+                    if new_value != original_value:
+                        modified_fields.add(field.name)
+        
+        return modified_fields
+
     @transaction.atomic
     def bulk_create(
         self, objs, batch_size=None, ignore_conflicts=False, bypass_hooks=False

{django_bulk_hooks-0.1.58.dist-info → django_bulk_hooks-0.1.61.dist-info}/METADATA

@@ -1,8 +1,7 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: django-bulk-hooks
-Version: 0.1.58
+Version: 0.1.61
 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
 Keywords: django,bulk,hooks
 Author: Konrad Beck
@@ -14,6 +13,7 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: Django (>=4.0)
+Project-URL: Homepage, https://github.com/AugendLimited/django-bulk-hooks
 Project-URL: Repository, https://github.com/AugendLimited/django-bulk-hooks
 Description-Content-Type: text/markdown
 

{django_bulk_hooks-0.1.58.dist-info → django_bulk_hooks-0.1.61.dist-info}/RECORD

@@ -1,17 +1,17 @@
 django_bulk_hooks/__init__.py,sha256=6VFU52Kz5Yjhjr_RhID41nMaXUMZ4na_MVDXEhIMXc4,98
 django_bulk_hooks/conditions.py,sha256=UQP5Y7yyJXqDdUcUukRPxCbvrsV6HqNmIxERFZqBh58,5317
 django_bulk_hooks/constants.py,sha256=Jks1BIADYbap2fpq3Ry0e7w-CiXBCsR9b5h1yan1qoc,192
-django_bulk_hooks/context.py,sha256=VyjgaubjJTTXQHgyoGY4OLBbdCpBwo-RX4bvxRfZ1c4,383
-django_bulk_hooks/decorators.py,sha256=wbT9lbpKZqDbG5vPvtTC6MHYWvjLaVccCwFM9U2nHkA,3159
+django_bulk_hooks/context.py,sha256=hElvgYZoH1nEM2dR7JIi7pbRt67za90ZVy0OzyFWMNg,367
+django_bulk_hooks/decorators.py,sha256=YiczUdfUBI6bk0-7ShfpuvU_n8YCDB6QD4i57NnmvNE,4886
 django_bulk_hooks/engine.py,sha256=l7BzU3lfYhZHZazks8FgHhZB4iLVKuzmcnVcyEQkGpQ,1978
 django_bulk_hooks/enums.py,sha256=Zo8_tJzuzZ2IKfVc7gZ-0tWPT8q1QhqZbAyoh9ZVJbs,381
 django_bulk_hooks/handler.py,sha256=qUulFSPhi__gWHISC4GogeVQ9aDo45bz0Dj421Y6skE,4968
-django_bulk_hooks/manager.py,sha256=jsTgbjmRe5SeakVJxV8wYxqH6f3fRX5zm3q0Wy-M9Nc,4271
+django_bulk_hooks/manager.py,sha256=ajsb0-HaIcb48WCHsD-Cz78umjNWQdbZarMIvBiyA_I,6691
 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=YG9wlvOdujapmxzNhOXTO74Uph0sozUsfc8e5n9KniU,1409
 django_bulk_hooks/registry.py,sha256=yeTi0IhodL61J86ohb5OyITufE28T3ecMbt6RWvkzTs,585
-django_bulk_hooks-0.1.58.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
-django_bulk_hooks-0.1.58.dist-info/METADATA,sha256=kaiVFuUtCftxD836kAaBTkFgH_sBqMtiwAf25A_9iuE,3101
-django_bulk_hooks-0.1.58.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-django_bulk_hooks-0.1.58.dist-info/RECORD,,
+django_bulk_hooks-0.1.61.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
+django_bulk_hooks-0.1.61.dist-info/METADATA,sha256=HLbhpnq0Z5CuOyHm0t7YF_s83UCUMEYd_cbKCpw0la8,3113
+django_bulk_hooks-0.1.61.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+django_bulk_hooks-0.1.61.dist-info/RECORD,,

{django_bulk_hooks-0.1.58.dist-info → django_bulk_hooks-0.1.61.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.9.1
+Generator: poetry-core 2.1.3
 Root-Is-Purelib: true
 Tag: py3-none-any