-# Copyright (C) 2021 Wind River Systems, Inc.\r
-#\r
-# Licensed under the Apache License, Version 2.0 (the "License");\r
-# you may not use this file except in compliance with the License.\r
-# You may obtain a copy of the License at\r
-#\r
-# http://www.apache.org/licenses/LICENSE-2.0\r
-#\r
-# Unless required by applicable law or agreed to in writing, software\r
-# distributed under the License is distributed on an "AS IS" BASIS,\r
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
-# See the License for the specific language governing permissions and\r
-# limitations under the License.\r
-\r
-# import time\r
-import json\r
-# import asyncio\r
-# import requests\r
-import http.client\r
-from urllib.parse import urlparse\r
-from retry import retry\r
-\r
-from o2common.service.unit_of_work import AbstractUnitOfWork\r
-from o2common.config import config\r
-from o2ims.domain import commands\r
-from o2ims.domain.configuration_obj import ConfigurationTypeEnum, \\r
- RegistrationStatusEnum\r
-\r
-from o2common.helper import o2logging\r
-logger = o2logging.get_logger(__name__)\r
-\r
-\r
-def registry_to_smo(\r
- cmd: commands.Register2SMO,\r
- uow: AbstractUnitOfWork,\r
-):\r
- logger.info('In registry_to_smo')\r
- data = cmd.data\r
- logger.info('The Register2SMO all is {}'.format(data.all))\r
- if data.all:\r
- confs = uow.configrations.list()\r
- for conf in confs:\r
- if conf.conftype != ConfigurationTypeEnum.SMO:\r
- continue\r
- reg_data = conf.serialize()\r
- logger.debug('Configuration: {}'.format(\r
- reg_data['configurationId']))\r
-\r
- register_smo(uow, reg_data)\r
- else:\r
- with uow:\r
- conf = uow.configurations.get(data.id)\r
- if conf is None:\r
- return\r
- logger.debug('Configuration: {}'.format(conf.configurationId))\r
- conf_data = conf.serialize()\r
- register_smo(uow, conf_data)\r
-\r
-\r
-def register_smo(uow, reg_data):\r
- call_res = call_smo(reg_data)\r
- logger.debug('Call SMO response is {}'.format(call_res))\r
- if call_res:\r
- reg = uow.configurations.get(reg_data['configurationId'])\r
- if reg is None:\r
- return\r
- reg.status = RegistrationStatusEnum.NOTIFIED\r
- logger.debug('Updating Configurations: {}'.format(\r
- reg.configurationId))\r
- uow.configurations.update(reg)\r
- uow.commit()\r
-\r
-\r
-# def retry(fun, max_tries=2):\r
-# for i in range(max_tries):\r
-# try:\r
-# time.sleep(5*i)\r
-# # await asyncio.sleep(5*i)\r
-# res = fun()\r
-# logger.debug('retry function result: {}'.format(res))\r
-# return res\r
-# except Exception:\r
-# continue\r
-\r
-\r
-@retry((ConnectionRefusedError), tries=2, delay=2)\r
-def call_smo(reg_data: dict):\r
- callback_data = json.dumps({\r
- 'consumerSubscriptionId': reg_data['configurationId'],\r
- 'imsUrl': config.get_api_url()\r
- })\r
- logger.info('URL: {}, data: {}'.format(\r
- reg_data['callback'], callback_data))\r
-\r
- o = urlparse(reg_data['callback'])\r
- conn = http.client.HTTPConnection(o.netloc)\r
- headers = {'Content-type': 'application/json'}\r
- conn.request('POST', o.path, callback_data, headers)\r
- resp = conn.getresponse()\r
- data = resp.read().decode('utf-8')\r
- # json_data = json.loads(data)\r
- if resp.status == 202 or resp.status == 200:\r
- logger.info('Registrer to SMO successed, response code {} {}, data {}'.\r
- format(resp.status, resp.reason, data))\r
- return True\r
- logger.error('Response code is: {}'.format(resp.status))\r
- return False\r
+# Copyright (C) 2021 Wind River Systems, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# import time
+import json
+# import asyncio
+# import requests
+import http.client
+from urllib.parse import urlparse
+from retry import retry
+
+from o2common.service.unit_of_work import AbstractUnitOfWork
+from o2common.config import config
+from o2ims.domain import commands
+from o2ims.domain.configuration_obj import ConfigurationTypeEnum, \
+ RegistrationStatusEnum
+
+from o2common.helper import o2logging
+logger = o2logging.get_logger(__name__)
+
+
+def registry_to_smo(
+ cmd: commands.Register2SMO,
+ uow: AbstractUnitOfWork,
+):
+ logger.info('In registry_to_smo')
+ data = cmd.data
+ logger.info('The Register2SMO all is {}'.format(data.all))
+ if data.all:
+ confs = uow.configrations.list()
+ for conf in confs:
+ if conf.conftype != ConfigurationTypeEnum.SMO:
+ continue
+ reg_data = conf.serialize()
+ logger.debug('Configuration: {}'.format(
+ reg_data['configurationId']))
+
+ register_smo(uow, reg_data)
+ else:
+ with uow:
+ conf = uow.configurations.get(data.id)
+ if conf is None:
+ return
+ logger.debug('Configuration: {}'.format(conf.configurationId))
+ conf_data = conf.serialize()
+ register_smo(uow, conf_data)
+
+
+def register_smo(uow, reg_data):
+ call_res = call_smo(reg_data)
+ logger.debug('Call SMO response is {}'.format(call_res))
+ if call_res:
+ reg = uow.configurations.get(reg_data['configurationId'])
+ if reg is None:
+ return
+ reg.status = RegistrationStatusEnum.NOTIFIED
+ logger.debug('Updating Configurations: {}'.format(
+ reg.configurationId))
+ uow.configurations.update(reg)
+ uow.commit()
+
+
+# def retry(fun, max_tries=2):
+# for i in range(max_tries):
+# try:
+# time.sleep(5*i)
+# # await asyncio.sleep(5*i)
+# res = fun()
+# logger.debug('retry function result: {}'.format(res))
+# return res
+# except Exception:
+# continue
+
+
+@retry((ConnectionRefusedError), tries=2, delay=2)
+def call_smo(reg_data: dict):
+ callback_data = json.dumps({
+ 'consumerSubscriptionId': reg_data['configurationId'],
+ 'imsUrl': config.get_api_url()
+ })
+ logger.info('URL: {}, data: {}'.format(
+ reg_data['callback'], callback_data))
+
+ o = urlparse(reg_data['callback'])
+ conn = http.client.HTTPConnection(o.netloc)
+ headers = {'Content-type': 'application/json'}
+ conn.request('POST', o.path, callback_data, headers)
+ resp = conn.getresponse()
+ data = resp.read().decode('utf-8')
+ # json_data = json.loads(data)
+ if resp.status == 202 or resp.status == 200:
+ logger.info('Registrer to SMO successed, response code {} {}, data {}'.
+ format(resp.status, resp.reason, data))
+ return True
+ logger.error('Response code is: {}'.format(resp.status))
+ return False
Code Review / pti / o2.git / blobdiff