diff --git a/src/calculation_time_constrains/schema.py b/src/calculation_time_constrains/schema.py index 6650fed..3da97b6 100644 --- a/src/calculation_time_constrains/schema.py +++ b/src/calculation_time_constrains/schema.py @@ -94,4 +94,4 @@ class CalculationTimeConstrainsSimulationRead(CalculationTimeConstrainsBase): class CalculationSelectedEquipmentUpdate(CalculationTimeConstrainsBase): is_included: bool - assetnum: str + location_tag: str diff --git a/src/calculation_time_constrains/service.py b/src/calculation_time_constrains/service.py index f828a86..4297143 100644 --- a/src/calculation_time_constrains/service.py +++ b/src/calculation_time_constrains/service.py @@ -2091,15 +2091,15 @@ async def bulk_update_equipment( calculation_data_id: UUID, ): # Create a dictionary mapping assetnum to is_included status - case_mappings = {asset.assetnum: asset.is_included for asset in selected_equipments} + case_mappings = {asset.location_tag: asset.is_included for asset in selected_equipments} # Get all assetnums that need to be updated assetnums = list(case_mappings.keys()) # Create a list of when clauses for the case statement when_clauses = [ - (CalculationEquipmentResult.assetnum == assetnum, is_included) - for assetnum, is_included in case_mappings.items() + (CalculationEquipmentResult.location_tag == location_tag, is_included) + for location_tag, is_included in case_mappings.items() ] # Build the update statement