fix swagger error
[vfc/nfvo/lcm.git] / lcm / ns / views / sol / scale_ns_views.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
16 from drf_yasg.utils import swagger_auto_schema
17 from rest_framework import status
18 from rest_framework.response import Response
19 from rest_framework.views import APIView
20 from lcm.ns.biz.ns_manual_scale import NSManualScaleService
21 from lcm.ns.serializers.sol.scale_ns_serializers import ManualScaleNsReqSerializer
22 from lcm.pub.exceptions import NSLCMException
23 from lcm.pub.utils.jobutil import JobUtil, JOB_TYPE
24 from lcm.ns.const import NS_OCC_BASE_URI
25 from lcm.pub.exceptions import BadRequestException
26 from lcm.ns.serializers.sol.pub_serializers import ProblemDetailsSerializer
27
28 logger = logging.getLogger(__name__)
29
30
31 class ScaleNSView(APIView):
32     @swagger_auto_schema(
33         request_body=ManualScaleNsReqSerializer(help_text="NS Scale"),
34         responses={
35             status.HTTP_202_ACCEPTED: "HTTP_202_ACCEPTED",
36             status.HTTP_500_INTERNAL_SERVER_ERROR: ProblemDetailsSerializer()
37         }
38     )
39     def post(self, request, ns_instance_id):
40         logger.debug("Enter ScaleNSView::post %s, %s", request.data, ns_instance_id)
41         job_id = JobUtil.create_job("NS", JOB_TYPE.MANUAL_SCALE_VNF, ns_instance_id)
42         try:
43             req_serializer = ManualScaleNsReqSerializer(data=request.data)
44             if not req_serializer.is_valid():
45                 raise NSLCMException(req_serializer.errors)
46             nsManualScaleService = NSManualScaleService(ns_instance_id, request.data, job_id)
47             nsManualScaleService.start()
48             response = Response(data={}, status=status.HTTP_202_ACCEPTED)
49             logger.debug("Location: %s" % nsManualScaleService.occ_id)
50             response["Location"] = NS_OCC_BASE_URI % nsManualScaleService.occ_id
51             logger.debug("Leave ScaleNSView")
52             return response
53         except BadRequestException as e:
54             logger.error("Exception in ScaleNSView: %s", e.message)
55             JobUtil.add_job_status(job_id, 255, 'NS scale 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 ScaleNSView: %s", e.message)
60             JobUtil.add_job_status(job_id, 255, 'NS scale 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)