1 # Copyright 2017 ZTE Corporation.
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
7 # http://www.apache.org/licenses/LICENSE-2.0
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.
19 from drf_yasg import openapi
20 from drf_yasg.utils import swagger_auto_schema
21 from rest_framework import status
22 from rest_framework.decorators import api_view
23 from rest_framework.response import Response
24 from rest_framework.views import APIView
26 from mgr.pub.utils.values import ignore_case_get
27 from mgr.pub.utils.syscomm import fun_name
28 from mgr.pub.database.models import VnfRegModel
29 from mgr.pub.utils import restcall
30 from mgr.vnfreg.serializers import ErrorSerializer, VnfInfoSerializer, ResponseSerializer, NoneSerializer, \
33 logger = logging.getLogger(__name__)
36 def handler_exception(e):
37 logger.error(e.message)
38 logger.error(traceback.format_exc())
39 errorSerializer = ErrorSerializer(data={'error': e.message})
40 errorSerializer.is_valid()
41 return errorSerializer.data
44 class vnfmgr_addvnf(APIView):
45 @swagger_auto_schema(request_body=VnfInfoSerializer(),
47 201: ResponseSerializer(),
48 500: ErrorSerializer()})
49 def post(self, request):
50 logger.info("Enter %s, data is %s", fun_name(), request.data)
51 requestSerializer = VnfInfoSerializer(data=request.data)
52 request_isValid = requestSerializer.is_valid()
54 if not request_isValid:
55 raise Exception(requestSerializer.errors)
57 requestData = requestSerializer.data
58 vnf_inst_id = ignore_case_get(requestData, "vnfInstId")
59 if VnfRegModel.objects.filter(id=vnf_inst_id):
60 raise Exception("Vnf(%s) already exists." % vnf_inst_id)
63 ip=ignore_case_get(requestData, "ip"),
64 port=ignore_case_get(requestData, "port"),
65 username=ignore_case_get(requestData, "username"),
66 password=ignore_case_get(requestData, "password")).save()
68 responseSerializer = ResponseSerializer(data={"vnfInstId": vnf_inst_id})
69 isValid = responseSerializer.is_valid()
71 raise Exception(responseSerializer.errors)
72 except Exception as e:
73 errorData = handler_exception(e)
74 return Response(data=errorData, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
76 return Response(data=responseSerializer.data, status=status.HTTP_201_CREATED)
79 @swagger_auto_schema(method='put',
80 request_body=VnfInfoSerializer(),
82 202: NoneSerializer(),
83 500: ErrorSerializer()})
84 @swagger_auto_schema(method='delete',
86 204: NoneSerializer(),
87 500: ErrorSerializer()})
88 @swagger_auto_schema(methods=['get'],
90 openapi.Parameter('test',
93 type=openapi.TYPE_BOOLEAN
96 200: openapi.Response('response description', VnfInfoSerializer()),
97 500: ErrorSerializer()})
98 @api_view(http_method_names=['GET', 'PUT', 'DELETE'])
99 def access_vnf(request, *args, **kwargs):
100 requestSerializer = VnfInfoSerializer(data=request.data)
101 request_isValid = requestSerializer.is_valid()
102 vnf_inst_id = ignore_case_get(kwargs, "vnfInstId")
103 logger.info("Enter %s, method is %s, ", fun_name(), request.method)
104 logger.info("vnfInstId is %s, data is %s", vnf_inst_id, request.data)
106 vnf = VnfRegModel.objects.filter(id=vnf_inst_id)
108 err_msg = "Vnf(%s) does not exist." % vnf_inst_id
109 return Response(data={'error': err_msg}, status=status.HTTP_404_NOT_FOUND)
110 if request.method == 'GET':
112 "vnfInstId": vnf_inst_id,
115 "username": vnf[0].username,
116 "password": vnf[0].password
118 responseSerializer = VnfInfoSerializer(data=resp)
119 if not responseSerializer.is_valid():
120 raise Exception(responseSerializer.errors)
121 ret = responseSerializer.data
122 normal_status = status.HTTP_200_OK
123 elif request.method == 'PUT':
124 if not request_isValid:
125 raise Exception(requestSerializer.errors)
127 requestData = requestSerializer.data
128 ip = ignore_case_get(requestData, "ip")
129 port = ignore_case_get(requestData, "port")
130 username = ignore_case_get(requestData, "username")
131 password = ignore_case_get(requestData, "password")
137 vnf[0].username = username
139 vnf[0].password = password
142 normal_status = status.HTTP_202_ACCEPTED
146 normal_status = status.HTTP_204_NO_CONTENT
147 except Exception as e:
148 errorData = handler_exception(e)
149 return Response(data=errorData, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
150 return Response(data=ret, status=normal_status)
153 @swagger_auto_schema(method='post',
154 request_body=VnfConfigSerializer(),
156 202: NoneSerializer(),
157 500: ErrorSerializer()})
158 @api_view(http_method_names=['POST'])
159 def vnf_config(request, *args, **kwargs):
160 logger.info("Enter %s, data is %s", fun_name(), request.data)
161 requestSerializer = VnfConfigSerializer(data=request.data)
162 request_isValid = requestSerializer.is_valid()
164 if not request_isValid:
165 raise Exception(requestSerializer.errors)
167 requestData = requestSerializer.data
168 vnf_inst_id = ignore_case_get(requestData, "vnfInstanceId")
169 vnf = VnfRegModel.objects.filter(id=vnf_inst_id)
171 raise Exception("Vnf(%s) does not exist." % vnf_inst_id)
172 ret = restcall.call_req(
173 base_url="http://%s:%s/" % (vnf[0].ip, vnf[0].port),
174 user=vnf[0].username,
175 passwd=vnf[0].password,
176 auth_type=restcall.rest_no_auth,
177 resource="v1/vnfconfig",
179 content=json.dumps(requestData))
181 raise Exception("Failed to config Vnf(%s): %s" % (vnf_inst_id, ret[1]))
182 except Exception as e:
183 errorData = handler_exception(e)
184 return Response(data=errorData, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
185 return Response(data={}, status=status.HTTP_202_ACCEPTED)