django-bulk-hooks 0.1.240__py3-none-any.whl → 0.1.241__py3-none-any.whl

django_bulk_hooks/queryset.py

@@ -164,11 +164,24 @@ class HookQuerySetMixin:
                             output_field = field_obj
                             target_name = field_name
 
-                        when_statements.append(
-                            When(
-                                pk=obj_pk, then=Value(value, output_field=output_field)
+                        # Special handling for Subquery values in CASE statements
+                        if isinstance(value, Subquery):
+                            logger.debug(f"Creating When statement with Subquery for {field_name}")
+                            # Ensure the Subquery has proper output_field
+                            if not hasattr(value, 'output_field') or value.output_field is None:
+                                value.output_field = output_field
+                                logger.debug(f"Set output_field for Subquery in When statement to {output_field}")
+                            when_statements.append(
+                                When(
+                                    pk=obj_pk, then=value
+                                )
+                            )
+                        else:
+                            when_statements.append(
+                                When(
+                                    pk=obj_pk, then=Value(value, output_field=output_field)
+                                )
                             )
-                        )
 
                     if when_statements:
                         case_statements[target_name] = Case(
@@ -177,6 +190,22 @@ class HookQuerySetMixin:
 
                 # Merge extra CASE updates into kwargs for DB update
                 if case_statements:
+                    logger.debug(f"Adding case statements to kwargs: {list(case_statements.keys())}")
+                    for field_name, case_stmt in case_statements.items():
+                        logger.debug(f"Case statement for {field_name}: {type(case_stmt).__name__}")
+                        # Check if the case statement contains Subquery objects
+                        if hasattr(case_stmt, 'get_source_expressions'):
+                            source_exprs = case_stmt.get_source_expressions()
+                            for expr in source_exprs:
+                                if isinstance(expr, Subquery):
+                                    logger.debug(f"Case statement for {field_name} contains Subquery")
+                                elif hasattr(expr, 'get_source_expressions'):
+                                    # Check nested expressions (like Value objects)
+                                    nested_exprs = expr.get_source_expressions()
+                                    for nested_expr in nested_exprs:
+                                        if isinstance(nested_expr, Subquery):
+                                            logger.debug(f"Case statement for {field_name} contains nested Subquery")
+                    
                     kwargs = {**kwargs, **case_statements}
 
         # Use Django's built-in update logic directly
@@ -208,6 +237,40 @@ class HookQuerySetMixin:
                 else:
                     logger.debug(f"Subquery for field {key} already has output_field: {value.output_field}")
                 safe_kwargs[key] = value
+            elif hasattr(value, 'get_source_expressions') and hasattr(value, 'resolve_expression'):
+                # Handle Case statements and other complex expressions
+                logger.debug(f"Found complex expression for field {key}: {type(value).__name__}")
+                
+                # Check if this expression contains any Subquery objects
+                source_expressions = value.get_source_expressions()
+                has_nested_subquery = False
+                
+                for expr in source_expressions:
+                    if isinstance(expr, Subquery):
+                        has_nested_subquery = True
+                        logger.debug(f"Found nested Subquery in {type(value).__name__}")
+                        # Ensure the nested Subquery has proper output_field
+                        if not hasattr(expr, 'output_field') or expr.output_field is None:
+                            try:
+                                field = model_cls._meta.get_field(key)
+                                expr.output_field = field
+                                logger.debug(f"Set output_field for nested Subquery to {field}")
+                            except Exception as e:
+                                logger.error(f"Failed to set output_field for nested Subquery: {e}")
+                                raise
+                
+                if has_nested_subquery:
+                    logger.debug(f"Expression contains Subquery, ensuring proper output_field")
+                    # Try to resolve the expression to ensure it's properly formatted
+                    try:
+                        resolved_value = value.resolve_expression(None, None)
+                        safe_kwargs[key] = resolved_value
+                        logger.debug(f"Successfully resolved expression for {key}")
+                    except Exception as e:
+                        logger.error(f"Failed to resolve expression for {key}: {e}")
+                        raise
+                else:
+                    safe_kwargs[key] = value
             else:
                 logger.debug(f"Non-Subquery value for field {key}: {type(value).__name__}")
                 safe_kwargs[key] = value

{django_bulk_hooks-0.1.240.dist-info → django_bulk_hooks-0.1.241.dist-info}/METADATA

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

@@ -9,9 +9,9 @@ django_bulk_hooks/handler.py,sha256=Bx-W6yyiciKMyy-BRxUt3CmRPCrX9_LhQgU-5LaJTjg,
 django_bulk_hooks/manager.py,sha256=nfWiwU5-yAoxdnQsUMohxtyCpkV0MBv6X3wmipr9eQY,3697
 django_bulk_hooks/models.py,sha256=exnXYVKEVbYAXhChCP8VdWTnKCnm9DiTcokEIBee1I0,4350
 django_bulk_hooks/priority.py,sha256=HG_2D35nga68lBCZmSXTcplXrjFoRgZFRDOy4ROKonY,376
-django_bulk_hooks/queryset.py,sha256=bi8jE8yvl2ih0M3LbzW8TZvl7JFKbgysILRXIpci6KM,42055
+django_bulk_hooks/queryset.py,sha256=kUmV4izYquYsvtcR3PH8TkY3PBm-Kt8c8B4HO4ck0uo,46280
 django_bulk_hooks/registry.py,sha256=GRUTGVQEO2sdkC9OaZ9Q3U7mM-3Ix83uTyvrlTtpatw,1317
-django_bulk_hooks-0.1.240.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
-django_bulk_hooks-0.1.240.dist-info/METADATA,sha256=hkqwdtVcH2Te0py2ryH-RXgAJpX93e0IDFteueUBYtQ,9061
-django_bulk_hooks-0.1.240.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-django_bulk_hooks-0.1.240.dist-info/RECORD,,
+django_bulk_hooks-0.1.241.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
+django_bulk_hooks-0.1.241.dist-info/METADATA,sha256=DUraAu1YHa04565ngESG8QnmmUwrcvrD1-4oqoOnBhY,9061
+django_bulk_hooks-0.1.241.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+django_bulk_hooks-0.1.241.dist-info/RECORD,,