move code

Author: CarolineDenis

specifyweb/specify/record_merging.py

@@ -192,18 +192,6 @@ def record_merge_fx(model_name: str, old_model_ids: List[int], new_model_id: int
         except ValueError:
             continue
 
-        # Handle case of updating a large amount of record ids in a foreign table.
-        # Example: handle case of updating a large amount of agent ids in the audit logs.
-        # Fix by optimizing the query by consolidating it here
-        if model_name.lower() in MERGING_OPTIMIZATION_FIELDS and \
-            table_name.lower() in MERGING_OPTIMIZATION_FIELDS[model_name.lower()]:
-            if field_name in MERGING_OPTIMIZATION_FIELDS[model_name.lower()][table_name.lower()]:
-                query = Q(**{field_name: old_model_ids[0]})
-                for old_model_id in old_model_ids[1:]:
-                    query.add(Q(**{field_name: old_model_id}), Q.OR)
-                foreign_model.objects.filter(query).update(**{field_name: new_model_id})
-                progress(1, 0) if progress is not None else None
-                continue
 
         apply_order = add_ordering_to_key(table_name.lower().title())
         # BUG: timestampmodified could be null for one record, and not the other
@@ -221,6 +209,18 @@ def record_merge_fx(model_name: str, old_model_ids: List[int], new_model_id: int
             field_name_id = f'{field_name}_id'
             if not hasattr(foreign_model, field_name_id):
                 continue
+            # Handle case of updating a large amount of record ids in a foreign table.
+            # Example: handle case of updating a large amount of agent ids in the audit logs.
+            # Fix by optimizing the query by consolidating it here
+            if model_name.lower() in MERGING_OPTIMIZATION_FIELDS and \
+                table_name.lower() in MERGING_OPTIMIZATION_FIELDS[model_name.lower()]:
+                if field_name_id in MERGING_OPTIMIZATION_FIELDS[model_name.lower()][table_name.lower()]:
+                    query = Q(**{field_name_id: old_model_ids[0]})
+                    for old_model_id in old_model_ids[1:]:
+                        query.add(Q(**{field_name_id: old_model_id}), Q.OR)
+                    foreign_model.objects.filter(query).update(**{field_name_id: new_model_id})
+                    progress(1, 0) if progress is not None else None
+                    continue
 
             # Filter the objects in the foreign model that references the old target model
             foreign_objects = filter_and_lock_target_objects(foreign_model, old_model_ids, field_name_id)