1 # Copyright (C) 2021 Wind River Systems, Inc.
\r
3 # Licensed under the Apache License, Version 2.0 (the "License");
\r
4 # you may not use this file except in compliance with the License.
\r
5 # You may obtain a copy of the License at
\r
7 # http://www.apache.org/licenses/LICENSE-2.0
\r
9 # Unless required by applicable law or agreed to in writing, software
\r
10 # distributed under the License is distributed on an "AS IS" BASIS,
\r
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
\r
12 # See the License for the specific language governing permissions and
\r
13 # limitations under the License.
\r
20 from urllib.parse import urlparse
\r
21 from retry import retry
\r
23 from o2common.service.unit_of_work import AbstractUnitOfWork
\r
24 from o2common.config import config
\r
25 from o2ims.domain import commands
\r
26 from o2ims.domain.configuration_obj import ConfigurationTypeEnum, \
\r
27 RegistrationStatusEnum
\r
29 from o2common.helper import o2logging
\r
30 logger = o2logging.get_logger(__name__)
\r
33 def registry_to_smo(
\r
34 cmd: commands.Register2SMO,
\r
35 uow: AbstractUnitOfWork,
\r
37 logger.info('In registry_to_smo')
\r
39 logger.info('The Register2SMO all is {}'.format(data.all))
\r
41 confs = uow.configrations.list()
\r
43 if conf.conftype != ConfigurationTypeEnum.SMO:
\r
45 reg_data = conf.serialize()
\r
46 logger.debug('Configuration: {}'.format(
\r
47 reg_data['configurationId']))
\r
49 register_smo(uow, reg_data)
\r
52 conf = uow.configurations.get(data.id)
\r
55 logger.debug('Configuration: {}'.format(conf.configurationId))
\r
56 conf_data = conf.serialize()
\r
57 register_smo(uow, conf_data)
\r
60 def register_smo(uow, reg_data):
\r
61 call_res = call_smo(reg_data)
\r
62 logger.debug('Call SMO response is {}'.format(call_res))
\r
64 reg = uow.configurations.get(reg_data['configurationId'])
\r
67 reg.status = RegistrationStatusEnum.NOTIFIED
\r
68 logger.debug('Updating Configurations: {}'.format(
\r
69 reg.configurationId))
\r
70 uow.configurations.update(reg)
\r
74 # def retry(fun, max_tries=2):
\r
75 # for i in range(max_tries):
\r
78 # # await asyncio.sleep(5*i)
\r
80 # logger.debug('retry function result: {}'.format(res))
\r
86 @retry((ConnectionRefusedError), tries=2, delay=2)
\r
87 def call_smo(reg_data: dict):
\r
88 callback_data = json.dumps({
\r
89 'consumerSubscriptionId': reg_data['configurationId'],
\r
90 'imsUrl': config.get_api_url()
\r
92 logger.info('URL: {}, data: {}'.format(
\r
93 reg_data['callback'], callback_data))
\r
95 o = urlparse(reg_data['callback'])
\r
96 conn = http.client.HTTPConnection(o.netloc)
\r
97 headers = {'Content-type': 'application/json'}
\r
98 conn.request('POST', o.path, callback_data, headers)
\r
99 resp = conn.getresponse()
\r
100 data = resp.read().decode('utf-8')
\r
101 # json_data = json.loads(data)
\r
102 if resp.status == 202 or resp.status == 200:
\r
103 logger.info('Registrer to SMO successed, response code {} {}, data {}'.
\r
104 format(resp.status, resp.reason, data))
\r
106 logger.error('Response code is: {}'.format(resp.status))
\r