django-bulk-hooks 0.1.60__tar.gz → 0.1.62__tar.gz

{django_bulk_hooks-0.1.60 → django_bulk_hooks-0.1.62}/PKG-INFO

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

{django_bulk_hooks-0.1.60 → django_bulk_hooks-0.1.62}/django_bulk_hooks/decorators.py

@@ -4,7 +4,7 @@ from functools import wraps
 from django.core.exceptions import FieldDoesNotExist
 from django_bulk_hooks.enums import DEFAULT_PRIORITY
 from django_bulk_hooks.constants import BEFORE_UPDATE
-from django_bulk_hooks.handler import TriggerHandler
+from django_bulk_hooks.registry import register_hook
 
 
 def hook(event, *, model, condition=None, priority=DEFAULT_PRIORITY):
@@ -116,7 +116,22 @@ def bulk_hook(model_cls, event, when=None, priority=None):
         priority: Optional priority for hook execution order
     """
     def decorator(func):
-        # Register the hook
-        TriggerHandler.register_hook(model_cls, event, func, when, priority)
+        # Create a simple handler class for the function
+        class FunctionHandler:
+            def __init__(self):
+                self.func = func
+            
+            def handle(self, new_instances, original_instances):
+                return self.func(new_instances, original_instances)
+        
+        # Register the hook using the registry
+        register_hook(
+            model=model_cls,
+            event=event,
+            handler_cls=FunctionHandler,
+            method_name='handle',
+            condition=when,
+            priority=priority or DEFAULT_PRIORITY,
+        )
         return func
     return decorator

{django_bulk_hooks-0.1.60 → django_bulk_hooks-0.1.62}/django_bulk_hooks/manager.py

@@ -22,7 +22,7 @@ class BulkLifecycleManager(models.Manager):
         return LifecycleQuerySet(self.model, using=self._db)
 
     @transaction.atomic
-    def bulk_update(self, objs, fields, batch_size=None, bypass_hooks=False):
+    def bulk_update(self, objs, fields, bypass_hooks=False, **kwargs):
         if not objs:
             return []
 
@@ -53,9 +53,7 @@ class BulkLifecycleManager(models.Manager):
         for i in range(0, len(objs), self.CHUNK_SIZE):
             chunk = objs[i : i + self.CHUNK_SIZE]
             # Call the base implementation to avoid re-triggering this method
-            super(models.Manager, self).bulk_update(
-                chunk, fields, batch_size=batch_size
-            )
+            super(models.Manager, self).bulk_update(chunk, fields, **kwargs)
 
         if not bypass_hooks:
             engine.run(model_cls, AFTER_UPDATE, objs, originals, ctx=ctx)
@@ -106,9 +104,7 @@ class BulkLifecycleManager(models.Manager):
         return modified_fields
 
     @transaction.atomic
-    def bulk_create(
-        self, objs, batch_size=None, ignore_conflicts=False, bypass_hooks=False
-    ):
+    def bulk_create(self, objs, bypass_hooks=False, **kwargs):
         model_cls = self.model
 
         if any(not isinstance(obj, model_cls) for obj in objs):
@@ -125,9 +121,7 @@ class BulkLifecycleManager(models.Manager):
         for i in range(0, len(objs), self.CHUNK_SIZE):
             chunk = objs[i : i + self.CHUNK_SIZE]
             result.extend(
-                super(models.Manager, self).bulk_create(
-                    chunk, batch_size=batch_size, ignore_conflicts=ignore_conflicts
-                )
+                super(models.Manager, self).bulk_create(chunk, **kwargs)
             )
 
         if not bypass_hooks:

{django_bulk_hooks-0.1.60 → django_bulk_hooks-0.1.62}/pyproject.toml

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