From 7daf2c1d793e6ea083212395c695aca17abd8983 Mon Sep 17 00:00:00 2001 From: MrWaradana Date: Mon, 8 Dec 2025 15:06:49 +0700 Subject: [PATCH] add category_no changes in equipment master data --- src/equipment/service.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/equipment/service.py b/src/equipment/service.py index bbf512d..c2c6664 100644 --- a/src/equipment/service.py +++ b/src/equipment/service.py @@ -424,9 +424,10 @@ async def update( data = equipment_in.model_dump() update_data = equipment_in.model_dump(exclude_defaults=True) # Check if proportion from AcquisitionData changed and recalculate acquisition_cost - if "proportion" in update_data: + if "proportion" in update_data or "category_no" in update_data: + category_no = update_data.get("category_no", equipment.category_no) acquisition_data_query = Select(AcquisitionData).filter( - AcquisitionData.assetnum == equipment.assetnum + AcquisitionData.category_no == category_no ) acquisition_data_result = await db_session.execute(acquisition_data_query) acquisition_data = acquisition_data_result.scalars().one_or_none()