Fix vfc-vnflcm coverage issue
[vfc/gvnfm/vnflcm.git] / lcm / lcm / nf / views.py
1 # Copyright 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 import logging
16 import traceback
17
18 from drf_yasg.utils import swagger_auto_schema
19 from rest_framework import status
20 from rest_framework.response import Response
21 from rest_framework.views import APIView
22
23 from lcm.nf.serializers import CreateVnfReqSerializer, CreateVnfRespSerializer, VnfsInfoSerializer, \
24     InstOrTeriVnfResponseSerializer, InstantiateVnfRequestSerializer, VnfInfoSerializer, TerminateVnfRequestSerializer
25 from lcm.nf.vnf_cancel.delete_vnf_identifier import DeleteVnf
26 from lcm.nf.vnf_cancel.term_vnf import TermVnf
27 from lcm.nf.vnf_create.create_vnf_identifier import CreateVnf
28 from lcm.nf.vnf_create.inst_vnf import InstVnf
29 from lcm.nf.vnf_query.query_vnf import QueryVnf
30 from lcm.pub.exceptions import NFLCMException
31 from lcm.pub.utils.jobutil import JobUtil
32
33 logger = logging.getLogger(__name__)
34
35
36 class CreateVnfAndQueryVnfs(APIView):
37     @swagger_auto_schema(
38         responses={
39             status.HTTP_200_OK: VnfsInfoSerializer(),
40             status.HTTP_500_INTERNAL_SERVER_ERROR: "Internal error"
41         }
42     )
43     def get(self, request):
44         logger.debug("QueryMultiVnf--get::> %s" % request.data)
45         try:
46             resp_data = QueryVnf(request.data).query_multi_vnf()
47
48             vnfs_info_serializer = VnfsInfoSerializer(data=resp_data)
49             if not vnfs_info_serializer.is_valid():
50                 raise NFLCMException(vnfs_info_serializer.errors)
51
52             return Response(data=vnfs_info_serializer.data, status=status.HTTP_200_OK)
53         except NFLCMException as e:
54             logger.error(e.message)
55             return Response(data={'error': '%s' % e.message}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
56         except Exception as e:
57             logger.error(e.message)
58             logger.error(traceback.format_exc())
59             return Response(data={'error': 'Failed to get Vnfs'}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
60
61     @swagger_auto_schema(
62         request_body=CreateVnfReqSerializer(),
63         responses={
64             status.HTTP_201_CREATED: CreateVnfRespSerializer(),
65             status.HTTP_500_INTERNAL_SERVER_ERROR: "Internal error"
66         }
67     )
68     def post(self, request):
69         logger.debug("CreateVnfIdentifier--post::> %s" % request.data)
70         try:
71             req_serializer = CreateVnfReqSerializer(data=request.data)
72             if not req_serializer.is_valid():
73                 raise NFLCMException(req_serializer.errors)
74
75             nf_inst_id = CreateVnf(req_serializer.data).do_biz()
76
77             create_vnf_resp_serializer = CreateVnfRespSerializer(data={"vnfInstanceId": nf_inst_id})
78             if not create_vnf_resp_serializer.is_valid():
79                 raise NFLCMException(create_vnf_resp_serializer.errors)
80             return Response(data=create_vnf_resp_serializer.data, status=status.HTTP_201_CREATED)
81         except NFLCMException as e:
82             logger.error(e.message)
83             return Response(data={'error': '%s' % e.message}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
84         except Exception as e:
85             logger.error(e.message)
86             logger.error(traceback.format_exc())
87             return Response(data={'error': 'unexpected exception'}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
88
89
90 class InstantiateVnf(APIView):
91     @swagger_auto_schema(
92         request_body=InstantiateVnfRequestSerializer(),
93         responses={
94             status.HTTP_202_ACCEPTED: InstOrTeriVnfResponseSerializer(),
95             status.HTTP_500_INTERNAL_SERVER_ERROR: "Internal error"
96         }
97     )
98     def post(self, request, instanceid):
99         logger.debug("InstantiateVnf--post::> %s" % request.data)
100         try:
101             instantiate_vnf_request_serializer = InstantiateVnfRequestSerializer(data=request.data)
102             if not instantiate_vnf_request_serializer.is_valid():
103                 raise NFLCMException(instantiate_vnf_request_serializer.errors)
104
105             job_id = JobUtil.create_job('NF', 'INSTANTIATE', instanceid)
106             JobUtil.add_job_status(job_id, 0, "INST_VNF_READY")
107             InstVnf(instantiate_vnf_request_serializer.data, instanceid, job_id).start()
108
109             instantiate_vnf_response_serializer = InstOrTeriVnfResponseSerializer(data={"jobId": job_id})
110             resp_isvalid = instantiate_vnf_response_serializer.is_valid()
111             if not resp_isvalid:
112                 raise NFLCMException(instantiate_vnf_response_serializer.errors)
113
114             return Response(data=instantiate_vnf_response_serializer.data, status=status.HTTP_202_ACCEPTED)
115         except NFLCMException as e:
116             logger.error(e.message)
117             return Response(data={'error': '%s' % e.message}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
118         except Exception as e:
119             logger.error(e.message)
120             logger.error(traceback.format_exc())
121             return Response(data={'error': 'unexpected exception'}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
122
123
124 class DeleteVnfAndQueryVnf(APIView):
125     @swagger_auto_schema(
126         responses={
127             status.HTTP_200_OK: VnfInfoSerializer(),
128             status.HTTP_500_INTERNAL_SERVER_ERROR: "Internal error"
129         }
130     )
131     def get(self, request, instanceid):
132         logger.debug("QuerySingleVnf--get::> %s" % request.data)
133         try:
134             resp_data = QueryVnf(request.data, instanceid).query_single_vnf()
135
136             vnf_info_serializer = VnfInfoSerializer(data=resp_data)
137             if not vnf_info_serializer.is_valid():
138                 raise NFLCMException(vnf_info_serializer.errors)
139
140             return Response(data=vnf_info_serializer.data, status=status.HTTP_200_OK)
141         except NFLCMException as e:
142             logger.error(e.message)
143             return Response(data={'error': '%s' % e.message}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
144         except Exception as e:
145             logger.eror(e.message)
146             logger.error(traceback.format_exc())
147             return Response(data={'error': 'Failed to get Vnf(%s)' % instanceid},
148                             status=status.HTTP_500_INTERNAL_SERVER_ERROR)
149
150     @swagger_auto_schema(
151         responses={
152             status.HTTP_204_NO_CONTENT: "Successfully",
153             status.HTTP_500_INTERNAL_SERVER_ERROR: "Internal error"
154         }
155     )
156     def delete(self, request, instanceid):
157         logger.debug("DeleteVnfIdentifier--delete::> %s" % request.data)
158         try:
159             DeleteVnf(request.data, instanceid).do_biz()
160
161             return Response(data=None, status=status.HTTP_204_NO_CONTENT)
162         except NFLCMException as e:
163             logger.error(e.message)
164             return Response(data={'error': '%s' % e.message}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
165         except Exception as e:
166             logger.error(e.message)
167             logger.error(traceback.format_exc())
168             return Response(data={'error': 'unexpected exception'}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
169
170
171 class TerminateVnf(APIView):
172     @swagger_auto_schema(
173         request_body=TerminateVnfRequestSerializer(),
174         responses={
175             status.HTTP_202_ACCEPTED: InstOrTeriVnfResponseSerializer(),
176             status.HTTP_500_INTERNAL_SERVER_ERROR: "Internal error"
177         }
178     )
179     def post(self, request, instanceid):
180         logger.debug("TerminateVnf--post::> %s" % request.data)
181         try:
182             terminate_vnf_request_serializer = TerminateVnfRequestSerializer(data=request.data)
183             if not terminate_vnf_request_serializer.is_valid():
184                 raise NFLCMException(terminate_vnf_request_serializer.errors)
185
186             job_id = JobUtil.create_job('NF', 'TERMINATE', instanceid)
187             JobUtil.add_job_status(job_id, 0, "TERM_VNF_READY")
188             TermVnf(terminate_vnf_request_serializer.data, instanceid, job_id).start()
189
190             terminate_vnf_response_serializer = InstOrTeriVnfResponseSerializer(data={"jobId": job_id})
191             if not terminate_vnf_response_serializer.is_valid():
192                 raise NFLCMException(terminate_vnf_response_serializer.errors)
193
194             return Response(data=terminate_vnf_response_serializer.data, status=status.HTTP_202_ACCEPTED)
195         except NFLCMException as e:
196             logger.error(e.message)
197             return Response(data={'error': '%s' % e.message}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
198         except Exception as e:
199             logger.error(e.message)
200             logger.error(traceback.format_exc())
201             return Response(data={'error': 'unexpected exception'}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)