django-bulk-hooks 0.2.46__py3-none-any.whl → 0.2.47__py3-none-any.whl

django_bulk_hooks/operations/bulk_executor.py

@@ -80,7 +80,14 @@ class BulkExecutor:
                 existing_record_ids = set()
                 existing_pks_map = {}
                 if update_conflicts and unique_fields:
-                    existing_record_ids, existing_pks_map = self.record_classifier.classify_for_upsert(objs, unique_fields)
+                    # For MTI, find which model has the unique fields and query THAT model
+                    # This handles the schema migration case where parent exists but child doesn't
+                    query_model = self.mti_handler.find_model_with_unique_fields(unique_fields)
+                    logger.info(f"MTI upsert: querying {query_model.__name__} for unique fields {unique_fields}")
+                    
+                    existing_record_ids, existing_pks_map = self.record_classifier.classify_for_upsert(
+                        objs, unique_fields, query_model=query_model
+                    )
                     logger.info(f"MTI Upsert classification: {len(existing_record_ids)} existing, {len(objs) - len(existing_record_ids)} new")
                     logger.info(f"existing_record_ids: {existing_record_ids}")
                     logger.info(f"existing_pks_map: {existing_pks_map}")

django_bulk_hooks/operations/coordinator.py

@@ -136,7 +136,15 @@ class BulkOperationCoordinator:
         existing_record_ids = set()
         existing_pks_map = {}
         if update_conflicts and unique_fields:
-            existing_record_ids, existing_pks_map = self.record_classifier.classify_for_upsert(objs, unique_fields)
+            # For MTI models, query the parent model that has the unique fields
+            query_model = None
+            if self.mti_handler.is_mti_model():
+                query_model = self.mti_handler.find_model_with_unique_fields(unique_fields)
+                logger.info(f"MTI model detected: querying {query_model.__name__} for unique fields {unique_fields}")
+            
+            existing_record_ids, existing_pks_map = self.record_classifier.classify_for_upsert(
+                objs, unique_fields, query_model=query_model
+            )
             logger.info(f"Upsert operation: {len(existing_record_ids)} existing, {len(objs) - len(existing_record_ids)} new records")
             logger.debug(f"Existing record IDs: {existing_record_ids}")
             logger.debug(f"Existing PKs map: {existing_pks_map}")

django_bulk_hooks/operations/mti_handler.py

@@ -111,6 +111,35 @@ class MTIHandler:
         """
         return list(model_cls._meta.local_fields)
 
+    def find_model_with_unique_fields(self, unique_fields):
+        """
+        Find which model in the inheritance chain contains the unique fields.
+        
+        This is critical for MTI upserts: we need to query the model that has
+        the unique constraint, not necessarily the child model.
+        
+        Args:
+            unique_fields: List of field names forming the unique constraint
+        
+        Returns:
+            Model class that contains all the unique fields (closest to root)
+        """
+        if not unique_fields:
+            return self.model_cls
+        
+        inheritance_chain = self.get_inheritance_chain()
+        
+        # Start from root and find the first model that has all unique fields
+        for model_cls in inheritance_chain:
+            model_field_names = {f.name for f in model_cls._meta.local_fields}
+            
+            # Check if this model has all the unique fields
+            if all(field in model_field_names for field in unique_fields):
+                return model_cls
+        
+        # Fallback to child model (shouldn't happen if unique_fields are valid)
+        return self.model_cls
+
     # ==================== MTI BULK CREATE PLANNING ====================
 
     def build_create_plan(

django_bulk_hooks/operations/record_classifier.py

@@ -31,7 +31,7 @@ class RecordClassifier:
         """
         self.model_cls = model_cls
 
