django-bulk-hooks 0.2.60__py3-none-any.whl → 0.2.62__py3-none-any.whl

django_bulk_hooks/operations/mti_plans.py

@@ -1,99 +1,103 @@
-"""
-MTI operation plans - Data structures for multi-table inheritance operations.
-
-These are pure data structures returned by MTIHandler to be executed by BulkExecutor.
-This separates planning (logic) from execution (database operations).
-"""
-
-from dataclasses import dataclass
-from dataclasses import field
-from typing import Any
-
-
-@dataclass
-class ParentLevel:
-    """
-    Represents one level in the parent hierarchy for MTI bulk create.
-
-    Attributes:
-        model_class: The parent model class for this level
-        objects: List of parent instances to create
-        original_object_map: Maps parent instance id() -> original object id()
-        update_conflicts: Whether to enable UPSERT for this level
-        unique_fields: Fields for conflict detection (if update_conflicts=True)
-        update_fields: Fields to update on conflict (if update_conflicts=True)
-    """
-
-    model_class: Any
-    objects: list[Any]
-    original_object_map: dict[int, int] = field(default_factory=dict)
-    update_conflicts: bool = False
-    unique_fields: list[str] = field(default_factory=list)
-    update_fields: list[str] = field(default_factory=list)
-
-
-@dataclass
-class MTICreatePlan:
-    """
-    Plan for executing bulk_create on an MTI model.
-
-    This plan describes WHAT to create, not HOW to create it.
-    The executor is responsible for executing this plan.
-
-    Attributes:
-        inheritance_chain: List of model classes from root to child
-        parent_levels: List of ParentLevel objects, one per parent model
-        child_objects: List of child instances to create (not yet with parent links)
-        child_model: The child model class
-        original_objects: Original objects provided by user
-        batch_size: Batch size for operations
-        existing_record_ids: Set of id() of original objects that represent existing DB records
-        update_conflicts: Whether this is an upsert operation
-        unique_fields: Fields used for conflict detection
-        update_fields: Fields to update on conflict
-    """
-
-    inheritance_chain: list[Any]
-    parent_levels: list[ParentLevel]
-    child_objects: list[Any]
-    child_model: Any
-    original_objects: list[Any]
-    batch_size: int = None
-    existing_record_ids: set = field(default_factory=set)
-    update_conflicts: bool = False
-    unique_fields: list[str] = field(default_factory=list)
-    update_fields: list[str] = field(default_factory=list)
-
-
-@dataclass
-class ModelFieldGroup:
-    """
-    Represents fields to update for one model in the inheritance chain.
-
-    Attributes:
-        model_class: The model class
-        fields: List of field names to update on this model
-        filter_field: Field to use for filtering (e.g., 'pk' or parent link attname)
-    """
-
-    model_class: Any
-    fields: list[str]
-    filter_field: str = "pk"
-
-
-@dataclass
-class MTIUpdatePlan:
-    """
-    Plan for executing bulk_update on an MTI model.
-
-    Attributes:
-        inheritance_chain: List of model classes from root to child
-        field_groups: List of ModelFieldGroup objects
-        objects: Objects to update
-        batch_size: Batch size for operations
-    """
-
-    inheritance_chain: list[Any]
-    field_groups: list[ModelFieldGroup]
-    objects: list[Any]
-    batch_size: int = None
+"""
+MTI operation plans - Data structures for multi-table inheritance operations.
+
+These are pure data structures returned by MTIHandler to be executed by BulkExecutor.
+This separates planning (logic) from execution (database operations).
+"""
+
+from dataclasses import dataclass
+from dataclasses import field
+from typing import Any
+
+
+@dataclass
+class ParentLevel:
+    """
+    Represents one level in the parent hierarchy for MTI bulk create.
+
+    Attributes:
+        model_class: The parent model class for this level
+        objects: List of parent instances to create
+        original_object_map: Maps parent instance id() -> original object id()
+        update_conflicts: Whether to enable UPSERT for this level
+        unique_fields: Fields for conflict detection (if update_conflicts=True)
+        update_fields: Fields to update on conflict (if update_conflicts=True)
+    """
+
+    model_class: Any
+    objects: list[Any]
+    original_object_map: dict[int, int] = field(default_factory=dict)
+    update_conflicts: bool = False
+    unique_fields: list[str] = field(default_factory=list)
+    update_fields: list[str] = field(default_factory=list)
+
+
+@dataclass
+class MTICreatePlan:
+    """
+    Plan for executing bulk_create on an MTI model.
+
+    This plan describes WHAT to create, not HOW to create it.
+    The executor is responsible for executing this plan.
+
+    Attributes:
+        inheritance_chain: List of model classes from root to child
+        parent_levels: List of ParentLevel objects, one per parent model
+        child_objects: List of child instances to create (not yet with parent links)
+        child_model: The child model class
+        original_objects: Original objects provided by user
+        batch_size: Batch size for operations
+        existing_record_ids: Set of id() of original objects that represent existing DB records
+        update_conflicts: Whether this is an upsert operation
+        unique_fields: Fields used for conflict detection (original, unfiltered)
+        update_fields: Fields to update on conflict (original, unfiltered)
+        child_unique_fields: Pre-filtered field objects for child table conflict detection
+        child_update_fields: Pre-filtered field objects for child table updates
+    """
+
+    inheritance_chain: list[Any]
+    parent_levels: list[ParentLevel]
+    child_objects: list[Any]
+    child_model: Any
+    original_objects: list[Any]
+    batch_size: int = None
+    existing_record_ids: set = field(default_factory=set)
+    update_conflicts: bool = False
+    unique_fields: list[str] = field(default_factory=list)
+    update_fields: list[str] = field(default_factory=list)
+    child_unique_fields: list = field(default_factory=list)  # Field objects for child table
+    child_update_fields: list = field(default_factory=list)  # Field objects for child table
+
+
+@dataclass
+class ModelFieldGroup:
+    """
+    Represents fields to update for one model in the inheritance chain.
+
+    Attributes:
+        model_class: The model class
+        fields: List of field names to update on this model
+        filter_field: Field to use for filtering (e.g., 'pk' or parent link attname)
+    """
+
+    model_class: Any
+    fields: list[str]
+    filter_field: str = "pk"
+
+
+@dataclass
+class MTIUpdatePlan:
+    """
+    Plan for executing bulk_update on an MTI model.
+
+    Attributes:
+        inheritance_chain: List of model classes from root to child
+        field_groups: List of ModelFieldGroup objects
+        objects: Objects to update
+        batch_size: Batch size for operations
+    """
+
+    inheritance_chain: list[Any]
+    field_groups: list[ModelFieldGroup]
+    objects: list[Any]
+    batch_size: int = None

