fix swagger error
[vfc/nfvo/lcm.git] / lcm / ns / views / sol / heal_ns_view.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 from drf_yasg.utils import swagger_auto_schema
16 from rest_framework import status
17 from rest_framework.response import Response
18 from rest_framework.views import APIView
19 from lcm.ns.biz.ns_heal import NSHealService
20 from lcm.ns.serializers.sol.heal_serializers import HealNsReqSerializer
21 from lcm.pub.exceptions import BadRequestException
22 from lcm.pub.utils.jobutil import JobUtil, JOB_TYPE
23 from lcm.ns.const import NS_OCC_BASE_URI
24 from lcm.ns.serializers.sol.pub_serializers import ProblemDetailsSerializer
25
26 logger = logging.getLogger(__name__)
27
28
29 class HealNSView(APIView):
30     @swagger_auto_schema(
31         request_body=HealNsReqSerializer(),
32         responses={
33             status.HTTP_202_ACCEPTED: "HTTP_202_ACCEPTED",
34             status.HTTP_500_INTERNAL_SERVER_ERROR: ProblemDetailsSerializer()
35         }
36     )
37     def post(self, request, ns_instance_id):
38         job_id = JobUtil.create_job("VNF", JOB_TYPE.HEAL_VNF, ns_instance_id)
39         try:
40             logger.debug("Enter HealNSView::post %s", request.data)
41             logger.debug("Enter HealNSView:: %s", ns_instance_id)
42             req_serializer = HealNsReqSerializer(data=request.data)
43             if not req_serializer.is_valid():
44                 logger.debug("request.data is not valid,error: %s" % req_serializer.errors)
45                 raise BadRequestException(req_serializer.errors)
46             nsHealService = NSHealService(ns_instance_id, request.data, job_id)
47             nsHealService.start()
48             response = Response(data={}, status=status.HTTP_202_ACCEPTED)
49             logger.debug("Location: %s" % nsHealService.occ_id)
50             response["Location"] = NS_OCC_BASE_URI % nsHealService.occ_id
51             logger.debug("Leave NSHealView")
52             return response
53         except BadRequestException as e:
54             logger.error("Exception in HealNS: %s", e.message)
55             JobUtil.add_job_status(job_id, 255, 'NS heal failed: %s' % e.message)
56             data = {'status': status.HTTP_400_BAD_REQUEST, 'detail': e.message}
57             return Response(data=data, status=status.HTTP_400_BAD_REQUEST)
58         except Exception as e:
59             logger.error("Exception in HealNSView: %s", e.message)
60             JobUtil.add_job_status(job_id, 255, 'NS heal failed: %s' % e.message)
61             data = {'status': status.HTTP_500_INTERNAL_SERVER_ERROR, 'detail': e.message}
62             return Response(data=data, status=status.HTTP_500_INTERNAL_SERVER_ERROR)