django-bulk-hooks 0.1.103__py3-none-any.whl → 0.1.105__py3-none-any.whl

django_bulk_hooks/manager.py

@@ -22,6 +22,143 @@ class BulkHookManager(models.Manager):
     def get_queryset(self):
         return HookQuerySet(self.model, using=self._db)
 
+    def _has_multi_table_inheritance(self, model_cls):
+        """
+        Check if this model uses multi-table inheritance.
+        """
+        return (
+            model_cls._meta.parents and 
+            not all(parent._meta.abstract for parent in model_cls._meta.parents.values())
+        )
+    
+    def _get_base_model(self, model_cls):
+        """
+        Get the base model (first non-abstract parent or self).
+        """
+        base_model = model_cls
+        while base_model._meta.parents:
+            parent = next(iter(base_model._meta.parents.values()))
+            if not parent._meta.abstract:
+                base_model = parent
+            else:
+                break
+        return base_model
+    
+    def _extract_base_objects(self, objs, model_cls):
+        """
+        Extract base model objects from inherited objects.
+        """
+        base_model = self._get_base_model(model_cls)
+        base_objects = []
+        
+        for obj in objs:
+            base_obj = base_model()
+            for field in base_model._meta.fields:
+                # Skip ID field
+                if field.name == 'id':
+                    continue
+                
+                # Safely copy field values
+                try:
+                    if hasattr(obj, field.name):
+                        setattr(base_obj, field.name, getattr(obj, field.name))
+                except (AttributeError, ValueError):
+                    # Skip fields that can't be copied
+                    continue
+            
+            base_objects.append(base_obj)
+        
+        return base_objects
+    
+    def _extract_child_objects(self, objs, model_cls):
+        """
+        Extract child model objects from inherited objects.
+        """
+        child_objects = []
+        
+        for obj in objs:
+            child_obj = model_cls()
+            child_obj.pk = obj.pk  # Set the same PK as base
+            
+            # Copy only fields specific to this model
+            for field in model_cls._meta.fields:
+                # Skip ID field and fields that don't belong to this model
+                if field.name == 'id':
+                    continue
+                
+                # Check if this field belongs to the current model
+                # Use a safer way to check field ownership
+                try:
+                    if hasattr(field, 'model') and field.model == model_cls:
+                        # This field belongs to the current model
+                        if hasattr(obj, field.name):
+                            setattr(child_obj, field.name, getattr(obj, field.name))
+                except AttributeError:
+                    # Skip fields that don't have proper model reference
+                    continue
+            
+            child_objects.append(child_obj)
+        
+        return child_objects
+    
+    def _bulk_create_inherited(self, objs, **kwargs):
+        """
+        Handle bulk create for inherited models by handling each table separately.
+        """
+        if not objs:
+            return []
+        
+        model_cls = self.model
+        result = []
+        
+        # Group objects by their actual class
+        objects_by_class = {}
+        for obj in objs:
+            obj_class = obj.__class__
+            if obj_class not in objects_by_class:
+                objects_by_class[obj_class] = []
+            objects_by_class[obj_class].append(obj)
+        
+        for obj_class, class_objects in objects_by_class.items():
+            try:
+                # Check if this class has multi-table inheritance
+                parent_models = [p for p in obj_class._meta.get_parent_list() 
+                               if not p._meta.abstract]
+                
+                if not parent_models:
+                    # No inheritance, use standard bulk_create
+                    chunk_result = super(models.Manager, self).bulk_create(class_objects, **kwargs)
+                    result.extend(chunk_result)
+                    continue
+                
+                # Handle multi-table inheritance
+                # Step 1: Bulk create base objects without hooks
+                base_objects = self._extract_base_objects(class_objects, obj_class)
+                created_base = super(models.Manager, self).bulk_create(base_objects, **kwargs)
+                
+                # Step 2: Update original objects with base IDs
+                for obj, base_obj in zip(class_objects, created_base):
+                    obj.pk = base_obj.pk
+                    obj._state.adding = False
+                
+                # Step 3: Bulk create child objects without hooks
+                child_objects = self._extract_child_objects(class_objects, obj_class)
+                if child_objects:
+                    # Use _base_manager to avoid recursion
+                    obj_class._base_manager.bulk_create(child_objects, **kwargs)
+                
+                result.extend(class_objects)
+                
+            except Exception as e:
+                # Add debugging information
+                import logging
+                logger = logging.getLogger(__name__)
+                logger.error(f"Error in _bulk_create_inherited for {obj_class}: {e}")
+                logger.error(f"Model fields: {[f.name for f in obj_class._meta.fields]}")
+                raise
+        
+        return result
+
     @transaction.atomic
     def bulk_update(
         self, objs, fields, bypass_hooks=False, bypass_validation=False, **kwargs
@@ -116,6 +253,9 @@ class BulkHookManager(models.Manager):
                 f"bulk_create expected instances of {model_cls.__name__}, but got {set(type(obj).__name__ for obj in objs)}"
             )
 
