1 # Copyright (C) 2021 Wind River Systems, Inc.
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
7 # http://www.apache.org/licenses/LICENSE-2.0
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.
20 from urllib.parse import urlparse
21 from retry import retry
23 from o2common.service.unit_of_work import AbstractUnitOfWork
24 from o2common.config import config
25 from o2ims.domain import commands
26 from o2ims.domain.configuration_obj import ConfigurationTypeEnum, \
27 RegistrationStatusEnum
29 from o2common.helper import o2logging
30 logger = o2logging.get_logger(__name__)
34 cmd: commands.Register2SMO,
35 uow: AbstractUnitOfWork,
37 logger.info('In registry_to_smo')
39 logger.info('The Register2SMO all is {}'.format(data.all))
41 confs = uow.configrations.list()
43 if conf.conftype != ConfigurationTypeEnum.SMO:
45 reg_data = conf.serialize()
46 logger.debug('Configuration: {}'.format(
47 reg_data['configurationId']))
49 register_smo(uow, reg_data)
52 conf = uow.configurations.get(data.id)
55 logger.debug('Configuration: {}'.format(conf.configurationId))
56 conf_data = conf.serialize()
57 register_smo(uow, conf_data)
60 def register_smo(uow, reg_data):
61 call_res = call_smo(reg_data)
62 logger.debug('Call SMO response is {}'.format(call_res))
64 reg = uow.configurations.get(reg_data['configurationId'])
67 reg.status = RegistrationStatusEnum.NOTIFIED
68 logger.debug('Updating Configurations: {}'.format(
70 uow.configurations.update(reg)
74 # def retry(fun, max_tries=2):
75 # for i in range(max_tries):
78 # # await asyncio.sleep(5*i)
80 # logger.debug('retry function result: {}'.format(res))
86 @retry((ConnectionRefusedError), tries=2, delay=2)
87 def call_smo(reg_data: dict):
88 callback_data = json.dumps({
89 'consumerSubscriptionId': reg_data['configurationId'],
90 'imsUrl': config.get_api_url()
92 logger.info('URL: {}, data: {}'.format(
93 reg_data['callback'], callback_data))
95 o = urlparse(reg_data['callback'])
96 conn = http.client.HTTPConnection(o.netloc)
97 headers = {'Content-type': 'application/json'}
98 conn.request('POST', o.path, callback_data, headers)
99 resp = conn.getresponse()
100 data = resp.read().decode('utf-8')
101 # json_data = json.loads(data)
102 if resp.status == 202 or resp.status == 200:
103 logger.info('Registrer to SMO successed, response code {} {}, data {}'.
104 format(resp.status, resp.reason, data))
106 logger.error('Response code is: {}'.format(resp.status))