1 # Copyright 2018 ZTE Corporation.
3 # Licensed under the Apache License, Version 2.0 (the "License");
4 # you may not use this file except in compliance with the License.
5 # You may obtain a copy of the License at
7 # http://www.apache.org/licenses/LICENSE-2.0
9 # Unless required by applicable law or agreed to in writing, software
10 # distributed under the License is distributed on an "AS IS" BASIS,
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
18 from lcm.pub.database.models import SubscriptionModel
19 from lcm.pub.exceptions import NSLCMException
20 from lcm.pub.msapi.extsys import get_vnfm_by_id
21 from lcm.pub.utils.restcall import req_by_msb
22 from lcm.pub.config import config as pub_config
24 logger = logging.getLogger(__name__)
27 class SubscriptionCreation(object):
28 def __init__(self, data):
30 self.vnf_instance_id = self.data['vnfInstanceId']
31 self.vnfm_id = self.data['vnfmId']
32 self.subscription_request_data = {}
33 self.subscription_response_data = {}
37 logger.debug('Start subscribing...')
38 self.prepare_lccn_subscription_request_data()
39 self.send_subscription_request()
40 self.save_subscription_response_data()
41 logger.debug('Subscribing has completed.')
43 def prepare_lccn_subscription_request_data(self):
44 vnfm_info = get_vnfm_by_id(self.vnfm_id)
45 call_back = "%s/api/gvnfmdriver/v1/vnfs/lifecyclechangesnotification" % pub_config.MSB_BASE_URL
46 self.subscription_request_data = {
48 "notificationTypes": ["VnfLcmOperationOccurrenceNotification"],
69 "vnfInstanceSubscriptionFilter": {
71 "vnfInstanceIds": [self.vnf_instance_id],
72 # "vnfInstanceNames": [],
73 # "vnfProductsFromProviders": {}
76 "callbackUri": call_back, # TODO: need reconfirming
78 "authType": ["BASIC"],
80 # "userName": vnfm_info['userName'],
81 # "password": vnfm_info['password']
85 if vnfm_info['userName']:
86 self.subscription_request_data["authentication"]["paramsBasic"]["userName"] = vnfm_info['userName']
87 if vnfm_info['password']:
88 self.subscription_request_data["authentication"]["paramsBasic"]["password"] = vnfm_info['password']
90 def send_subscription_request(self):
91 ret = req_by_msb('api/gvnfmdriver/v1/%s/subscriptions' % self.vnfm_id, 'POST', json.JSONEncoder().encode(self.subscription_request_data))
93 logger.error("Status code is %s, detail is %s.", ret[2], ret[1])
94 raise NSLCMException("Failed to subscribe from vnfm(%s)." % self.vnfm_id)
95 self.subscription_response_data = json.JSONDecoder().decode(ret[1])
97 def save_subscription_response_data(self):
98 logger.debug("Save subscription[%s] to the database" % self.subscription_response_data['id'])
99 lccn_filter = self.subscription_response_data['filter']
100 SubscriptionModel.objects.create(
101 subscription_id=self.subscription_response_data['id'],
102 notification_types=json.dumps(lccn_filter['notificationTypes']),
103 operation_types=json.dumps(lccn_filter['operationTypes']),
104 operation_states=json.dumps(lccn_filter['operationStates']),
105 vnf_instance_filter=json.dumps(lccn_filter['vnfInstanceSubscriptionFilter']),
106 callback_uri=self.subscription_response_data['callbackUri'],
107 links=json.dumps(self.subscription_response_data['_links'])
109 logger.debug('Subscription[%s] has been created', self.subscription_response_data['id'])
112 class SubscriptionDeletion(object):
113 def __init__(self, vnfm_id, vnf_instance_id):
114 self.vnfm_id = vnfm_id
115 self.vnf_instance_id = vnf_instance_id
116 self.subscription_id = None
117 self.subscription = None
120 self.filter_subscription()
121 self.send_subscription_deletion_request()
122 self.delete_subscription_in_db()
124 def filter_subscription(self):
125 subscritptions = SubscriptionModel.objects.filter(vnf_instance_filter__contains=self.vnf_instance_id)
126 if not subscritptions.exists():
127 logger.debug("Subscription contains VNF(%s) does not exist." % self.vnf_instance_id)
128 self.subscription = subscritptions.first()
130 def send_subscription_deletion_request(self):
131 if self.subscription:
132 self.subscription_id = self.subscription.subscription_id
133 ret = req_by_msb('api/gvnfmdriver/v1/%s/subscriptions/%s' % (self.vnfm_id, self.subscription_id), 'DELETE')
135 logger.error('Status code is %s, detail is %s.', ret[2], ret[1])
136 raise NSLCMException("Failed to subscribe from vnfm(%s)." % self.vnfm_id)
137 logger.debug('Subscripton(%s) in vnfm(%s) has been deleted.' % (self.subscription, self.vnfm_id))
139 def delete_subscription_in_db(self):
140 if self.subscription:
141 self.subscription.delete()