django_bulk_hooks/operations/record_classifier.py

@@ -64,7 +64,7 @@ class RecordClassifier:
             # Build lookup dict for this object's unique fields
             lookup = {}
             normalized_values = []
-            
+
             for field_name in unique_fields:
                 # Use centralized field value extraction for consistent FK handling
                 value = get_field_value_for_db(obj, field_name, query_model)

django_bulk_hooks/queryset.py

@@ -11,6 +11,8 @@ import logging
 from django.db import models
 from django.db import transaction
 
+from django_bulk_hooks.helpers import extract_pks
+
 logger = logging.getLogger(__name__)
 
 
@@ -132,7 +134,11 @@ class HookQuerySet(models.QuerySet):
 
     @transaction.atomic
     def bulk_delete(
-        self, objs, bypass_hooks=False, bypass_validation=False, **kwargs,
+        self,
+        objs,
+        bypass_hooks=False,
+        bypass_validation=False,
+        **kwargs,
     ):
         """
         Delete multiple objects with hook support.
@@ -148,7 +154,7 @@ class HookQuerySet(models.QuerySet):
             Tuple of (count, details dict)
         """
         # Filter queryset to only these objects
-        pks = [obj.pk for obj in objs if obj.pk is not None]
+        pks = extract_pks(objs)
         if not pks:
             return 0
 

django_bulk_hooks/registry.py

@@ -116,7 +116,6 @@ class HookRegistry:
             if not self._hooks[key]:
                 del self._hooks[key]
 
-
     def clear(self) -> None:
         """
         Clear all registered hooks.
@@ -132,7 +131,6 @@ class HookRegistry:
             HookMeta._registered.clear()
             HookMeta._class_hook_map.clear()
 
-
     def list_all(self) -> dict[tuple[type, str], list[HookInfo]]:
         """
         Get all registered hooks for debugging.

