19ff8c01eade7ced380ad6c816c777c0d55c2397
[vfc/nfvo/lcm.git] / lcm / ns / views / 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 import json
15 import logging
16 import traceback
17
18 from drf_yasg.utils import swagger_auto_schema
19 from lcm.ns.biz.ns_get import GetNSInfoService
20 from lcm.ns.biz.ns_heal import NSHealService
21 from lcm.ns.biz.ns_manual_scale import NSManualScaleService
22 from lcm.ns.biz.ns_terminate import TerminateNsService
23 from rest_framework import status
24 from rest_framework.response import Response
25 from rest_framework.views import APIView
26
27 from lcm.ns.biz.ns_delete import DeleteNsService
28 from lcm.ns.serializers.ns_serializers import HealNsReqSerializer
29 from lcm.ns.serializers.ns_serializers import InstNsPostDealReqSerializer
30 from lcm.ns.serializers.ns_serializers import ManualScaleNsReqSerializer
31 from lcm.ns.serializers.ns_serializers import NsOperateJobSerializer
32 from lcm.ns.serializers.ns_serializers import QueryNsRespSerializer
33 from lcm.ns.serializers.ns_serializers import TerminateNsReqSerializer
34 from lcm.pub.database.models import NSInstModel, ServiceBaseInfoModel
35 from lcm.pub.exceptions import NSLCMException
36 from lcm.pub.utils.jobutil import JobUtil, JOB_TYPE
37 from lcm.pub.utils.restcall import req_by_msb
38 from lcm.pub.utils.values import ignore_case_get
39
40 logger = logging.getLogger(__name__)
41
42
43 class TerminateNSView(APIView):
44     @swagger_auto_schema(
45         request_body=TerminateNsReqSerializer(),
46         responses={
47             status.HTTP_202_ACCEPTED: NsOperateJobSerializer(),
48             status.HTTP_500_INTERNAL_SERVER_ERROR: "Inner error"
49         }
50     )
51     def post(self, request, ns_instance_id):
52         try:
53             logger.debug("Enter TerminateNSView::post %s", request.data)
54             req_serializer = TerminateNsReqSerializer(data=request.data)
55             if not req_serializer.is_valid():
56                 raise NSLCMException(req_serializer.errors)
57
58             termination_type = ignore_case_get(request.data, 'terminationType')
59             graceful_termination_timeout = ignore_case_get(request.data, 'gracefulTerminationTimeout')
60             job_id = JobUtil.create_job("VNF", JOB_TYPE.TERMINATE_VNF, ns_instance_id)
61             TerminateNsService(ns_instance_id, termination_type, graceful_termination_timeout, job_id).start()
62
63             resp_serializer = NsOperateJobSerializer(data={'jobId': job_id})
64             if not resp_serializer.is_valid():
65                 raise NSLCMException(resp_serializer.errors)
66             logger.debug("Leave TerminateNSView::post ret=%s", resp_serializer.data)
67             return Response(data=resp_serializer.data, status=status.HTTP_202_ACCEPTED)
68         except Exception as e:
69             logger.error("Exception in CreateNS: %s", e.message)
70             return Response(data={'error': e.message}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
71
72
73 class NSHealView(APIView):
74     @swagger_auto_schema(
75         request_body=HealNsReqSerializer(),
76         responses={
77             status.HTTP_202_ACCEPTED: NsOperateJobSerializer(),
78             status.HTTP_500_INTERNAL_SERVER_ERROR: "Inner error"
79         }
80     )
81     def post(self, request, ns_instance_id):
82         try:
83             logger.debug("Enter HealNSView::post %s", request.data)
84             logger.debug("Enter HealNSView:: %s", ns_instance_id)
85             req_serializer = HealNsReqSerializer(data=request.data)
86             if not req_serializer.is_valid():
87                 raise NSLCMException(req_serializer.errors)
88
89             job_id = JobUtil.create_job("VNF", JOB_TYPE.HEAL_VNF, ns_instance_id)
90             NSHealService(ns_instance_id, request.data, job_id).start()
91
92             resp_serializer = NsOperateJobSerializer(data={'jobId': job_id})
93             if not resp_serializer.is_valid():
94                 raise NSLCMException(resp_serializer.errors)
95
96             logger.debug("Leave HealNSView::post ret=%s", resp_serializer.data)
97             return Response(data=resp_serializer.data, status=status.HTTP_202_ACCEPTED)
98         except Exception as e:
99             logger.error("Exception in HealNSView: %s", e.message)
100             return Response(data={'error': e.message}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
101
102
103 class NSDetailView(APIView):
104     @swagger_auto_schema(
105         request_body=None,
106         responses={
107             status.HTTP_200_OK: QueryNsRespSerializer(help_text="NS instance", many=True),
108             status.HTTP_500_INTERNAL_SERVER_ERROR: "Inner error",
109             status.HTTP_404_NOT_FOUND: "Ns instance does not exist"
110         }
111     )
112     def get(self, request, ns_instance_id):
113         try:
114             logger.debug("Enter NSDetailView::get ns(%s)", ns_instance_id)
115             ns_filter = {"ns_inst_id": ns_instance_id}
116             ret = GetNSInfoService(ns_filter).get_ns_info()
117             if not ret:
118                 return Response(status=status.HTTP_404_NOT_FOUND)
119             logger.debug("Leave NSDetailView::get::ret=%s", ret)
120             resp_serializer = QueryNsRespSerializer(data=ret, many=True)
121             if not resp_serializer.is_valid():
122                 raise NSLCMException(resp_serializer.errors)
123             return Response(data=resp_serializer.data, status=status.HTTP_200_OK)
124         except Exception as e:
125             logger.error(traceback.format_exc())
126             logger.error("Exception in GetNSDetail: %s", e.message)
127             return Response(data={'error': e.message}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
128
129     @swagger_auto_schema(
130         request_body=None,
131         responses={
132             status.HTTP_204_NO_CONTENT: 'successful',
133             status.HTTP_500_INTERNAL_SERVER_ERROR: "Inner error"
134         }
135     )
136     def delete(self, request, ns_instance_id):
137         try:
138             logger.debug("Enter NSDetailView::delete ns(%s)", ns_instance_id)
139             DeleteNsService(ns_instance_id).do_biz()
140             return Response(data={}, status=status.HTTP_204_NO_CONTENT)
141         except Exception as e:
142             logger.error(traceback.format_exc())
143             logger.error("Exception in delete NS: %s", e.message)
144             return Response(data={'error': e.message}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
145
146
147 class NSInstPostDealView(APIView):
148     @swagger_auto_schema(
149         request_body=InstNsPostDealReqSerializer(help_text="NS instant post deal"),
150         responses={
151             status.HTTP_202_ACCEPTED: "NS instant post deal success",
152             status.HTTP_500_INTERNAL_SERVER_ERROR: "Inner error"
153         }
154     )
155     def post(self, request, ns_instance_id):
156         logger.debug("Enter NSInstPostDealView::post %s, %s", request.data, ns_instance_id)
157         ns_post_status = ignore_case_get(request.data, 'status')
158         ns_status = 'ACTIVE' if ns_post_status == 'true' else 'FAILED'
159         ns_opr_status = 'success' if ns_post_status == 'true' else 'failed'
160         try:
161             req_serializer = InstNsPostDealReqSerializer(data=request.data)
162             if not req_serializer.is_valid():
163                 raise NSLCMException(req_serializer.errors)
164             NSInstModel.objects.filter(id=ns_instance_id).update(status=ns_status)
165             ServiceBaseInfoModel.objects.filter(service_id=ns_instance_id).update(
166                 active_status=ns_status, status=ns_opr_status)
167             nsd_info = NSInstModel.objects.filter(id=ns_instance_id)
168             nsd_id = nsd_info[0].nsd_id
169             nsd_model = json.loads(nsd_info[0].nsd_model)
170             if "policies" in nsd_model and nsd_model["policies"]:
171                 policy = nsd_model["policies"][0]
172                 if "properties" in policy and policy["properties"]:
173                     file_url = ignore_case_get(policy["properties"][0], "drl_file_url")
174                 else:
175                     file_url = ""
176                 self.send_policy_request(ns_instance_id, nsd_id, file_url)
177         except:
178             logger.error(traceback.format_exc())
179             return Response(data={'error': 'Failed to update status of NS(%s)' % ns_instance_id},
180                             status=status.HTTP_500_INTERNAL_SERVER_ERROR)
181         logger.debug("*****NS INST %s, %s******", ns_status, ns_opr_status)
182         return Response(data={'success': 'Update status of NS(%s) to %s' % (ns_instance_id, ns_status)},
183                         status=status.HTTP_202_ACCEPTED)
184
185     def send_policy_request(self, ns_instance_id, nsd_id, file_url):
186         input_data = {
187             "nsid": ns_instance_id,
188             "nsdid": nsd_id,
189             "fileUri": file_url
190         }
191         req_param = json.JSONEncoder().encode(input_data)
192         policy_engine_url = 'api/polengine/v1/policyinfo'
193         ret = req_by_msb(policy_engine_url, "POST", req_param)
194         if ret[0] != 0:
195             logger.error("Failed to send ns policy req")
196
197
198 class NSManualScaleView(APIView):
199     @swagger_auto_schema(
200         request_body=ManualScaleNsReqSerializer(help_text="NS manual scale"),
201         responses={
202             status.HTTP_202_ACCEPTED: NsOperateJobSerializer(),
203             status.HTTP_500_INTERNAL_SERVER_ERROR: "Inner error"
204         }
205     )
206     def post(self, request, ns_instance_id):
207         logger.debug("Enter NSManualScaleView::post %s, %s", request.data, ns_instance_id)
208         job_id = JobUtil.create_job("NS", JOB_TYPE.MANUAL_SCALE_VNF, ns_instance_id)
209         try:
210             req_serializer = ManualScaleNsReqSerializer(data=request.data)
211             if not req_serializer.is_valid():
212                 raise NSLCMException(req_serializer.errors)
213
214             NSManualScaleService(ns_instance_id, request.data, job_id).start()
215
216             resp_serializer = NsOperateJobSerializer(data={'jobId': job_id})
217             if not resp_serializer.is_valid():
218                 raise NSLCMException(resp_serializer.errors)
219
220             return Response(data=resp_serializer.data, status=status.HTTP_202_ACCEPTED)
221         except Exception as e:
222             logger.error(traceback.format_exc())
223             JobUtil.add_job_status(job_id, 255, 'NS scale failed: %s' % e.message)
224             return Response(data={'error': 'NS scale failed: %s' % ns_instance_id},
225                             status=status.HTTP_500_INTERNAL_SERVER_ERROR)