-    def classify_for_upsert(self, objs, unique_fields):
+    def classify_for_upsert(self, objs, unique_fields, query_model=None):
         """
         Classify records as new or existing based on unique_fields.
 
@@ -41,6 +41,7 @@ class RecordClassifier:
         Args:
             objs: List of model instances
             unique_fields: List of field names that form the unique constraint
+            query_model: Optional model class to query (for MTI, may be different from self.model_cls)
 
         Returns:
             Tuple of (existing_record_ids, existing_pks_map)
@@ -50,6 +51,9 @@ class RecordClassifier:
         if not unique_fields or not objs:
             return set(), {}
 
+        # Use query_model if provided (for MTI scenarios), otherwise use self.model_cls
+        query_model = query_model or self.model_cls
+
         # Build a query to find existing records
         queries = []
         obj_to_unique_values = {}
@@ -77,10 +81,10 @@ class RecordClassifier:
         for q in queries[1:]:
             combined_query |= q
 
-        logger.info(f"Classifying for upsert: model={self.model_cls.__name__}, query={combined_query}, unique_fields={unique_fields}")
-        queryset = self.model_cls.objects.filter(combined_query)
+        logger.info(f"Classifying for upsert: model={query_model.__name__}, query={combined_query}, unique_fields={unique_fields}")
+        queryset = query_model.objects.filter(combined_query)
         logger.info(f"Queryset SQL: {queryset.query}")
-        logger.info(f"All records in table: {self.model_cls.objects.all().count()}")
+        logger.info(f"All records in table: {query_model.objects.all().count()}")
         existing_records = list(queryset.values("pk", *unique_fields))
         logger.info(f"Found {len(existing_records)} existing records: {existing_records}")
 

{django_bulk_hooks-0.2.46.dist-info → django_bulk_hooks-0.2.47.dist-info}/METADATA

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

@@ -13,14 +13,14 @@ django_bulk_hooks/manager.py,sha256=3mFzB0ZzHHeXWdKGObZD_H0NlskHJc8uYBF69KKdAXU,
 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=4gCvG1Co3rqUR4TmlA4af-McHBLvOk3ekNAEguPsIBA,22678
-django_bulk_hooks/operations/coordinator.py,sha256=6qalGxbKqgglYwwYXVS49T8pd15uO95qexcTP_V2buk,30354
-django_bulk_hooks/operations/mti_handler.py,sha256=qx1xuOzxT3ibz2WJcfPsKxOiG-HU-axBRSmz76neD_w,19561
+django_bulk_hooks/operations/bulk_executor.py,sha256=nS8oRxMTPesJnwsqR4KdbJU6Sj0iD7ZWU2z9RJgtxao,23170
+django_bulk_hooks/operations/coordinator.py,sha256=iGavJLqe3eYRqFay8cMn6muwyRYzQo-HFGphsS5hL6g,30799
+django_bulk_hooks/operations/mti_handler.py,sha256=GIpTzzoo_hx1iI5osx2COk3cVQgqTsD-o41d9S7aiqo,20748
 django_bulk_hooks/operations/mti_plans.py,sha256=7STQ2oA2ZT8cEG3-t-6xciRAdf7OeSf0gRLXR_BRG-Q,3363
-django_bulk_hooks/operations/record_classifier.py,sha256=KiMmTjEZ3QdkSImCJnG5LBu2uSmcuIYQP4rSIzofdvQ,6437
+django_bulk_hooks/operations/record_classifier.py,sha256=vNi0WSNiPAVb8pTZZJ26b81oX59snk7LIxWMzyenDCk,6694
 django_bulk_hooks/queryset.py,sha256=aQitlbexcVnmeAdc0jtO3hci39p4QEu4srQPEzozy5s,5546
 django_bulk_hooks/registry.py,sha256=uum5jhGI3TPaoiXuA1MdBdu4gbE3rQGGwQ5YDjiMcjk,7949
-django_bulk_hooks-0.2.46.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
-django_bulk_hooks-0.2.46.dist-info/METADATA,sha256=mvL_yi_iUtU2JrP7pZ8lDfVUFAGBBK_dmQ0-VjhhYZE,9265
-django_bulk_hooks-0.2.46.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-django_bulk_hooks-0.2.46.dist-info/RECORD,,
+django_bulk_hooks-0.2.47.dist-info/LICENSE,sha256=dguKIcbDGeZD-vXWdLyErPUALYOvtX_fO4Zjhq481uk,1088
+django_bulk_hooks-0.2.47.dist-info/METADATA,sha256=Dp_OKvGqoa7j0zax8Cc-j5uH-acwsaCd1o0y-nw30U0,9265
+django_bulk_hooks-0.2.47.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+django_bulk_hooks-0.2.47.dist-info/RECORD,,