Update resilientCommunity.py

updated addEquipmentInfoToOmd code
dpinney · Apr 9, 2024 · 935d6cb · 935d6cb
1 parent 1fbf0c9
commit 935d6cb
Showing 1 changed file with 15 additions and 13 deletions.
diff --git a/omf/models/resilientCommunity.py b/omf/models/resilientCommunity.py
@@ -882,26 +882,28 @@ def addLoadInfoToOmd(loadsDict, omdDict):
     return omdDict
 
 
-def addEquipmentInfoToOmd(obDict, omdDict):
+def addEquipmentInfoToOmd(obDict, omdDict, equipList):
     '''
     adds criticality values to omd file for all objects
     loadsDict -> dict of loads
     omdDict -> dict of omd objects
     returns new dict of omd objects
     '''
     for ob in omdDict.get('tree', {}).values():
-        obType = ob['object']
-        obName = ob['name']
-        k = obType + '.' + obName
-
-        bcs_score = obDict[k]['base crit score']
-        ccs_score = obDict[k]['community crit score']
-        bcs_index = obDict[k]['base crit score']
-        ccs_index = obDict[k]['community crit index']
-        ob['base crit score'] = bcs_score
-        ob['community crit score'] = ccs_score
-        ob['community crit index'] = ccs_index
-        ob['base crit index'] = bcs_index
+        if (ob['object'] in equipList):
+            obType = ob['object']
+            obName = ob['name']
+            k = obType + '.' + obName
+            bcs_score = obDict[k]['base crit score']
+            ccs_score = obDict[k]['community crit score']
+            bcs_index = obDict[k]['base crit score']
+            ccs_index = obDict[k]['community crit index']
+            ob['base crit score'] = bcs_score
+            ob['community crit score'] = ccs_score
+            ob['community crit index'] = ccs_index
+            ob['base crit index'] = bcs_index
+        else:
+            continue
     return omdDict