Fix vfc-lcm/ns pep8 issue
[vfc/nfvo/lcm.git] / lcm / ns / ns_create.py
1 # Copyright 2016 ZTE Corporation.
2 #
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
6 #
7 #         http://www.apache.org/licenses/LICENSE-2.0
8 #
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.
14 import logging
15 import uuid
16
17 from lcm.pub.config.config import REPORT_TO_AAI
18 from lcm.pub.database.models import NSInstModel
19 from lcm.pub.exceptions import NSLCMException
20 from lcm.pub.msapi.aai import create_customer_aai
21 from lcm.pub.msapi.sdc_run_catalog import query_nspackage_by_id
22 from lcm.pub.utils.timeutil import now_time
23 from lcm.pub.utils.values import ignore_case_get
24
25 logger = logging.getLogger(__name__)
26
27
28 class CreateNSService(object):
29     def __init__(self, nsd_id, ns_name, description):
30         self.nsd_id = nsd_id
31         self.ns_name = ns_name
32         self.description = description
33         self.ns_inst_id = ''
34         self.ns_package_id = ''
35
36     def do_biz(self):
37         self.check_nsd_valid()
38         self.check_ns_inst_name_exist()
39         self.create_ns_inst()
40         if REPORT_TO_AAI:
41             self.create_ns_in_aai()
42         logger.debug("CreateNSService::do_biz::ns_inst_id=%s" % self.ns_inst_id)
43         return self.ns_inst_id
44
45     def check_nsd_valid(self):
46         logger.debug("CreateNSService::check_nsd_valid::nsd_id=%s" % self.nsd_id)
47         ns_package_info = query_nspackage_by_id(self.nsd_id)
48         if not ns_package_info:
49             raise NSLCMException("nsd(%s) not exists." % self.nsd_id)
50         packageInfo = ns_package_info["packageInfo"]
51         self.ns_package_id = ignore_case_get(packageInfo, "nsPackageId")
52         logger.debug("CreateNSService::check_nsd_valid::ns_package_id=%s" % self.ns_package_id)
53
54     def check_ns_inst_name_exist(self):
55         is_exist = NSInstModel.objects.filter(name=self.ns_name).exists()
56         logger.debug("CreateNSService::check_ns_inst_name_exist::is_exist=%s" % is_exist)
57         if is_exist:
58             raise NSLCMException("ns(%s) already existed." % self.ns_name)
59
60     def create_ns_inst(self):
61         self.ns_inst_id = str(uuid.uuid4())
62         logger.debug("CreateNSService::create_ns_inst::ns_inst_id=%s" % self.ns_inst_id)
63         NSInstModel(id=self.ns_inst_id, name=self.ns_name, nspackage_id=self.ns_package_id,
64                     nsd_id=self.nsd_id, description=self.description, status='empty',
65                     lastuptime=now_time()).save()
66
67     def create_ns_in_aai(self):
68         logger.debug("CreateNSService::create_ns_in_aai::report ns instance[%s] to aai." % self.ns_inst_id)
69         global_customer_id = "global-customer-id-" + self.ns_inst_id
70         data = {
71             "global-customer-id": "global-customer-id-" + self.ns_inst_id,
72             "subscriber-name": "subscriber-name-" + self.ns_inst_id,
73             "subscriber-type": "subscriber-type-" + self.ns_inst_id,
74             "service-subscriptions": {
75                 "service-subscription": [
76                     {
77                         "service-type": "service-type-" + self.ns_inst_id,
78                         "service-instances": {
79                             "service-instance": [
80                                 {
81                                     "service-instance-id": self.ns_inst_id,
82                                     "service-instance-name": self.ns_name,
83                                     "service-type": "service-type-" + self.ns_inst_id,
84                                     "service-role": "service-role-" + self.ns_inst_id
85                                 }
86                             ]
87                         }
88                     }
89                 ]
90             }
91         }
92         resp_data, resp_status = create_customer_aai(global_customer_id, data)
93         if resp_data:
94             logger.debug("Fail to create ns[%s] to aai: [%s].", self.ns_inst_id, resp_status)
95         else:
96             logger.debug("Success to create ns[%s] to aai: [%s].", self.ns_inst_id, resp_status)