diff options
Diffstat (limited to 'catalog/packages/biz/notificationsutil.py')
| -rw-r--r-- | catalog/packages/biz/notificationsutil.py | 228 |
1 files changed, 145 insertions, 83 deletions
diff --git a/catalog/packages/biz/notificationsutil.py b/catalog/packages/biz/notificationsutil.py index 8a653b4..8cdfd80 100644 --- a/catalog/packages/biz/notificationsutil.py +++ b/catalog/packages/biz/notificationsutil.py @@ -17,51 +17,50 @@ import uuid import requests from rest_framework import status from catalog.packages import const -from catalog.pub.database.models import VnfPkgSubscriptionModel, NsdmSubscriptionModel -from catalog.pub.database.models import VnfPackageModel +from catalog.pub.database.models import VnfPackageModel, VnfPkgSubscriptionModel, NsdmSubscriptionModel import catalog.pub.utils.timeutil from catalog.pub.utils.values import remove_none_key from catalog.pub.config import config as pub_config import traceback from django.db.models import Q +from catalog.packages.serializers.vnf_pkg_notifications import PkgChangeNotificationSerializer, \ + PkgOnboardingNotificationSerializer logger = logging.getLogger(__name__) class NotificationsUtil(object): - def __init__(self): + def __init__(self, notification_type): + self.notification_type = notification_type + self.notifyserializer = None + + def prepare_notification(self, **kwargs): pass - def send_notification(self, notification, filters, isvnfpkg): - subscriptions_filter = {v + "__contains": notification[k] for k, v in filters.items()} + def send_notification(self): + notification = self.prepare_notification() + + subscriptions_filter = {v + "__contains": notification[k] for k, v in self.filters.items()} subscriptions_filter = remove_none_key(subscriptions_filter) logger.debug('send_notification subscriptions_filter = %s' % subscriptions_filter) q1 = Q() q1.connector = 'OR' for k, v in subscriptions_filter.items(): q1.children.append((k, v)) - if isvnfpkg: - subscriptions = VnfPkgSubscriptionModel.objects.filter(q1) - subscription_root_uri = const.VNFPKG_SUBSCRIPTION_ROOT_URI - else: - subscriptions = NsdmSubscriptionModel.objects.filter(q1) - subscription_root_uri = const.NSDM_SUBSCRIPTION_ROOT_URI + subscriptions = self.SubscriptionModel.objects.filter(q1) if not subscriptions.exists(): logger.info("No subscriptions created for the filters %s" % notification) return logger.info("Start sending notifications") for sub in subscriptions: # set subscription id - if isvnfpkg: - notification["subscriptionId"] = sub.subscription_id - else: - notification["subscriptionId"] = sub.subscriptionid + notification["subscriptionId"] = sub.get_subscription_id() notification['_links']['subscription'] = { 'href': 'http://%s:%s/%s%s' % (pub_config.MSB_SERVICE_IP, pub_config.MSB_SERVICE_PORT, - subscription_root_uri, + self.subscription_root_uri, notification["subscriptionId"]) } callbackuri = sub.callback_uri @@ -70,6 +69,10 @@ class NotificationsUtil(object): if auth_info["authType"] == const.OAUTH2_CLIENT_CREDENTIALS: pass """ + if self.notifyserializer: + serialized_data = self.notifyserializer(data=notification) + if not serialized_data.is_valid(): + logger.error('Notification Data is invalid:%s.' % serialized_data.errors) self.post_notification(callbackuri, notification) def post_notification(self, callbackuri, notification): @@ -96,76 +99,135 @@ class NotificationsUtil(object): logger.error(traceback.format_exc()) -def prepare_vnfpkg_notification(vnf_pkg_id, notification_type, pkg_change_type, operational_state): - logger.info('Start to prepare notification') - vnf_pkg = VnfPackageModel.objects.filter(vnfPackageId=vnf_pkg_id) - vnfd_id = None - if vnf_pkg: - vnfd_id = vnf_pkg[0].vnfdId - notification_content = { - 'id': str(uuid.uuid4()), # shall be the same if sent multiple times due to multiple subscriptions. - 'notificationType': notification_type, - # set 'subscriptionId' after filtering for subscribers - 'timeStamp': catalog.pub.utils.timeutil.now_time(), - 'vnfPkgId': vnf_pkg_id, - 'vnfdId': vnfd_id, - '_links': { - 'vnfPackage': { - 'href': 'http://%s:%s/%s/vnf_packages/%s' % (pub_config.MSB_SERVICE_IP, - pub_config.MSB_SERVICE_PORT, - const.PKG_URL_PREFIX, - vnf_pkg_id) +class PkgNotifications(NotificationsUtil): + def __init__(self, notification_type, vnf_pkg_id, change_type=None, operational_state=None): + super(PkgNotifications, self).__init__(notification_type) + self.filters = { + 'vnfdId': 'vnfd_id', + 'vnfPkgId': 'vnf_pkg_id' + } + self.vnf_pkg_id = vnf_pkg_id + self.change_type = change_type + self.operational_state = operational_state + self.SubscriptionModel = VnfPkgSubscriptionModel + self.subscription_root_uri = const.VNFPKG_SUBSCRIPTION_ROOT_URI + if self.notification_type == "VnfPackageChangeNotification": + self.notifyserializer = PkgChangeNotificationSerializer + else: + self.notifyserializer = PkgOnboardingNotificationSerializer + + def prepare_notification(self): + logger.info('Start to prepare Pkgnotification') + + vnf_pkg = VnfPackageModel.objects.filter(vnfPackageId=self.vnf_pkg_id) + vnfd_id = None + if vnf_pkg: + vnfd_id = vnf_pkg[0].vnfdId + notification_content = { + 'id': str(uuid.uuid4()), # shall be the same if sent multiple times due to multiple subscriptions. + 'notificationType': self.notification_type, + # set 'subscriptionId' after filtering for subscribers + 'timeStamp': catalog.pub.utils.timeutil.now_time(), + 'vnfPkgId': self.vnf_pkg_id, + 'vnfdId': vnfd_id, + '_links': { + 'vnfPackage': { + 'href': 'http://%s:%s/%s/vnf_packages/%s' % (pub_config.MSB_SERVICE_IP, + pub_config.MSB_SERVICE_PORT, + const.PKG_URL_PREFIX, + self.vnf_pkg_id) + } } } - } - - if notification_type == "VnfPackageChangeNotification": - notification_content['changeType'] = pkg_change_type - notification_content['operationalState'] = operational_state - - return notification_content - - -def prepare_nsd_notification(nsd_info_id, nsd_id, notification_type, failure_details=None, operational_state=None): - logger.info('Start to prepare notification') - notification_content = { - 'id': str(uuid.uuid4()), # shall be the same if sent multiple times due to multiple subscriptions. - 'notificationType': notification_type, - # set 'subscriptionId' after filtering for subscribers - 'timeStamp': catalog.pub.utils.timeutil.now_time(), - 'nsdInfoId': nsd_info_id, - 'nsdId': nsd_id, - 'onboardingFailureDetails': failure_details, - 'nsdOperationalState': operational_state, - '_links': { - 'nsdInfo': { - 'href': 'http://%s:%s/%s/ns_descriptors/%s' % (pub_config.MSB_SERVICE_IP, - pub_config.MSB_SERVICE_PORT, - const.NSD_URL_PREFIX, - nsd_info_id) + + if self.notification_type == "VnfPackageChangeNotification": + notification_content['changeType'] = self.change_type + notification_content['operationalState'] = self.operational_state + + return notification_content + + +class NsdNotifications(NotificationsUtil): + def __init__(self, notification_type, nsd_info_id, nsd_id, failure_details=None, operational_state=None): + super(NsdNotifications, self).__init__(notification_type) + self.filters = { + 'nsdInfoId': 'nsdInfoId', + 'nsdId': 'nsdId', + } + self.SubscriptionModel = NsdmSubscriptionModel + self.subscription_root_uri = const.NSDM_SUBSCRIPTION_ROOT_URI + self.nsd_info_id = nsd_info_id + self.nsd_id = nsd_id + self.failure_details = failure_details + self.operational_state = operational_state + # todo: + # if self.notification_type == "VnfPackageChangeNotification": + # self.notifyserializer = PkgChangeNotificationSerializer + # else: + # self.notifyserializer = PkgOnboardingNotificationSerializer + + def prepare_notification(self): + logger.info('Start to prepare Nsdnotification') + + notification_content = { + 'id': str(uuid.uuid4()), # shall be the same if sent multiple times due to multiple subscriptions. + 'notificationType': self.notification_type, + # set 'subscriptionId' after filtering for subscribers + 'timeStamp': catalog.pub.utils.timeutil.now_time(), + 'nsdInfoId': self.nsd_info_id, + 'nsdId': self.nsd_id, + '_links': { + 'nsdInfo': { + 'href': 'http://%s:%s/%s/ns_descriptors/%s' % (pub_config.MSB_SERVICE_IP, + pub_config.MSB_SERVICE_PORT, + const.NSD_URL_PREFIX, + self.nsd_info_id) + } } } - } - return notification_content - - -def prepare_pnfd_notification(pnfd_info_id, pnfd_id, notification_type, failure_details=None): - logger.info('Start to prepare notification') - notification_content = { - 'id': str(uuid.uuid4()), # shall be the same if sent multiple times due to multiple subscriptions. - 'notificationType': notification_type, - # set 'subscriptionId' after filtering for subscribers - 'timeStamp': catalog.pub.utils.timeutil.now_time(), - 'pnfdInfoIds': pnfd_info_id, - 'pnfdId': pnfd_id, - 'onboardingFailureDetails': failure_details, - '_links': { - 'pnfdInfo': { - 'href': 'http://%s:%s/%s/pnf_descriptors/%s' % (pub_config.MSB_SERVICE_IP, - pub_config.MSB_SERVICE_PORT, - const.NSD_URL_PREFIX, - pnfd_info_id) + if self.notification_type == "NsdOnboardingFailureNotification": + notification_content['onboardingFailureDetails'] = self.failure_details + if self.notification_type == "NsdChangeNotification": + notification_content['nsdOperationalState'] = self.operational_state + return notification_content + + +class PnfNotifications(NotificationsUtil): + def __init__(self, notification_type, pnfd_info_id, pnfd_id, failure_details=None): + super(PnfNotifications, self).__init__(notification_type) + self.filters = { + 'pnfdId': 'pnfdId', + 'pnfdInfoIds': 'pnfdInfoIds', + } + self.SubscriptionModel = NsdmSubscriptionModel + self.subscription_root_uri = const.NSDM_SUBSCRIPTION_ROOT_URI + self.pnfd_info_id = pnfd_info_id + self.pnfd_id = pnfd_id + self.failure_details = failure_details + # todo + # if self.notification_type == "VnfPackageChangeNotification": + # self.notifyserializer = PkgChangeNotificationSerializer + # else: + # self.notifyserializer = PkgOnboardingNotificationSerializer + + def prepare_notification(self, *args, **kwargs): + logger.info('Start to prepare Pnfnotification') + notification_content = { + 'id': str(uuid.uuid4()), # shall be the same if sent multiple times due to multiple subscriptions. + 'notificationType': self.notification_type, + # set 'subscriptionId' after filtering for subscribers + 'timeStamp': catalog.pub.utils.timeutil.now_time(), + 'pnfdInfoIds': self.pnfd_info_id, + 'pnfdId': self.pnfd_id, + '_links': { + 'pnfdInfo': { + 'href': 'http://%s:%s/%s/pnf_descriptors/%s' % (pub_config.MSB_SERVICE_IP, + pub_config.MSB_SERVICE_PORT, + const.NSD_URL_PREFIX, + self.pnfd_info_id) + } } } - } - return notification_content + if self.notification_type == "PnfdOnboardingFailureNotification": + notification_content['onboardingFailureDetails'] = self.failure_details + return notification_content |