django-bulk-hooks 0.1.182__py3-none-any.whl → 0.1.184__py3-none-any.whl

django_bulk_hooks/queryset.py

@@ -14,6 +14,7 @@ from django_bulk_hooks.constants import (
     VALIDATE_UPDATE,
 )
 from django_bulk_hooks.context import HookContext
+from django.db.models import When, Value, Case
 
 
 class HookQuerySet(models.QuerySet):
@@ -554,35 +555,46 @@ class HookQuerySet(models.QuerySet):
             if not model_fields:
                 continue
 
-            # Get objects that have this model's fields
-            model_objs = []
-            for obj in objs:
-                # Create a temporary object with just this model's fields
-                temp_obj = model()
+            # 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
+            if model == inheritance_chain[0]:
+                # Root model - use primary keys directly
+                pks = [obj.pk for obj in objs]
+                filter_field = 'pk'
+            else:
+                # Child model - use parent link field
+                parent_link = None
+                for parent_model in inheritance_chain:
+                    if parent_model in model._meta.parents:
+                        parent_link = model._meta.parents[parent_model]
+                        break
                 
-                # Set the primary key properly using the model's pk field
-                pk_field = model._meta.pk
-                pk_value = obj.pk
-                setattr(temp_obj, pk_field.attname, pk_value)
+                if parent_link is None:
+                    # This shouldn't happen in proper MTI, but handle gracefully
+                    continue
                 
-                # Also set the pk property for compatibility
-                temp_obj.pk = pk_value
+                # Get the parent link values (these should be the same as the root PKs)
+                pks = [getattr(obj, parent_link.attname) for obj in objs]
+                filter_field = parent_link.attname
+            
+            if pks:
+                base_qs = model._base_manager.using(self.db)
                 
-                # Mark the object as not being added (it's an existing object)
-                temp_obj._state.adding = False
-                temp_obj._state.db = self.db
-
-                # Copy only the fields for this model
+                # Build CASE statements for each field
+                case_statements = {}
                 for field_name in model_fields:
-                    if hasattr(obj, field_name):
-                        setattr(temp_obj, field_name, getattr(obj, field_name))
-
-                model_objs.append(temp_obj)
-
-            # Use Django's bulk_update for this model's table
-            # This will automatically use batch_size from kwargs
-            base_qs = model._base_manager.using(self.db)
-            updated_count = base_qs.bulk_update(model_objs, model_fields, **kwargs)
-            total_updated += updated_count
+                    field = model._meta.get_field(field_name)
+                    when_statements = []
+                    
+                    for pk, obj in zip(pks, objs):
+                        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)
+                
+                # Execute the update using the appropriate filter field
+                updated_count = base_qs.filter(**{f"{filter_field}__in": pks}).update(**case_statements)
+                total_updated += updated_count
 
         return total_updated

{django_bulk_hooks-0.1.182.dist-info → django_bulk_hooks-0.1.184.dist-info}/METADATA

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

@@ -9,9 +9,9 @@ django_bulk_hooks/handler.py,sha256=xZt8iNdYF-ACz-MnKMY0co6scWINU5V5wC1lyDn844k,
 django_bulk_hooks/manager.py,sha256=KLEjpQRt4WlzgBAf_X3XOAPUQM8Jmc1fIt8yr62FPQc,3044
 django_bulk_hooks/models.py,sha256=7fnx5xd4HWXfLVlFhhiRzR92JRWFEuxgk6aSWLEsyJg,3996
 django_bulk_hooks/priority.py,sha256=HG_2D35nga68lBCZmSXTcplXrjFoRgZFRDOy4ROKonY,376
-django_bulk_hooks/queryset.py,sha256=mWrbt6I0N8--IZ01eXyWsfFuVj62bbvtr06frM9PCes,24844
+django_bulk_hooks/queryset.py,sha256=gQvwa506wYi206B13I-lwkZi5Ua9nWcKctwCCgLGuaQ,25614
 django_bulk_hooks/registry.py,sha256=-mQBizJ06nz_tajZBinViKx_uP2Tbc1tIpTEMv7lwKA,705
-django_bulk_hooks-0.1.182.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
-django_bulk_hooks-0.1.182.dist-info/METADATA,sha256=sBgWwmOAkiN9zxOCkaf0C-0v2MVMsbPum20NPQhAArI,6951
-django_bulk_hooks-0.1.182.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-django_bulk_hooks-0.1.182.dist-info/RECORD,,
+django_bulk_hooks-0.1.184.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
+django_bulk_hooks-0.1.184.dist-info/METADATA,sha256=zj7hawgW0Z19ZUhnTcTMcx-pMXy_PvepfzUdstynZjU,6951
+django_bulk_hooks-0.1.184.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+django_bulk_hooks-0.1.184.dist-info/RECORD,,