django-bulk-hooks 0.2.16__py3-none-any.whl → 0.2.17__py3-none-any.whl

django_bulk_hooks/operations/mti_plans.py

@@ -5,8 +5,9 @@ These are pure data structures returned by MTIHandler to be executed by BulkExec
 This separates planning (logic) from execution (database operations).
 """
 
-from dataclasses import dataclass, field
-from typing import Dict, List, Any
+from dataclasses import dataclass
+from dataclasses import field
+from typing import Any
 
 
 @dataclass
@@ -23,11 +24,11 @@ class ParentLevel:
         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)
+    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)
+    unique_fields: list[str] = field(default_factory=list)
+    update_fields: list[str] = field(default_factory=list)
 
 
 @dataclass
@@ -50,16 +51,16 @@ class MTICreatePlan:
         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]
+    inheritance_chain: list[Any]
+    parent_levels: list[ParentLevel]
+    child_objects: list[Any]
     child_model: Any
-    original_objects: List[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)
+    unique_fields: list[str] = field(default_factory=list)
+    update_fields: list[str] = field(default_factory=list)
 
 
 @dataclass
@@ -73,7 +74,7 @@ class ModelFieldGroup:
         filter_field: Field to use for filtering (e.g., 'pk' or parent link attname)
     """
     model_class: Any
-    fields: List[str]
+    fields: list[str]
     filter_field: str = "pk"
 
 
@@ -88,8 +89,8 @@ class MTIUpdatePlan:
         objects: Objects to update
         batch_size: Batch size for operations
     """
-    inheritance_chain: List[Any]
-    field_groups: List[ModelFieldGroup]
-    objects: List[Any]
+    inheritance_chain: list[Any]
+    field_groups: list[ModelFieldGroup]
+    objects: list[Any]
     batch_size: int = None
 

django_bulk_hooks/operations/record_classifier.py

@@ -8,6 +8,7 @@ Separates data access concerns from business logic.
 """
 
 import logging
+
 from django.db.models import Q
 
 logger = logging.getLogger(__name__)
@@ -48,11 +49,11 @@ class RecordClassifier:
         """
         if not unique_fields or not objs:
             return set(), {}
-        
+
         # Build a query to find existing records
         queries = []
         obj_to_unique_values = {}
-        
+
         for obj in objs:
             # Build lookup dict for this object's unique fields
             lookup = {}
@@ -67,36 +68,36 @@ class RecordClassifier:
                 if lookup:
                     queries.append(Q(**lookup))
                     obj_to_unique_values[id(obj)] = tuple(lookup.values())
-        
+
         if not queries:
             return set(), {}
-        
+
         # Query for existing records
         combined_query = queries[0]
         for q in queries[1:]:
             combined_query |= q
-        
+
         existing_records = list(
-            self.model_cls.objects.filter(combined_query).values('pk', *unique_fields)
+            self.model_cls.objects.filter(combined_query).values("pk", *unique_fields),
         )
-        
+
         # Map existing records back to original objects
         existing_record_ids = set()
         existing_pks_map = {}
-        
+
         for record in existing_records:
             record_values = tuple(record[field] for field in unique_fields)
             # Find which object(s) match these values
             for obj_id, obj_values in obj_to_unique_values.items():
                 if obj_values == record_values:
                     existing_record_ids.add(obj_id)
-                    existing_pks_map[obj_id] = record['pk']
-        
+                    existing_pks_map[obj_id] = record["pk"]
+
         logger.info(
             f"Classified {len(existing_record_ids)} existing and "
-            f"{len(objs) - len(existing_record_ids)} new records for upsert"
+            f"{len(objs) - len(existing_record_ids)} new records for upsert",
         )
-        
+
         return existing_record_ids, existing_pks_map
 
     def fetch_by_pks(self, pks, select_related=None, prefetch_related=None):
@@ -113,15 +114,15 @@ class RecordClassifier:
         """
         if not pks:
             return {}
-        
+
         queryset = self.model_cls._base_manager.filter(pk__in=pks)
