django-bulk-hooks 0.1.182__tar.gz → 0.1.183__tar.gz

{django_bulk_hooks-0.1.182 → django_bulk_hooks-0.1.183}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: django-bulk-hooks
-Version: 0.1.182
+Version: 0.1.183
 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 → django_bulk_hooks-0.1.183}/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,26 @@ 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()
+            # Use Django's update() with CASE statements for bulk update
+            pks = [obj.pk for obj in objs]
+            
+            if pks:
+                base_qs = model._base_manager.using(self.db)
                 
-                # 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)
-                
-                # Also set the pk property for compatibility
-                temp_obj.pk = pk_value
-                
-                # 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(pk=pk, then=Value(value, output_field=field)))
+                    
+                    case_statements[field_name] = Case(*when_statements, output_field=field)
+                
+                # Execute the update
+                updated_count = base_qs.filter(pk__in=pks).update(**case_statements)
+                total_updated += updated_count
 
         return total_updated

{django_bulk_hooks-0.1.182 → django_bulk_hooks-0.1.183}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "django-bulk-hooks"
-version = "0.1.182"
+version = "0.1.183"
 description = "Hook-style hooks for Django bulk operations like bulk_create and bulk_update."
 authors = ["Konrad Beck <konrad.beck@merchantcapital.co.za>"]
 readme = "README.md"