Fix INF-344 resourceType fields on alarmDictionary
[pti/o2.git] / o2ims / service / auditor / pserver_dev_handler.py
1 # Copyright (C) 2021 Wind River Systems, Inc.
2 #
3 #  Licensed under the Apache License, Version 2.0 (the "License");
4 #  you may not use this file except in compliance with the License.
5 #  You may obtain a copy of the License at
6 #
7 #      http://www.apache.org/licenses/LICENSE-2.0
8 #
9 #  Unless required by applicable law or agreed to in writing, software
10 #  distributed under the License is distributed on an "AS IS" BASIS,
11 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 #  See the License for the specific language governing permissions and
13 #  limitations under the License.
14
15 # pylint: disable=unused-argument
16 from __future__ import annotations
17 import uuid
18 # import json
19
20 from o2ims.domain import commands, events
21 from o2ims.domain.stx_object import StxGenericModel
22 from o2ims.domain.subscription_obj import NotificationEventEnum
23 from o2common.service.unit_of_work import AbstractUnitOfWork
24 from o2ims.domain.resource_type import MismatchedModel
25 from o2ims.domain.ocloud import Resource, ResourceType
26
27 from o2common.helper import o2logging
28 logger = o2logging.get_logger(__name__)
29
30
31 class InvalidResourceType(Exception):
32     pass
33
34
35 def update_pserver_dev(
36     cmd: commands.UpdatePserverDev,
37     uow: AbstractUnitOfWork
38 ):
39     stxobj = cmd.data
40     with uow:
41         p_resource = uow.resources.get(cmd.parentid)
42         # resourcepool = uow.resource_pools.get(p_resource.resourcePoolId)
43
44         res = uow.session.execute(
45             '''
46             SELECT "resourceTypeId", "name"
47             FROM "resourceType"
48             WHERE "resourceTypeEnum" = :resource_type_enum
49             ''',
50             dict(resource_type_enum=stxobj.type.name)
51         )
52         first = res.first()
53         if first is None:
54             res_type_name = 'pserver_dev'
55             resourcetype_id = str(uuid.uuid3(
56                 uuid.NAMESPACE_URL, res_type_name))
57             res_type = ResourceType(
58                 resourcetype_id,
59                 res_type_name, stxobj.type,
60                 description='A Device resource type of Physical Server')
61             dict_id = str(uuid.uuid3(
62                 uuid.NAMESPACE_URL,
63                 str(f"{res_type_name}_alarmdictionary")))
64             alarm_dictionary = uow.alarm_dictionaries.get(dict_id)
65             if alarm_dictionary:
66                 res_type.alarmDictionary = alarm_dictionary
67             uow.resource_types.add(res_type)
68         else:
69             resourcetype_id = first['resourceTypeId']
70
71         resource = uow.resources.get(stxobj.id)
72         if not resource:
73             logger.info("add the device of pserver:" + stxobj.name
74                         + " update_at: " + str(stxobj.updatetime)
75                         + " id: " + str(stxobj.id)
76                         + " hash: " + str(stxobj.hash))
77             localmodel = create_by(stxobj, p_resource, resourcetype_id)
78             uow.resources.add(localmodel)
79
80             logger.info("Add the device of pserver: " + stxobj.id
81                         + ", name: " + stxobj.name)
82         else:
83             localmodel = resource
84             if is_outdated(localmodel, stxobj):
85                 logger.info("update device of pserver:" + stxobj.name
86                             + " update_at: " + str(stxobj.updatetime)
87                             + " id: " + str(stxobj.id)
88                             + " hash: " + str(stxobj.hash))
89                 update_by(localmodel, stxobj, p_resource)
90                 uow.resources.update(localmodel)
91
92             logger.info("Update the device of pserver: " + stxobj.id
93                         + ", name: " + stxobj.name)
94         uow.commit()
95
96
97 def is_outdated(resource: Resource, stxobj: StxGenericModel):
98     return True if resource.hash != stxobj.hash else False
99
100
101 def create_by(stxobj: StxGenericModel, parent: Resource, resourcetype_id: str)\
102         -> Resource:
103     # content = json.loads(stxobj.content)
104     resourcetype_id = resourcetype_id
105     resourcepool_id = parent.resourcePoolId
106     parent_id = parent.resourceId
107     gAssetId = ''  # TODO: global ID
108     description = "%s : A device resource of the physical server"\
109         % stxobj.name
110     resource = Resource(stxobj.id, resourcetype_id, resourcepool_id,
111                         stxobj.name, parent_id, gAssetId, stxobj.content,
112                         description)
113     resource.createtime = stxobj.createtime
114     resource.updatetime = stxobj.updatetime
115     resource.hash = stxobj.hash
116
117     return resource
118
119
120 def update_by(target: Resource, stxobj: StxGenericModel,
121               parentid: str) -> None:
122     if target.resourceId != stxobj.id:
123         raise MismatchedModel("Mismatched Id")
124     target.createtime = stxobj.createtime
125     target.updatetime = stxobj.updatetime
126     target.hash = stxobj.hash
127     target.version_number = target.version_number + 1
128     target.events.append(events.ResourceChanged(
129         id=stxobj.id,
130         resourcePoolId=target.resourcePoolId,
131         notificationEventType=NotificationEventEnum.MODIFY,
132         updatetime=stxobj.updatetime
133     ))