Refactor the notification process code
[modeling/etsicatalog.git] / catalog / packages / biz / notificationsutil.py
index 8a653b4..8cdfd80 100644 (file)
@@ -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