From 37637713f0a032c5ba18454d25f1e3ef76c5c0af Mon Sep 17 00:00:00 2001 From: Cizz22 Date: Fri, 25 Jul 2025 16:48:27 +0700 Subject: [PATCH] add powerplant reliability --- src/calculation_time_constrains/schema.py | 2 +- src/calculation_time_constrains/service.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) 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