-        
+
         if select_related:
             queryset = queryset.select_related(*select_related)
-        
+
         if prefetch_related:
             queryset = queryset.prefetch_related(*prefetch_related)
-        
+
         return {obj.pk: obj for obj in queryset}
 
     def fetch_by_unique_constraint(self, field_values_map):
@@ -141,7 +142,7 @@ class RecordClassifier:
         except self.model_cls.MultipleObjectsReturned:
             logger.warning(
                 f"Multiple {self.model_cls.__name__} records found for "
-                f"unique constraint {field_values_map}"
+                f"unique constraint {field_values_map}",
             )
             return self.model_cls.objects.filter(**field_values_map).first()
 
@@ -157,11 +158,11 @@ class RecordClassifier:
         """
         if not pks:
             return set()
-        
+
         existing_pks = self.model_cls.objects.filter(
-            pk__in=pks
-        ).values_list('pk', flat=True)
-        
+            pk__in=pks,
+        ).values_list("pk", flat=True)
+
         return set(existing_pks)
 
     def count_by_unique_fields(self, objs, unique_fields):

django_bulk_hooks/queryset.py

@@ -7,7 +7,9 @@ complex coordination required for bulk operations with hooks.
 """
 
 import logging
-from django.db import models, transaction
+
+from django.db import models
+from django.db import transaction
 
 logger = logging.getLogger(__name__)
 
@@ -98,7 +100,7 @@ class HookQuerySet(models.QuerySet):
             fields = self.coordinator.analyzer.detect_changed_fields(objs)
             if not fields:
                 logger.debug(
-                    f"bulk_update: No fields changed for {len(objs)} {self.model.__name__} objects"
+                    f"bulk_update: No fields changed for {len(objs)} {self.model.__name__} objects",
                 )
                 return 0
 
@@ -133,7 +135,7 @@ 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.

django_bulk_hooks/registry.py

@@ -8,14 +8,13 @@ deterministic priority ordering.
 import logging
 import threading
 from collections.abc import Callable
-from typing import Dict, List, Optional, Tuple, Type, Union
 
 from django_bulk_hooks.enums import Priority
 
 logger = logging.getLogger(__name__)
 
 # Type alias for hook info tuple
-HookInfo = Tuple[Type, str, Optional[Callable], int]
+HookInfo = tuple[type, str, Callable | None, int]
 
 
 class HookRegistry:
@@ -30,17 +29,17 @@ class HookRegistry:
 
     def __init__(self):
         """Initialize an empty registry with thread-safe storage."""
-        self._hooks: Dict[Tuple[Type, str], List[HookInfo]] = {}
+        self._hooks: dict[tuple[type, str], list[HookInfo]] = {}
         self._lock = threading.RLock()
 
     def register(
         self,
-        model: Type,
+        model: type,
         event: str,
-        handler_cls: Type,
+        handler_cls: type,
         method_name: str,
-        condition: Optional[Callable],
-        priority: Union[int, Priority],
+        condition: Callable | None,
+        priority: int | Priority,
     ) -> None:
         """
         Register a hook handler for a model and event.
@@ -64,16 +63,14 @@ class HookRegistry:
                 # Sort by priority (lower values first)
                 hooks.sort(key=lambda x: x[3])
                 logger.debug(
-                    f"Registered {handler_cls.__name__}.{method_name} "
-                    f"for {model.__name__}.{event} (priority={priority})"
+                    f"Registered {handler_cls.__name__}.{method_name} for {model.__name__}.{event} (priority={priority})",
                 )
             else:
                 logger.debug(
-                    f"Hook {handler_cls.__name__}.{method_name} "
-                    f"already registered for {model.__name__}.{event}"
+                    f"Hook {handler_cls.__name__}.{method_name} already registered for {model.__name__}.{event}",
                 )
 
-    def get_hooks(self, model: Type, event: str) -> List[HookInfo]:
+    def get_hooks(self, model: type, event: str) -> list[HookInfo]:
         """
         Get all hooks for a model and event.
 
