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