Merge "Update unit test for swagger"
[vfc/nfvo/lcm.git] / lcm / ns / views / deprecated / scale_ns_views.py
1 # Copyright 2019 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 import traceback
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
21 from lcm.ns.biz.ns_manual_scale import NSManualScaleService
22 from lcm.ns.serializers.deprecated.ns_serializers import _NsOperateJobSerializer
23 from lcm.ns.serializers.deprecated.ns_serializers import _ManualScaleNsReqSerializer
24 from lcm.pub.exceptions import NSLCMException
25 from lcm.pub.utils.jobutil import JobUtil, JOB_TYPE
26
27 logger = logging.getLogger(__name__)
28
29
30 class NSManualScaleView(APIView):
31     @swagger_auto_schema(
32         request_body=_ManualScaleNsReqSerializer(help_text="NS manual scale"),
33         responses={
34             status.HTTP_202_ACCEPTED: _NsOperateJobSerializer(),
35             status.HTTP_500_INTERNAL_SERVER_ERROR: "Inner error"
36         }
37     )
38     def post(self, request, ns_instance_id):
39         logger.debug("Enter NSManualScaleView::post %s, %s", request.data, ns_instance_id)
40         job_id = JobUtil.create_job("NS", JOB_TYPE.MANUAL_SCALE_VNF, ns_instance_id)
41         try:
42             req_serializer = _ManualScaleNsReqSerializer(data=request.data)
43             if not req_serializer.is_valid():
44                 raise NSLCMException(req_serializer.errors)
45             req = request.data
46             scale_data = {}
47             scale_data['scaleType'] = req['scaleType']
48             scaleNsData = req['scaleNsData'][0]
49             scale_data['scaleNsData'] = {"scaleNsByStepsData": scaleNsData['scaleNsByStepsData'][0]}
50             NSManualScaleService(ns_instance_id, scale_data, job_id).start()
51
52             resp_serializer = _NsOperateJobSerializer(data={'jobId': job_id})
53             if not resp_serializer.is_valid():
54                 raise NSLCMException(resp_serializer.errors)
55
56             return Response(data=resp_serializer.data, status=status.HTTP_202_ACCEPTED)
57         except Exception as e:
58             logger.error(traceback.format_exc())
59             JobUtil.add_job_status(job_id, 255, 'NS scale failed: %s' % e.message)
60             return Response(data={'error': 'NS scale failed: %s' % ns_instance_id},
61                             status=status.HTTP_500_INTERNAL_SERVER_ERROR)