@@ -97,13 +94,17 @@ class HookRegistry:
                 "before_create",
             ]:
                 logger.debug(
-                    f"get_hooks {model.__name__}.{event} found {len(hooks)} hooks"
+                    f"get_hooks {model.__name__}.{event} found {len(hooks)} hooks",
                 )
 
             return hooks
 
     def unregister(
-        self, model: Type, event: str, handler_cls: Type, method_name: str
+        self,
+        model: type,
+        event: str,
+        handler_cls: type,
+        method_name: str,
     ) -> None:
         """
         Unregister a specific hook handler.
@@ -124,9 +125,7 @@ class HookRegistry:
             hooks = self._hooks[key]
             # Filter out the specific hook
             self._hooks[key] = [
-                (h_cls, m_name, cond, pri)
-                for h_cls, m_name, cond, pri in hooks
-                if not (h_cls == handler_cls and m_name == method_name)
+                (h_cls, m_name, cond, pri) for h_cls, m_name, cond, pri in hooks if not (h_cls == handler_cls and m_name == method_name)
             ]
 
             # Clean up empty hook lists
@@ -134,8 +133,7 @@ class HookRegistry:
                 del self._hooks[key]
 
             logger.debug(
-                f"Unregistered {handler_cls.__name__}.{method_name} "
-                f"for {model.__name__}.{event}"
+                f"Unregistered {handler_cls.__name__}.{method_name} for {model.__name__}.{event}",
             )
 
     def clear(self) -> None:
@@ -155,7 +153,7 @@ class HookRegistry:
 
             logger.debug("Cleared all registered hooks")
 
-    def list_all(self) -> Dict[Tuple[Type, str], List[HookInfo]]:
+    def list_all(self) -> dict[tuple[type, str], list[HookInfo]]:
         """
         Get all registered hooks for debugging.
 
@@ -166,17 +164,19 @@ class HookRegistry:
             return dict(self._hooks)
 
     @property
-    def hooks(self) -> Dict[Tuple[Type, str], List[HookInfo]]:
+    def hooks(self) -> dict[tuple[type, str], list[HookInfo]]:
         """
         Expose internal hooks dictionary for testing purposes.
-        
+
         This property provides direct access to the internal hooks storage
         to allow tests to clear the registry state between test runs.
         """
         return self._hooks
 
     def count_hooks(
-        self, model: Optional[Type] = None, event: Optional[str] = None
+        self,
+        model: type | None = None,
+        event: str | None = None,
     ) -> int:
         """
         Count registered hooks, optionally filtered by model and/or event.
@@ -192,27 +192,19 @@ class HookRegistry:
             if model is None and event is None:
                 # Count all hooks
                 return sum(len(hooks) for hooks in self._hooks.values())
-            elif model is not None and event is not None:
+            if model is not None and event is not None:
                 # Count hooks for specific model and event
                 return len(self._hooks.get((model, event), []))
-            elif model is not None:
+            if model is not None:
                 # Count all hooks for a model
-                return sum(
-                    len(hooks)
-                    for (m, _), hooks in self._hooks.items()
-                    if m == model
-                )
-            else:  # event is not None
-                # Count all hooks for an event
-                return sum(
-                    len(hooks)
-                    for (_, e), hooks in self._hooks.items()
-                    if e == event
-                )
+                return sum(len(hooks) for (m, _), hooks in self._hooks.items() if m == model)
+            # event is not None
+            # Count all hooks for an event
+            return sum(len(hooks) for (_, e), hooks in self._hooks.items() if e == event)
 
 
 # Global singleton registry
-_registry: Optional[HookRegistry] = None
+_registry: HookRegistry | None = None
 _registry_lock = threading.Lock()
 
 
@@ -239,12 +231,12 @@ def get_registry() -> HookRegistry:
 
 # Backward-compatible module-level functions
 def register_hook(
-    model: Type,
+    model: type,
     event: str,
-    handler_cls: Type,
+    handler_cls: type,
     method_name: str,
-    condition: Optional[Callable],
-    priority: Union[int, Priority],
+    condition: Callable | None,
+    priority: int | Priority,
 ) -> None:
     """
     Register a hook handler (backward-compatible function).
