From 5a626c3d67a8dc742f78aa20684d9ed2041f1bd2 Mon Sep 17 00:00:00 2001
From: Cizz22 <cisatraa@gmail.com>
Date: Thu, 27 Nov 2025 11:08:45 +0700
Subject: [PATCH] fix custom

---
 src/aeros_equipment/service.py | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/src/aeros_equipment/service.py b/src/aeros_equipment/service.py
index a867677..3100694 100644
--- a/src/aeros_equipment/service.py
+++ b/src/aeros_equipment/service.py
@@ -412,11 +412,22 @@ async def update_equipment_for_simulation(*, db_session: DbSession,aeros_db_sess
                         continue
                     
                     # Check eq dengan "TRIP" di WO, jika ada masukkan parameter Reliabilitu, jika tidak ada MTBF = duration
+                    MTBF = 1e+6 / float(custom_param['failure_rate'])
+                    
                         
                     eq["cmDisP1"] = custom_param["mttr"]
-                    eq["relDisType"] = "Fixed"
-                    eq["relDisP1"] = float(custom_param["failure_rate"])
-                    eq["relDisP2"] = 0
+                    eq["relDisType"] = custom_param["distribution"] if "distribution" in custom_param else "Fixed"
+                    # eq["relDisP1"] = float(custom_param["failure_rate"])
+                    # eq["relDisP2"] = 0
+                    
+                    if eq["relDisType"] == "Fixed":
+                        eq["relDisP1"] = MTBF
+                        eq["relDisP2"] = 0
+                    else:
+                        eq["relDisP1"] = 1
+                        eq["relDisP2"] = MTBF
+                    
+                    
                     eq["ohDisP1"] = overhaul_duration
                     eq["ohDisUnitCode"] = "UHour"