Move heal ns view as single
[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_manual_scale import NSManualScaleService
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.biz.ns_delete import DeleteNsService
26 from lcm.ns.serializers.ns_serializers import InstNsPostDealReqSerializer
27 from lcm.ns.serializers.ns_serializers import ManualScaleNsReqSerializer
28 from lcm.ns.serializers.ns_serializers import NsOperateJobSerializer
29 from lcm.ns.serializers.ns_serializers import QueryNsRespSerializer
30 from lcm.pub.database.models import NSInstModel, ServiceBaseInfoModel
31 from lcm.pub.exceptions import NSLCMException
32 from lcm.pub.utils.jobutil import JobUtil, JOB_TYPE
33 from lcm.pub.utils.restcall import req_by_msb
34 from lcm.pub.utils.values import ignore_case_get
35
36 logger = logging.getLogger(__name__)
37
38
39 class NSDetailView(APIView):
40     @swagger_auto_schema(
41         request_body=None,
42         responses={
43             status.HTTP_200_OK: QueryNsRespSerializer(help_text="NS instance", many=True),
44             status.HTTP_500_INTERNAL_SERVER_ERROR: "Inner error",
45             status.HTTP_404_NOT_FOUND: "Ns instance does not exist"
46         }
47     )
48     def get(self, request, ns_instance_id):
49         try:
50             logger.debug("Enter NSDetailView::get ns(%s)", ns_instance_id)
51             ns_filter = {"ns_inst_id": ns_instance_id}
52             ret = GetNSInfoService(ns_filter).get_ns_info()
53             if not ret:
54                 return Response(status=status.HTTP_404_NOT_FOUND)
55             logger.debug("Leave NSDetailView::get::ret=%s", ret)
56             resp_serializer = QueryNsRespSerializer(data=ret, many=True)
57             if not resp_serializer.is_valid():
58                 raise NSLCMException(resp_serializer.errors)
59             return Response(data=resp_serializer.data, status=status.HTTP_200_OK)
60         except Exception as e:
61             logger.error(traceback.format_exc())
62             logger.error("Exception in GetNSDetail: %s", e.message)
63             return Response(data={'error': e.message}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
64
65     @swagger_auto_schema(
66         request_body=None,
67         responses={
68             status.HTTP_204_NO_CONTENT: 'successful',
69             status.HTTP_500_INTERNAL_SERVER_ERROR: "Inner error"
70         }
71     )
72     def delete(self, request, ns_instance_id):
73         try:
74             logger.debug("Enter NSDetailView::delete ns(%s)", ns_instance_id)
75             DeleteNsService(ns_instance_id).do_biz()
76             return Response(data={}, status=status.HTTP_204_NO_CONTENT)
77         except Exception as e:
78             logger.error(traceback.format_exc())
79             logger.error("Exception in delete NS: %s", e.message)
80             return Response(data={'error': e.message}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
81
82
83 class NSInstPostDealView(APIView):
84     @swagger_auto_schema(
85         request_body=InstNsPostDealReqSerializer(help_text="NS instant post deal"),
86         responses={
87             status.HTTP_202_ACCEPTED: "NS instant post deal success",
88             status.HTTP_500_INTERNAL_SERVER_ERROR: "Inner error"
89         }
90     )
91     def post(self, request, ns_instance_id):
92         logger.debug("Enter NSInstPostDealView::post %s, %s", request.data, ns_instance_id)
93         ns_post_status = ignore_case_get(request.data, 'status')
94         ns_status = 'ACTIVE' if ns_post_status == 'true' else 'FAILED'
95         ns_opr_status = 'success' if ns_post_status == 'true' else 'failed'
96         try:
97             req_serializer = InstNsPostDealReqSerializer(data=request.data)
98             if not req_serializer.is_valid():
99                 raise NSLCMException(req_serializer.errors)
100             NSInstModel.objects.filter(id=ns_instance_id).update(status=ns_status)
101             ServiceBaseInfoModel.objects.filter(service_id=ns_instance_id).update(
102                 active_status=ns_status, status=ns_opr_status)
103             nsd_info = NSInstModel.objects.filter(id=ns_instance_id)
104             nsd_id = nsd_info[0].nsd_id
105             nsd_model = json.loads(nsd_info[0].nsd_model)
106             if "policies" in nsd_model and nsd_model["policies"]:
107                 policy = nsd_model["policies"][0]
108                 if "properties" in policy and policy["properties"]:
109                     file_url = ignore_case_get(policy["properties"][0], "drl_file_url")
110                 else:
111                     file_url = ""
112                 self.send_policy_request(ns_instance_id, nsd_id, file_url)
113         except:
114             logger.error(traceback.format_exc())
115             return Response(data={'error': 'Failed to update status of NS(%s)' % ns_instance_id},
116                             status=status.HTTP_500_INTERNAL_SERVER_ERROR)
117         logger.debug("*****NS INST %s, %s******", ns_status, ns_opr_status)
118         return Response(data={'success': 'Update status of NS(%s) to %s' % (ns_instance_id, ns_status)},
119                         status=status.HTTP_202_ACCEPTED)
120
121     def send_policy_request(self, ns_instance_id, nsd_id, file_url):
122         input_data = {
123             "nsid": ns_instance_id,
124             "nsdid": nsd_id,
125             "fileUri": file_url
126         }
127         req_param = json.JSONEncoder().encode(input_data)
128         policy_engine_url = 'api/polengine/v1/policyinfo'
129         ret = req_by_msb(policy_engine_url, "POST", req_param)
130         if ret[0] != 0:
131             logger.error("Failed to send ns policy req")
132
133
134 class NSManualScaleView(APIView):
135     @swagger_auto_schema(
136         request_body=ManualScaleNsReqSerializer(help_text="NS manual scale"),
137         responses={
138             status.HTTP_202_ACCEPTED: NsOperateJobSerializer(),
139             status.HTTP_500_INTERNAL_SERVER_ERROR: "Inner error"
140         }
141     )
142     def post(self, request, ns_instance_id):
143         logger.debug("Enter NSManualScaleView::post %s, %s", request.data, ns_instance_id)
144         job_id = JobUtil.create_job("NS", JOB_TYPE.MANUAL_SCALE_VNF, ns_instance_id)
145         try:
146             req_serializer = ManualScaleNsReqSerializer(data=request.data)
147             if not req_serializer.is_valid():
148                 raise NSLCMException(req_serializer.errors)
149
150             NSManualScaleService(ns_instance_id, request.data, job_id).start()
151
152             resp_serializer = NsOperateJobSerializer(data={'jobId': job_id})
153             if not resp_serializer.is_valid():
154                 raise NSLCMException(resp_serializer.errors)
155
156             return Response(data=resp_serializer.data, status=status.HTTP_202_ACCEPTED)
157         except Exception as e:
158             logger.error(traceback.format_exc())
159             JobUtil.add_job_status(job_id, 255, 'NS scale failed: %s' % e.message)
160             return Response(data={'error': 'NS scale failed: %s' % ns_instance_id},
161                             status=status.HTTP_500_INTERNAL_SERVER_ERROR)