Add the command that registers to the SMO; Make the create registration and create...

[pti/o2.git] / o2ims / service / command / registration_handler.py

diff --git a/o2ims/service/command/registration_handler.py b/o2ims/service/command/registration_handler.py
new file mode 100644 (file)
index 0000000..0a4395d
--- /dev/null
+++ b/o2ims/service/command/registration_handler.py
@@ -0,0 +1,103 @@
+# 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.subscription_obj import 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
+        regs = uow.registrations.list()\r
+        for reg in regs:\r
+            reg_data = reg.serialize()\r
+            logger.debug('Registration: {}'.format(reg_data['registrationId']))\r
+\r
+            register_smo(uow, reg_data)\r
+    else:\r
+        with uow:\r
+            reg = uow.registrations.get(data.id)\r
+            if reg is None:\r
+                return\r
+            logger.debug('Registration: {}'.format(reg.registrationId))\r
+            reg_data = reg.serialize()\r
+            register_smo(uow, reg_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.registrations.get(reg_data['registrationId'])\r
+        if reg is None:\r
+            return\r
+        reg.status = RegistrationStatusEnum.NOTIFIED\r
+        logger.debug('Updating Registration: {}'.format(\r
+            reg.registrationId))\r
+        uow.registrations.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['registrationId'],\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