{django_bulk_hooks-0.2.60.dist-info → django_bulk_hooks-0.2.62.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: django-bulk-hooks
-Version: 0.2.60
+Version: 0.2.62
 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.62.dist-info/RECORD

@@ -0,0 +1,27 @@
+django_bulk_hooks/__init__.py,sha256=ujBvX-GY4Pg4ACBh7RXm_MOmi2eAowf5s7pG2SXWdpo,2276
+django_bulk_hooks/changeset.py,sha256=qnMD3bR2cNh8ZM8J6ASR5ly5Rjx-tPzXBYkqIjKGW98,6568
+django_bulk_hooks/conditions.py,sha256=ar4pGjtxLKmgSIlO4S6aZFKmaBNchLtxMmWpkn4g9RU,8114
+django_bulk_hooks/constants.py,sha256=PxpEETaO6gdENcTPoXS586lerGKVP3nmjpDvOkmhYxI,509
+django_bulk_hooks/context.py,sha256=mqaC5-yESDTA5ruI7fuXlt8qSgKuOFp0mjq7h1-4HdQ,1926
+django_bulk_hooks/decorators.py,sha256=TdkO4FJyFrVU2zqK6Y_6JjEJ4v3nbKkk7aa22jN10sk,11994
+django_bulk_hooks/dispatcher.py,sha256=yHfmAWChj9rsZwO4jhYDVtJde_Z8bPe9WHRjkYBMfEo,8151
+django_bulk_hooks/enums.py,sha256=Zo8_tJzuzZ2IKfVc7gZ-0tWPT8q1QhqZbAyoh9ZVJbs,381
+django_bulk_hooks/factory.py,sha256=ezrVM5U023KZqOBbJXb6lYUP-pE7WJmi8Olh2Ew-7RA,18085
+django_bulk_hooks/handler.py,sha256=SRCrMzgolrruTkvMnYBFmXLR-ABiw0JiH3605PEdCZM,4207
+django_bulk_hooks/helpers.py,sha256=tMxUI5oWhbtWByzCCR0Qcj1CgZ6iP5Jfx03EqVmEhxU,7597
+django_bulk_hooks/manager.py,sha256=aDuP87DZLWWbDK2qeA7usl3pxoIjHFIWnQNi_jEq6z0,4446
+django_bulk_hooks/models.py,sha256=TWN_F-SsLGPx9jrkNT9pmJFR5VsZ0Z_QaVOZOmt7bpw,2434
+django_bulk_hooks/operations/__init__.py,sha256=BtJYjmRhe_sScivLsniDaZmBkm0ZLvcmzXFKL7QY2Xg,550
+django_bulk_hooks/operations/analyzer.py,sha256=Pz8mc-EL8KDOfLQFYiRuN-r0OmINW3nIBhRJJCma-yo,10360
+django_bulk_hooks/operations/bulk_executor.py,sha256=po8V_2H3ULiE0RYJ-wbaRIz52SKhss81UHwuQjlz3H8,26214
+django_bulk_hooks/operations/coordinator.py,sha256=NPPkiEIMgbJXEIOtBqvy3OH4l0pOK_NL5jwT7Og9I4o,33765
+django_bulk_hooks/operations/field_utils.py,sha256=cQ9w4xdk-z3PrMLFvRzVV07Wc0D2qbpSepwoupqwQH8,7888
+django_bulk_hooks/operations/mti_handler.py,sha256=Vmz0C0gtYDvbybmb4cDzIaGglSaQK4DQVkaBK-WuQeE,25855
+django_bulk_hooks/operations/mti_plans.py,sha256=HIRJgogHPpm6MV7nZZ-sZhMLUnozpZPV2SzwQHLRzYc,3667
+django_bulk_hooks/operations/record_classifier.py,sha256=It85hJC2K-UsEOLbTR-QBdY5UPV-acQIJ91TSGa7pYo,7053
+django_bulk_hooks/queryset.py,sha256=g_9OtOTC8FXY0hBwYr2FCqQ3mYXbfJTFPLlFV3SHmWQ,5600
+django_bulk_hooks/registry.py,sha256=4HxP1mVK2z4VzvlohbEw2359wM21UJZJYagJJ1komM0,7947
+django_bulk_hooks-0.2.62.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
+django_bulk_hooks-0.2.62.dist-info/METADATA,sha256=Ib--glZc6N4AtvIcIGqMAQUTdAk0AeCmrV2wv6u8iWQ,9265
+django_bulk_hooks-0.2.62.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+django_bulk_hooks-0.2.62.dist-info/RECORD,,

django_bulk_hooks-0.2.60.dist-info/RECORD

@@ -1,27 +0,0 @@
-django_bulk_hooks/__init__.py,sha256=ujBvX-GY4Pg4ACBh7RXm_MOmi2eAowf5s7pG2SXWdpo,2276
-django_bulk_hooks/changeset.py,sha256=wU6wckJEQ_hG5UZq_9g_kWODoKwEj99JrtacVUQ9BxA,7445
-django_bulk_hooks/conditions.py,sha256=v2DMFmWI7bppBQw5qdbO5CmQRN_QtUwnBjcyKBJLLbw,8030
-django_bulk_hooks/constants.py,sha256=PxpEETaO6gdENcTPoXS586lerGKVP3nmjpDvOkmhYxI,509
-django_bulk_hooks/context.py,sha256=mqaC5-yESDTA5ruI7fuXlt8qSgKuOFp0mjq7h1-4HdQ,1926
-django_bulk_hooks/decorators.py,sha256=P7cvzFgORJRW-YQHNAxNXqQOP9OywBmA7Rz9kiJoxUk,12237
-django_bulk_hooks/dispatcher.py,sha256=CiKYe5ecUPu5TYUZq8ToaRT40TkLc5l5mczgf5XDzGA,8217
-django_bulk_hooks/enums.py,sha256=Zo8_tJzuzZ2IKfVc7gZ-0tWPT8q1QhqZbAyoh9ZVJbs,381
-django_bulk_hooks/factory.py,sha256=ezrVM5U023KZqOBbJXb6lYUP-pE7WJmi8Olh2Ew-7RA,18085
-django_bulk_hooks/handler.py,sha256=38ejMdQ9reYA07_XQ9tC8xv0lW3amO-m8gPzuRNOyj0,4200
-django_bulk_hooks/helpers.py,sha256=Nw8eXryLUUquW7AgiuKp0PQT3Pq6HAHsdP-xAtqhmjA,3216
-django_bulk_hooks/manager.py,sha256=3mFzB0ZzHHeXWdKGObZD_H0NlskHJc8uYBF69KKdAXU,4068
-django_bulk_hooks/models.py,sha256=4Vvi2LiGP0g4j08a5liqBROfsO8Wd_ermBoyjKwfrPU,2512
-django_bulk_hooks/operations/__init__.py,sha256=BtJYjmRhe_sScivLsniDaZmBkm0ZLvcmzXFKL7QY2Xg,550
-django_bulk_hooks/operations/analyzer.py,sha256=wAG8sAG9NwfwNqG9z81VfGR7AANDzRmMGE_o82MWji4,10689
-django_bulk_hooks/operations/bulk_executor.py,sha256=0_r9qaOMl9RJpi9eE3GrdUAlXA6ZU016nHYeIZ-yiPE,28279
-django_bulk_hooks/operations/coordinator.py,sha256=1Ka5eZJXTFjx3tr-BD6Tr350Y2T57SUOX3vjagBYBvM,32193
-django_bulk_hooks/operations/field_utils.py,sha256=Tvr5bcZLG8imH-r2S85oui1Cbw6hGv3VtuIMn4OvsU4,2895
-django_bulk_hooks/operations/mti_handler.py,sha256=pguym-6vqaLxu-lY7gnb5laKT9RJQlUciNw1SflZ1Zs,25574
-django_bulk_hooks/operations/mti_plans.py,sha256=7STQ2oA2ZT8cEG3-t-6xciRAdf7OeSf0gRLXR_BRG-Q,3363
-django_bulk_hooks/operations/record_classifier.py,sha256=kqML4aO11X9K3SSJ5DUlUukwI172j_Tk12Kr77ee8q8,7065
-django_bulk_hooks/queryset.py,sha256=aQitlbexcVnmeAdc0jtO3hci39p4QEu4srQPEzozy5s,5546
-django_bulk_hooks/registry.py,sha256=uum5jhGI3TPaoiXuA1MdBdu4gbE3rQGGwQ5YDjiMcjk,7949
-django_bulk_hooks-0.2.60.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
-django_bulk_hooks-0.2.60.dist-info/METADATA,sha256=tYgGJtWSG7Zx360peLawbsJAiIkT-BwF7WuJg4JfkJM,9265
-django_bulk_hooks-0.2.60.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-django_bulk_hooks-0.2.60.dist-info/RECORD,,