-# Copyright (C) 2021 Wind River Systems, Inc.
+# Copyright (C) 2021-2022 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.
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 o2common.config import config, conf
+from o2common.service.command.handler import get_https_conn_default
+from o2common.service.command.handler import get_http_conn
+from o2common.service.command.handler import get_https_conn_selfsigned
+from o2common.service.command.handler import post_data
+import ssl
from o2ims.domain import commands
-from o2ims.domain.configuration_obj import ConfigurationTypeEnum, \
- RegistrationStatusEnum
+from o2ims.domain.subscription_obj import NotificationEventEnum
from o2common.helper import o2logging
logger = o2logging.get_logger(__name__)
):
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)
+ logger.info('The Register2SMO notificationEventType is {}'.format(
+ data.notificationEventType))
+ with uow:
+ ocloud = uow.oclouds.get(data.id)
+ if ocloud is None:
+ return
+ logger.debug('O-Cloud Global UUID: {}'.format(ocloud.globalCloudId))
+ ocloud_dict = ocloud.serialize()
+ if data.notificationEventType == NotificationEventEnum.CREATE:
+ register_smo(uow, ocloud_dict)
+
+
+class RegIMSToSMOExp(Exception):
+ def __init__(self, value):
+ self.value = value
-def register_smo(uow, reg_data):
- call_res = call_smo(reg_data)
+def register_smo(uow, ocloud_data):
+ call_res, status = call_smo(ocloud_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()
+ if call_res is True:
+ logger.info('Register to smo success response is {}'.format(status))
+ else:
+ raise RegIMSToSMOExp('Register o2ims to SMO failed')
+ # TODO: record the result for the smo register
# def retry(fun, max_tries=2):
@retry((ConnectionRefusedError), tries=2, delay=2)
def call_smo(reg_data: dict):
+ smo_token = conf.DEFAULT.smo_token_data
+ smo_token_info = {
+ 'iss': 'o2ims',
+ 'aud': 'smo',
+ 'smo_token_payload': smo_token,
+ 'smo_token_type': 'jwt',
+ 'smo_token_expiration': '',
+ 'smo_token_algo': 'RS256'
+ }
+
callback_data = json.dumps({
- 'consumerSubscriptionId': reg_data['configurationId'],
- 'imsUrl': config.get_api_url()
+ 'globalCloudId': reg_data['globalCloudId'],
+ 'oCloudId': reg_data['oCloudId'],
+ 'IMS_EP': config.get_api_url(),
+ 'smo_token_data': smo_token_info
})
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
+ conf.DEFAULT.smo_register_url, callback_data))
+ o = urlparse(conf.DEFAULT.smo_register_url)
+ if o.scheme == 'https':
+ conn = get_https_conn_default(o.netloc)
+ else:
+ conn = get_http_conn(o.netloc)
+
+ try:
+ return post_data(conn, o.path, callback_data)
+ except ssl.SSLCertVerificationError as e:
+ logger.debug('Try to register to smo with \
+ trusted ca failed: {}'.format(e))
+ if 'self signed' in str(e):
+ conn = get_https_conn_selfsigned(o.netloc)
+ try:
+ return post_data(conn, o.path, callback_data)
+ except Exception as e:
+ logger.info(
+ 'Register to smo with self-signed ca failed: {}'.format(e))
+ # TODO: write the status to extension db table.
+ return False, None
+ return False, None
+ except Exception as e:
+ logger.critical('Register to smo except: {}'.format(e))
+ return False, None