Fix vfc-catalog/swagger pep8 issue.
[vfc/nfvo/lcm.git] / lcm / ns / views.py
1 # Copyright 2016-2017 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
15
16 import json
17 import logging
18 import os
19 import traceback
20
21 from rest_framework import status
22 from rest_framework.response import Response
23 from rest_framework.views import APIView
24
25 from lcm.ns.ns_create import CreateNSService
26 from lcm.ns.ns_delete import DeleteNsService
27 from lcm.ns.ns_get import GetNSInfoService
28 from lcm.ns.ns_heal import NSHealService
29 from lcm.ns.ns_instant import InstantNSService
30 from lcm.ns.ns_manual_scale import NSManualScaleService
31 from lcm.ns.ns_terminate import TerminateNsService
32 from lcm.pub.database.models import NSInstModel, ServiceBaseInfoModel
33 from lcm.pub.utils.jobutil import JobUtil, JOB_TYPE
34 from lcm.pub.utils.restcall import req_by_msb
35 from lcm.pub.utils.values import ignore_case_get
36
37 logger = logging.getLogger(__name__)
38
39
40 class CreateNSView(APIView):
41     def get(self, request):
42         logger.debug("CreateNSView::get")
43         filter = None
44         csarId = ignore_case_get(request.META, 'csarId')
45         if csarId:
46             filter = {"csarId": csarId}
47
48         ret = GetNSInfoService(filter).get_ns_info()
49         logger.debug("CreateNSView::get::ret=%s", ret)
50         return Response(data=ret, status=status.HTTP_200_OK)
51
52     def post(self, request):
53         logger.debug("Enter CreateNS: %s", request.data)
54         if ignore_case_get(request.data, 'test') == "test":
55             return Response(data={'nsInstanceId': "test"}, status=status.HTTP_201_CREATED)
56         nsd_id = ignore_case_get(request.data, 'nsdId')
57         ns_name = ignore_case_get(request.data, 'nsName')
58         description = ignore_case_get(request.data, 'description')
59         try:
60             ns_inst_id = CreateNSService(nsd_id, ns_name, description).do_biz()
61         except Exception as e:
62             logger.error("Exception in CreateNS: %s", e.message)
63             return Response(data={'error': e.message}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
64         logger.debug("CreateNSView::post::ret={'nsInstanceId':%s}", ns_inst_id)
65         return Response(data={'nsInstanceId': ns_inst_id}, status=status.HTTP_201_CREATED)
66
67
68 class NSInstView(APIView):
69     def post(self, request, ns_instance_id):
70         ack = InstantNSService(ns_instance_id, request.data).do_biz()
71         logger.debug("Leave NSInstView::post::ack=%s", ack)
72         return Response(data=ack['data'], status=ack['status'])
73
74
75 class TerminateNSView(APIView):
76     def post(self, request, ns_instance_id):
77         logger.debug("Enter TerminateNSView::post %s", request.data)
78         termination_type = ignore_case_get(request.data, 'terminationType')
79         graceful_termination_timeout = ignore_case_get(request.data, 'gracefulTerminationTimeout')
80         job_id = JobUtil.create_job("VNF", JOB_TYPE.TERMINATE_VNF, ns_instance_id)
81         try:
82             TerminateNsService(ns_instance_id, termination_type, graceful_termination_timeout, job_id).start()
83         except Exception as e:
84             logger.error("Exception in CreateNS: %s", e.message)
85             return Response(data={'error': e.message}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
86         ret = {'jobId': job_id}
87         logger.debug("Leave TerminateNSView::post ret=%s", ret)
88         return Response(data=ret, status=status.HTTP_202_ACCEPTED)
89
90
91 class NSHealView(APIView):
92     def post(self, request, ns_instance_id):
93         logger.debug("Enter HealNSView::post %s", request.data)
94         job_id = JobUtil.create_job("VNF", JOB_TYPE.HEAL_VNF, ns_instance_id)
95         try:
96             NSHealService(ns_instance_id, request.data, job_id).start()
97         except Exception as e:
98             logger.error("Exception in HealNSView: %s", e.message)
99             return Response(data={'error': e.message}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
100         ret = {'jobId': job_id}
101         logger.debug("Leave HealNSView::post ret=%s", ret)
102         return Response(data=ret, status=status.HTTP_202_ACCEPTED)
103
104
105 class NSDetailView(APIView):
106     def get(self, request, ns_instance_id):
107         logger.debug("Enter NSDetailView::get ns(%s)", ns_instance_id)
108         ns_filter = {"ns_inst_id": ns_instance_id}
109         ret = GetNSInfoService(ns_filter).get_ns_info()
110         if not ret:
111             return Response(status=status.HTTP_404_NOT_FOUND)
112         logger.debug("Leave NSDetailView::get::ret=%s", ret)
113         return Response(data=ret, status=status.HTTP_200_OK)
114
115     def delete(self, request, ns_instance_id):
116         logger.debug("Enter NSDetailView::delete ns(%s)", ns_instance_id)
117         DeleteNsService(ns_instance_id).do_biz()
118         return Response(data={}, status=status.HTTP_204_NO_CONTENT)
119
120
121 class SwaggerJsonView(APIView):
122     def get(self, request):
123         json_file = os.path.join(os.path.dirname(__file__), 'swagger.json')
124         f = open(json_file)
125         json_data = json.JSONDecoder().decode(f.read())
126         f.close()
127         return Response(json_data)
128
129
130 class NSInstPostDealView(APIView):
131     def post(self, request, ns_instance_id):
132         logger.debug("Enter NSInstPostDealView::post %s, %s", request.data, ns_instance_id)
133         ns_post_status = ignore_case_get(request.data, 'status')
134         ns_status = 'ACTIVE' if ns_post_status == 'true' else 'FAILED'
135         ns_opr_status = 'success' if ns_post_status == 'true' else 'failed'
136         try:
137             NSInstModel.objects.filter(id=ns_instance_id).update(status=ns_status)
138             ServiceBaseInfoModel.objects.filter(service_id=ns_instance_id).update(
139                 active_status=ns_status, status=ns_opr_status)
140             nsd_info = NSInstModel.objects.filter(id=ns_instance_id)
141             nsd_id = nsd_info[0].nsd_id
142             nsd_model = json.loads(nsd_info[0].nsd_model)
143             if "policies" in nsd_model and nsd_model["policies"]:
144                 policy = nsd_model["policies"][0]
145                 if "properties" in policy and policy["properties"]:
146                     file_url = ignore_case_get(policy["properties"][0], "drl_file_url")
147                 else:
148                     file_url = ""
149                 self.send_policy_request(ns_instance_id, nsd_id, file_url)
150         except:
151             logger.error(traceback.format_exc())
152             return Response(data={'error': 'Failed to update status of NS(%s)' % ns_instance_id},
153                             status=status.HTTP_500_INTERNAL_SERVER_ERROR)
154         logger.debug("*****NS INST %s, %s******", ns_status, ns_opr_status)
155         return Response(data={'success': 'Update status of NS(%s) to %s' % (ns_instance_id, ns_status)},
156                         status=status.HTTP_202_ACCEPTED)
157
158     def send_policy_request(self,ns_instance_id, nsd_id, file_url):
159         input_data = {
160             "nsid": ns_instance_id,
161             "nsdid": nsd_id,
162             "fileUri":file_url
163         }
164         req_param = json.JSONEncoder().encode(input_data)
165         policy_engine_url = 'api/polengine/v1/policyinfo'
166         ret = req_by_msb(policy_engine_url, "POST", req_param)
167         if ret[0] != 0:
168             logger.error("Failed to send ns policy req")
169
170
171 class NSManualScaleView(APIView):
172     def post(self, request, ns_instance_id):
173         logger.debug("Enter NSManualScaleView::post %s, %s", request.data, ns_instance_id)
174         job_id = JobUtil.create_job("NS", JOB_TYPE.MANUAL_SCALE_VNF, ns_instance_id)
175         try:
176             NSManualScaleService(ns_instance_id, request.data, job_id).start()
177         except Exception as e:
178             logger.error(traceback.format_exc())
179             JobUtil.add_job_status(job_id, 255, 'NS scale failed: %s' % e.message)
180             return Response(data={'error': 'NS scale failed: %s' % ns_instance_id},
181                             status=status.HTTP_500_INTERNAL_SERVER_ERROR)
182         return Response(data={'jobId': job_id}, status=status.HTTP_202_ACCEPTED)