django-bulk-hooks 0.2.8__tar.gz → 0.2.9__tar.gz

{django_bulk_hooks-0.2.8 → django_bulk_hooks-0.2.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: django-bulk-hooks
-Version: 0.2.8
+Version: 0.2.9
 Summary: Hook-style hooks for Django bulk operations like bulk_create and bulk_update.
 License: MIT
 Keywords: django,bulk,hooks

{django_bulk_hooks-0.2.8 → django_bulk_hooks-0.2.9}/django_bulk_hooks/operations/analyzer.py

@@ -257,6 +257,10 @@ class ModelAnalyzer:
         F() expressions, Subquery, Case, etc.) into concrete values and applies
         them to the instances.
         
+        CRITICAL: When setting FK fields by their attname (e.g., business_id),
+        we must manually clear the relationship cache (e.g., business) to match
+        Django's ForeignKey descriptor behavior.
+        
         Args:
             instances: List of model instances to update
             update_kwargs: Dict of {field_name: value_or_expression}
@@ -270,8 +274,66 @@ class ModelAnalyzer:
         fields_updated = list(update_kwargs.keys())
         
         for field_name, value in update_kwargs.items():
+            # Determine if this is a FK field being set by its attname
+            field_info = self._get_fk_field_info(field_name)
+            
             for instance in instances:
                 resolved_value = self.resolve_expression(field_name, value, instance)
                 setattr(instance, field_name, resolved_value)
+                
+                # Clear relationship cache when FK field is set directly
+                # This replicates Django's ForeignKey descriptor behavior
+                if field_info and field_info['is_fk_attname']:
+                    self._clear_fk_cache(instance, field_info['accessor_name'])
+        
+        return fields_updated
+
+    def _get_fk_field_info(self, field_name):
+        """
+        Get information about a FK field if field_name is a FK attname.
+        
+        Args:
+            field_name: Field name to check
+            
+        Returns:
+            Dict with FK info or None if not a FK field
+        """
+        try:
+            # Check all fields to find if this is a FK attname
+            for field in self.model_cls._meta.get_fields():
+                if (field.is_relation and 
+                    not field.many_to_many and 
+                    not field.one_to_many and
+                    hasattr(field, 'attname') and 
+                    field.attname == field_name):
+                    # This is a FK field being set by its attname (e.g., business_id)
+                    return {
+                        'is_fk_attname': True,
+                        'accessor_name': field.name,  # e.g., 'business'
+                        'field': field
+                    }
+        except Exception as e:
+            logger.debug(f"Error checking FK field info for {field_name}: {e}")
+        
+        return None
+
+    def _clear_fk_cache(self, instance, accessor_name):
+        """
+        Clear cached relationship when FK field is set directly.
         
-        return fields_updated
+        This replicates what Django's ForeignKey descriptor __set__ does:
+        when you set a FK field, Django clears the cached related object.
+        
+        Args:
+            instance: Model instance
+            accessor_name: Name of the relationship accessor (e.g., 'business')
+        """
+        try:
+            if hasattr(instance, '_state') and hasattr(instance._state, 'fields_cache'):
+                instance._state.fields_cache.pop(accessor_name, None)
+                logger.debug(
+                    f"Cleared FK cache for '{accessor_name}' on {self.model_cls.__name__}"
+                )
+        except Exception as e:
+            # Don't fail the operation, just log
+            logger.debug(f"Could not clear FK cache for {accessor_name}: {e}")

{django_bulk_hooks-0.2.8 → django_bulk_hooks-0.2.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "django-bulk-hooks"
-version = "0.2.8"
+version = "0.2.9"
 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"