django-bulk-hooks 0.1.205__py3-none-any.whl → 0.1.207__py3-none-any.whl

django_bulk_hooks/queryset.py

@@ -1,5 +1,5 @@
 from django.db import models, transaction
-from django.db.models import AutoField
+from django.db.models import AutoField, Case, Field, Value, When
 
 from django_bulk_hooks import engine
 from django_bulk_hooks.constants import (
@@ -14,7 +14,6 @@ from django_bulk_hooks.constants import (
     VALIDATE_UPDATE,
 )
 from django_bulk_hooks.context import HookContext
-from django.db.models import When, Value, Case, Field
 
 
 class HookQuerySetMixin:
@@ -22,7 +21,7 @@ class HookQuerySetMixin:
     A mixin that provides bulk hook functionality to any QuerySet.
     This can be dynamically injected into querysets from other managers.
     """
-    
+
     @transaction.atomic
     def delete(self):
         objs = list(self)
@@ -62,6 +61,12 @@ class HookQuerySetMixin:
         }
         originals = [original_map.get(obj.pk) for obj in instances]
 
+        # Check if any of the update values are Subquery objects
+        has_subquery = any(
+            hasattr(value, "query") and hasattr(value, "resolve_expression")
+            for value in kwargs.values()
+        )
+
         # Apply field updates to instances
         for obj in instances:
             for field, value in kwargs.items():
@@ -75,6 +80,26 @@ class HookQuerySetMixin:
         # Call the base QuerySet implementation to avoid recursion
         update_count = super().update(**kwargs)
 
+        # If we used Subquery objects, refresh the instances to get computed values
+        if has_subquery and instances:
+            # Single bulk query to get all updated values
+            # Use values() to get only the fields we need for maximum performance
+            field_names = [f.name for f in model_cls._meta.fields if f.name != "id"]
+            refreshed_data = {
+                obj["pk"]: obj
+                for obj in model_cls._base_manager.filter(pk__in=pks).values(
+                    "pk", *field_names
+                )
+            }
+
+            # Bulk update all instances in memory
+            for instance in instances:
+                if instance.pk in refreshed_data:
+                    data = refreshed_data[instance.pk]
+                    # Bulk copy all field values except primary key
+                    for field_name in field_names:
+                        setattr(instance, field_name, data[field_name])
+
         # Run AFTER_UPDATE hooks
         engine.run(model_cls, AFTER_UPDATE, instances, originals, ctx=ctx)
 
@@ -303,14 +328,14 @@ class HookQuerySetMixin:
         while current_model:
             if not current_model._meta.proxy:
                 chain.append(current_model)
-            
+
             parents = [
                 parent
                 for parent in current_model._meta.parents.keys()
                 if not parent._meta.proxy
             ]
             current_model = parents[0] if parents else None
-        
+
         chain.reverse()
         return chain
 
@@ -584,10 +609,10 @@ class HookQuerySetMixin:
             for field in model._meta.local_fields:
                 if hasattr(field, "auto_now") and field.auto_now:
                     auto_now_fields.add(field.name)
-        
+
         # Combine original fields with auto_now fields
         all_fields = list(fields) + list(auto_now_fields)
-        
+
         # Group fields by model in the inheritance chain
         field_groups = {}
         for field_name in all_fields:
@@ -603,7 +628,7 @@ class HookQuerySetMixin:
         # Process in batches
         batch_size = django_kwargs.get("batch_size") or len(objs)
         total_updated = 0
-        
+
         with transaction.atomic(using=self.db, savepoint=False):
             for i in range(0, len(objs), batch_size):
                 batch = objs[i : i + batch_size]
@@ -614,35 +639,37 @@ class HookQuerySetMixin:
 
         return total_updated
 
-    def _process_mti_bulk_update_batch(self, batch, field_groups, inheritance_chain, **kwargs):
+    def _process_mti_bulk_update_batch(
+        self, batch, field_groups, inheritance_chain, **kwargs
+    ):
         """
         Process a single batch of objects for MTI bulk update.
         Updates each table in the inheritance chain for the batch.
         """
         total_updated = 0
-        
+
         # For MTI, we need to handle parent links correctly
         # The root model (first in chain) has its own PK
         # Child models use the parent link to reference the root PK
         root_model = inheritance_chain[0]
-        
+
         # Get the primary keys from the objects
         # If objects have pk set but are not loaded from DB, use those PKs
         root_pks = []
         for obj in batch:
             # Check both pk and id attributes
-            pk_value = getattr(obj, 'pk', None)
+            pk_value = getattr(obj, "pk", None)
             if pk_value is None:
-                pk_value = getattr(obj, 'id', None)
-            
+                pk_value = getattr(obj, "id", None)
+
             if pk_value is not None:
                 root_pks.append(pk_value)
             else:
                 continue
-        
+
         if not root_pks:
             return 0
-        
+
         # Update each table in the inheritance chain
         for model, model_fields in field_groups.items():
             if not model_fields:
@@ -651,7 +678,7 @@ class HookQuerySetMixin:
             if model == inheritance_chain[0]:
                 # Root model - use primary keys directly
                 pks = root_pks
-                filter_field = 'pk'
+                filter_field = "pk"
             else:
                 # Child model - use parent link field
                 parent_link = None
@@ -659,48 +686,58 @@ class HookQuerySetMixin:
                     if parent_model in model._meta.parents:
                         parent_link = model._meta.parents[parent_model]
                         break
-                
+
                 if parent_link is None:
                     continue
-                
+
                 # For child models, the parent link values should be the same as root PKs
                 pks = root_pks
                 filter_field = parent_link.attname
-            
+
             if pks:
                 base_qs = model._base_manager.using(self.db)
-                
+
                 # Check if records exist
                 existing_count = base_qs.filter(**{f"{filter_field}__in": pks}).count()
-                
+
                 if existing_count == 0:
                     continue
-                
+
                 # Build CASE statements for each field to perform a single bulk update
                 case_statements = {}
                 for field_name in model_fields:
                     field = model._meta.get_field(field_name)
                     when_statements = []
-                    
+
                     for pk, obj in zip(pks, batch):
                         # Check both pk and id attributes for the object
-                        obj_pk = getattr(obj, 'pk', None)
+                        obj_pk = getattr(obj, "pk", None)
                         if obj_pk is None:
-                            obj_pk = getattr(obj, 'id', None)
-                        
+                            obj_pk = getattr(obj, "id", None)
+
                         if obj_pk is None:
                             continue
                         value = getattr(obj, field_name)
-                        when_statements.append(When(**{filter_field: pk}, then=Value(value, output_field=field)))
-                    
-                    case_statements[field_name] = Case(*when_statements, output_field=field)
-                
+                        when_statements.append(
+                            When(
+                                **{filter_field: pk},
+                                then=Value(value, output_field=field),
+                            )
+                        )
+
+                    case_statements[field_name] = Case(
+                        *when_statements, output_field=field
+                    )
+
                 # Execute a single bulk update for all objects in this model
                 try:
-                    updated_count = base_qs.filter(**{f"{filter_field}__in": pks}).update(**case_statements)
+                    updated_count = base_qs.filter(
+                        **{f"{filter_field}__in": pks}
+                    ).update(**case_statements)
                     total_updated += updated_count
                 except Exception as e:
                     import traceback
+
                     traceback.print_exc()
 
         return total_updated
@@ -711,4 +748,5 @@ class HookQuerySet(HookQuerySetMixin, models.QuerySet):
     A QuerySet that provides bulk hook functionality.
     This is the traditional approach for backward compatibility.
     """
+
     pass

{django_bulk_hooks-0.1.205.dist-info → django_bulk_hooks-0.1.207.dist-info}/METADATA

@@ -1,7 +1,8 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.1
 Name: django-bulk-hooks
-Version: 0.1.205
+Version: 0.1.207
 Summary: Hook-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
@@ -13,7 +14,6 @@ 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
 
@@ -132,6 +132,46 @@ Account.objects.update(balance=0.00)
 Account.objects.delete()
 ```
 
+### Subquery Support in Updates
+
+When using `Subquery` objects in update operations, the computed values are automatically available in hooks. The system efficiently refreshes all instances in bulk for optimal performance:
+
+```python
+from django.db.models import Subquery, OuterRef, Sum
+
+def aggregate_revenue_by_ids(self, ids: Iterable[int]) -> int:
+    return self.find_by_ids(ids).update(
+        revenue=Subquery(
+            FinancialTransaction.objects.filter(daily_financial_aggregate_id=OuterRef("pk"))
+            .filter(is_revenue=True)
+            .values("daily_financial_aggregate_id")
+            .annotate(revenue_sum=Sum("amount"))
+            .values("revenue_sum")[:1],
+        ),
+    )
+
+# In your hooks, you can now access the computed revenue value:
+class FinancialAggregateHooks(Hook):
+    @hook(AFTER_UPDATE, model=DailyFinancialAggregate)
+    def log_revenue_update(self, new_records, old_records):
+        for new_record in new_records:
+            # This will now contain the computed value, not the Subquery object
+            print(f"Updated revenue: {new_record.revenue}")
+
+# Bulk operations are optimized for performance:
+def bulk_aggregate_revenue(self, ids: Iterable[int]) -> int:
+    # This will efficiently refresh all instances in a single query
+    return self.filter(id__in=ids).update(
+        revenue=Subquery(
+            FinancialTransaction.objects.filter(daily_financial_aggregate_id=OuterRef("pk"))
+            .filter(is_revenue=True)
+            .values("daily_financial_aggregate_id")
+            .annotate(revenue_sum=Sum("amount"))
+            .values("revenue_sum")[:1],
+        ),
+    )
+```
+
 ## 🧠 Why?
 
 Django's `bulk_` methods bypass signals and `save()`. This package fills that gap with:

{django_bulk_hooks-0.1.205.dist-info → django_bulk_hooks-0.1.207.dist-info}/RECORD

@@ -9,9 +9,9 @@ django_bulk_hooks/handler.py,sha256=xZt8iNdYF-ACz-MnKMY0co6scWINU5V5wC1lyDn844k,
 django_bulk_hooks/manager.py,sha256=nfWiwU5-yAoxdnQsUMohxtyCpkV0MBv6X3wmipr9eQY,3697
 django_bulk_hooks/models.py,sha256=7fnx5xd4HWXfLVlFhhiRzR92JRWFEuxgk6aSWLEsyJg,3996
 django_bulk_hooks/priority.py,sha256=HG_2D35nga68lBCZmSXTcplXrjFoRgZFRDOy4ROKonY,376
-django_bulk_hooks/queryset.py,sha256=e4rh5rwxIMp4Q1Z49XF5JSVGctulxyG7lkk0OAPxP40,30216
+django_bulk_hooks/queryset.py,sha256=YMw8Za_1aqLkCmTqhYKnznq95IVlie3Wx9NBDND7KcM,31387
 django_bulk_hooks/registry.py,sha256=-mQBizJ06nz_tajZBinViKx_uP2Tbc1tIpTEMv7lwKA,705
-django_bulk_hooks-0.1.205.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
-django_bulk_hooks-0.1.205.dist-info/METADATA,sha256=8CqllhoMI6B-20rLs6m2IfqUOhWhkSCHO7nsfwAzAeA,7418
-django_bulk_hooks-0.1.205.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-django_bulk_hooks-0.1.205.dist-info/RECORD,,
+django_bulk_hooks-0.1.207.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
+django_bulk_hooks-0.1.207.dist-info/METADATA,sha256=6XzEu4cdHtO1kVPzw-vuUmoFjbEUaVrTdEEKAFqEP60,9049
+django_bulk_hooks-0.1.207.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+django_bulk_hooks-0.1.207.dist-info/RECORD,,

{django_bulk_hooks-0.1.205.dist-info → django_bulk_hooks-0.1.207.dist-info}/WHEEL

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