+        # Check if this model uses multi-table inheritance
+        has_multi_table_inheritance = self._has_multi_table_inheritance(model_cls)
+
         result = []
 
         if not bypass_hooks:
@@ -128,9 +268,18 @@ class BulkHookManager(models.Manager):
             # Then run business logic hooks
             engine.run(model_cls, BEFORE_CREATE, objs, ctx=ctx)
 
+        # Perform bulk create in chunks
         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, **kwargs))
+            
+            if has_multi_table_inheritance:
+                # Use our multi-table bulk create
+                created_chunk = self._bulk_create_inherited(chunk, **kwargs)
+            else:
+                # Use Django's standard bulk create
+                created_chunk = super(models.Manager, self).bulk_create(chunk, **kwargs)
+            
+            result.extend(created_chunk)
 
         if not bypass_hooks:
             engine.run(model_cls, AFTER_CREATE, result, ctx=ctx)

{django_bulk_hooks-0.1.103.dist-info → django_bulk_hooks-0.1.105.dist-info}/METADATA

@@ -1,8 +1,7 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: django-bulk-hooks
-Version: 0.1.103
+Version: 0.1.105
 Summary: Hook-style hooks for Django bulk operations like bulk_create and bulk_update.
-Home-page: https://github.com/AugendLimited/django-bulk-hooks
 License: MIT
 Keywords: django,bulk,hooks
 Author: Konrad Beck
@@ -14,6 +13,7 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: Django (>=4.0)
+Project-URL: Homepage, https://github.com/AugendLimited/django-bulk-hooks
 Project-URL: Repository, https://github.com/AugendLimited/django-bulk-hooks
 Description-Content-Type: text/markdown
 

{django_bulk_hooks-0.1.103.dist-info → django_bulk_hooks-0.1.105.dist-info}/RECORD

@@ -6,12 +6,12 @@ django_bulk_hooks/decorators.py,sha256=tckDcxtOzKCbgvS9QydgeIAWTFDEl-ch3_Q--ruEG
 django_bulk_hooks/engine.py,sha256=3HbgV12JRYIy9IlygHPxZiHnFXj7EwzLyTuJNQeVIoI,1402
 django_bulk_hooks/enums.py,sha256=Zo8_tJzuzZ2IKfVc7gZ-0tWPT8q1QhqZbAyoh9ZVJbs,381
 django_bulk_hooks/handler.py,sha256=xZt8iNdYF-ACz-MnKMY0co6scWINU5V5wC1lyDn844k,4854
-django_bulk_hooks/manager.py,sha256=JkhOw9XbEUt8VpunhUSP0z-NlazIFUWmAesj7H4ge8w,7148
+django_bulk_hooks/manager.py,sha256=Q9Oss6kWCgyL-08t3XYplLSQSLILMAX7Qv9eg7-VgKA,13147
 django_bulk_hooks/models.py,sha256=7RG7GrOdHXFjGVPV4FPRZVNMIHHW-hMCi6hn9LH_hVI,3331
 django_bulk_hooks/priority.py,sha256=HG_2D35nga68lBCZmSXTcplXrjFoRgZFRDOy4ROKonY,376
 django_bulk_hooks/queryset.py,sha256=iet4z-9SKhnresA4FBQbxx9rdYnoaOWbw9LUlGftlP0,1466
 django_bulk_hooks/registry.py,sha256=-mQBizJ06nz_tajZBinViKx_uP2Tbc1tIpTEMv7lwKA,705
-django_bulk_hooks-0.1.103.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
-django_bulk_hooks-0.1.103.dist-info/METADATA,sha256=Y_okaTk73sz_FCHWBOczBT8tzRz-wQJmNzmofLWy3MI,6939
-django_bulk_hooks-0.1.103.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-django_bulk_hooks-0.1.103.dist-info/RECORD,,
+django_bulk_hooks-0.1.105.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
+django_bulk_hooks-0.1.105.dist-info/METADATA,sha256=PVNvKCg4CciKQTGbe7p4L38tsF9kuum22H7zaPu8oNs,6951
+django_bulk_hooks-0.1.105.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+django_bulk_hooks-0.1.105.dist-info/RECORD,,

{django_bulk_hooks-0.1.103.dist-info → django_bulk_hooks-0.1.105.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.9.1
+Generator: poetry-core 2.1.3
 Root-Is-Purelib: true
 Tag: py3-none-any