@@ -255,7 +247,7 @@ def register_hook(
     registry.register(model, event, handler_cls, method_name, condition, priority)
 
 
-def get_hooks(model: Type, event: str) -> List[HookInfo]:
+def get_hooks(model: type, event: str) -> list[HookInfo]:
     """
     Get hooks for a model and event (backward-compatible function).
 
@@ -266,7 +258,10 @@ def get_hooks(model: Type, event: str) -> List[HookInfo]:
 
 
 def unregister_hook(
-    model: Type, event: str, handler_cls: Type, method_name: str
+    model: type,
+    event: str,
+    handler_cls: type,
+    method_name: str,
 ) -> None:
     """
     Unregister a hook handler (backward-compatible function).
@@ -288,7 +283,7 @@ def clear_hooks() -> None:
     registry.clear()
 
 
-def list_all_hooks() -> Dict[Tuple[Type, str], List[HookInfo]]:
+def list_all_hooks() -> dict[tuple[type, str], list[HookInfo]]:
     """
     List all registered hooks (backward-compatible function).
 

{django_bulk_hooks-0.2.16.dist-info → django_bulk_hooks-0.2.17.dist-info}/METADATA

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

@@ -0,0 +1,26 @@
+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=fOWh1gtukdegMgku7qdc-Hix6G6ZCKMetONFiNKIBWI,8842
+django_bulk_hooks/enums.py,sha256=Zo8_tJzuzZ2IKfVc7gZ-0tWPT8q1QhqZbAyoh9ZVJbs,381
+django_bulk_hooks/factory.py,sha256=OgrOmLbIzhrSKTDx06oGMAVsEb0NoVOmW5IdLsMz_Qs,19938
+django_bulk_hooks/handler.py,sha256=i0M4mdx3vgXIb8mA1S5ZBW_8ezECd8yTZaj9QNUm8P8,4738
+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=8xNdsoeOCHsIIP7ztIuFHDewBR2dDSS6auOrSU8zmqE,9007
+django_bulk_hooks/operations/bulk_executor.py,sha256=QmfcEUJAVHaMPzYe4yDwQ0i-9FvxDaTWy17mcmn2-Yw,22371
+django_bulk_hooks/operations/coordinator.py,sha256=iSi-x3fq6_3UygWGeQY8Fq_gZWIqq_8N7WddPhVeygk,23526
+django_bulk_hooks/operations/mti_handler.py,sha256=G-pxkzIqHqXGshRGksqmsN1J3rlzePUZrSv4wm7D3cQ,19162
+django_bulk_hooks/operations/mti_plans.py,sha256=YP7LcV9Z8UqNS_x74OswF9_5swqruRTdAu6z-J_R6C0,3377
+django_bulk_hooks/operations/record_classifier.py,sha256=KzUoAhfoqzFVrOabNZAby9Akb54h-fAQZmb8O-fIx_0,6221
+django_bulk_hooks/queryset.py,sha256=rvJgQLwtSJztwc68nkJ6xwCsnbXQvkvS6_dbGGj8TFo,5886
+django_bulk_hooks/registry.py,sha256=QyeA2OqNdMAMaLjFU9UF0YGhKiPKbZkmFQpLgof7uNs,9038
+django_bulk_hooks-0.2.17.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
+django_bulk_hooks-0.2.17.dist-info/METADATA,sha256=ue7kA8E8s2Tha6v_EXhQ41oORwxZ1rTWMem14TNqLGg,9265
+django_bulk_hooks-0.2.17.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+django_bulk_hooks-0.2.17.dist-info/RECORD,,

django_bulk_hooks-0.2.16.dist-info/RECORD

@@ -1,26 +0,0 @@
-django_bulk_hooks/__init__.py,sha256=4QlWY5rqR9o2ddrNB-ypM4s1GtNJ1ZvL2ABhybaPpio,1823
-django_bulk_hooks/changeset.py,sha256=WALeiWDcjOBNdCKeidVKOPKAySKj9ZOvUJ-kWaVZYhM,7444
-django_bulk_hooks/conditions.py,sha256=qtGjToKXC8FPUPK31Mib-GMzc9GSdrH90M2pT3CIsh8,8111
-django_bulk_hooks/constants.py,sha256=PxpEETaO6gdENcTPoXS586lerGKVP3nmjpDvOkmhYxI,509
-django_bulk_hooks/context.py,sha256=mqaC5-yESDTA5ruI7fuXlt8qSgKuOFp0mjq7h1-4HdQ,1926
-django_bulk_hooks/decorators.py,sha256=uZhMGbFUxMz8fNnWBltB1D5Ez1d6vRNvoSFzqOTdGDo,12331
-django_bulk_hooks/dispatcher.py,sha256=ExEtApzwvprWIlbZifPUY17hJHeRUmeHe59BikDgs_g,8872
-django_bulk_hooks/enums.py,sha256=Zo8_tJzuzZ2IKfVc7gZ-0tWPT8q1QhqZbAyoh9ZVJbs,381
-django_bulk_hooks/factory.py,sha256=JmjQiJPfAnytXrO6r6qOadX5yX0-sfpbZ9V8nwX3MAg,20013
-django_bulk_hooks/handler.py,sha256=2-k0GPWGSQ6acfvV0qJgDH8aa0z51DqdpX5vSJ6Uawk,4759
-django_bulk_hooks/helpers.py,sha256=Yopvl588VbKOi2kHEsQcEcI5jw5jiNA2MuF6Ce1VP0c,3174
-django_bulk_hooks/manager.py,sha256=3mFzB0ZzHHeXWdKGObZD_H0NlskHJc8uYBF69KKdAXU,4068
-django_bulk_hooks/models.py,sha256=62tn5wL55EjJVOsZofMluhEJB8bH7CzBvH0vd214_RY,2570
-django_bulk_hooks/operations/__init__.py,sha256=5L5NnwiFw8Yn5WO6-38eGdCYBkA0URpwyDcAdeYfc5w,550
-django_bulk_hooks/operations/analyzer.py,sha256=s6FM53ho1raPdKU-VjjW0SWymXyrJe0I_Wu8XsXFdSY,9065
-django_bulk_hooks/operations/bulk_executor.py,sha256=MqfcwFNvAxpe1pfFfPEq70z8Ft1IxW1fmieDHeWbbbI,22767
-django_bulk_hooks/operations/coordinator.py,sha256=T9Y1Y5BAichETCG23mwC0eQ7VdZde-kUBKWKJVk3L1M,23717
-django_bulk_hooks/operations/mti_handler.py,sha256=y7tHDe2sPVg4bU0y4ITguMUdERwYXvfvjgSD-uvDBEM,19591
-django_bulk_hooks/operations/mti_plans.py,sha256=W2NjG0W9alY2ZxplzHgQv3TvWWmam-8pZuI0YU7HUqA,3365
-django_bulk_hooks/operations/record_classifier.py,sha256=ny_9Z0ATSX-X6I51ufuQUzgGKM0j3IcPs9SMOcgy5oc,6335
-django_bulk_hooks/queryset.py,sha256=ody4MXrRREL27Ts2ey1UpS0tb5Dxnw-6kN3unxPQ3zY,5860
-django_bulk_hooks/registry.py,sha256=fYNY4UzM4evNVozndeVeyD-yL1h0Lf8iykqJ3WqnMc8,9349
-django_bulk_hooks-0.2.16.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
-django_bulk_hooks-0.2.16.dist-info/METADATA,sha256=oDkpbXEOpjzhwtAlHboZDSSi_E-HrZ95D6_IoW85ufE,9265
-django_bulk_hooks-0.2.16.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-django_bulk_hooks-0.2.16.dist-info/RECORD,,