Fix INF-344 resourceType fields on alarmDictionary

[pti/o2.git] / o2ims / service / auditor / agg_storage_handler.py

diff --git a/o2ims/service/auditor/agg_storage_handler.py b/o2ims/service/auditor/agg_storage_handler.py
index 14ee152..6fddf19 100644 (file)
--- a/o2ims/service/auditor/agg_storage_handler.py
+++ b/o2ims/service/auditor/agg_storage_handler.py
@@ -42,7 +42,7 @@ def update_storage_aggregate(
     with uow:
         res = uow.session.execute(
             '''
-            SELECT "resourceTypeId", "oCloudId", "name"
+            SELECT "resourceTypeId", "name"
             FROM "resourceType"
             WHERE "resourceTypeEnum" = :resource_type_enum
             ''',
@@ -50,15 +50,21 @@ def update_storage_aggregate(
         )
         first = res.first()
         if first is None:
-            resourcepool = uow.resource_pools.get(cmd.parentid)
+            # resourcepool = uow.resource_pools.get(cmd.parentid)
             res_type_name = 'storage_aggregate'
             resourcetype_id = str(uuid.uuid3(
                 uuid.NAMESPACE_URL, res_type_name))
-            uow.resource_types.add(ResourceType(
+            res_type = ResourceType(
                 resourcetype_id,
                 res_type_name, stxobj.type,
-                resourcepool.oCloudId,
-                description='The storage Aggregate resource type'))
+                description='The storage Aggregate resource type')
+            dict_id = str(uuid.uuid3(
+                uuid.NAMESPACE_URL,
+                str(f"{res_type_name}_alarmdictionary")))
+            alarm_dictionary = uow.alarm_dictionaries.get(dict_id)
+            if alarm_dictionary:
+                res_type.alarmDictionary = alarm_dictionary
+            uow.resource_types.add(res_type)
         else:
             resourcetype_id = first['resourceTypeId']