import logging
import json
+import traceback
+from drf_yasg import openapi
+from drf_yasg.utils import swagger_auto_schema
from rest_framework import status
from rest_framework.decorators import api_view
from rest_framework.response import Response
+from rest_framework.views import APIView
from mgr.pub.utils.values import ignore_case_get
from mgr.pub.utils.syscomm import fun_name
from mgr.pub.database.models import VnfRegModel
from mgr.pub.utils import restcall
+from mgr.vnfreg.serializers import ErrorSerializer, VnfInfoSerializer, ResponseSerializer, NoneSerializer, \
+ VnfConfigSerializer
logger = logging.getLogger(__name__)
-@api_view(http_method_names=['POST'])
-def add_vnf(request, *args, **kwargs):
- logger.info("Enter %s, data is %s", fun_name(), request.data)
- vnf_inst_id = ignore_case_get(request.data, "vnfInstId")
- try:
- if VnfRegModel.objects.filter(id=vnf_inst_id):
- raise Exception("Vnf(%s) already exists." % vnf_inst_id)
- VnfRegModel(
- id=vnf_inst_id,
- ip=ignore_case_get(request.data, "ip"),
- port=ignore_case_get(request.data, "port"),
- username=ignore_case_get(request.data, "username"),
- password=ignore_case_get(request.data, "password")).save()
- except Exception as e:
- logger.error(e.message)
- return Response(data={'error': e.message}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
- return Response(data={"vnfInstId": vnf_inst_id}, status=status.HTTP_201_CREATED)
+def handler_exception(e):
+ logger.error(e.message)
+ logger.error(traceback.format_exc())
+ errorSerializer = ErrorSerializer(data={'error': e.message})
+ errorSerializer.is_valid()
+ return errorSerializer.data
+
+
+class vnfmgr_addvnf(APIView):
+ @swagger_auto_schema(request_body=VnfInfoSerializer(),
+ responses={
+ 201: ResponseSerializer(),
+ 500: ErrorSerializer()})
+ def post(self, request):
+ logger.info("Enter %s, data is %s", fun_name(), request.data)
+ requestSerializer = VnfInfoSerializer(data=request.data)
+ request_isValid = requestSerializer.is_valid()
+ try:
+ if not request_isValid:
+ raise Exception(requestSerializer.errors)
+ requestData = requestSerializer.data
+ vnf_inst_id = ignore_case_get(requestData, "vnfInstId")
+ if VnfRegModel.objects.filter(id=vnf_inst_id):
+ raise Exception("Vnf(%s) already exists." % vnf_inst_id)
+ VnfRegModel(
+ id=vnf_inst_id,
+ ip=ignore_case_get(requestData, "ip"),
+ port=ignore_case_get(requestData, "port"),
+ username=ignore_case_get(requestData, "username"),
+ password=ignore_case_get(requestData, "password")).save()
+ responseSerializer = ResponseSerializer(data={"vnfInstId": vnf_inst_id})
+ isValid = responseSerializer.is_valid()
+ if not isValid:
+ raise Exception(responseSerializer.errors)
+ except Exception as e:
+ errorData = handler_exception(e)
+ return Response(data=errorData, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+
+ return Response(data=responseSerializer.data, status=status.HTTP_201_CREATED)
+
+
+@swagger_auto_schema(method='put',
+ request_body=VnfInfoSerializer(),
+ responses={
+ 202: NoneSerializer(),
+ 500: ErrorSerializer()})
+@swagger_auto_schema(method='delete',
+ responses={
+ 204: NoneSerializer(),
+ 500: ErrorSerializer()})
+@swagger_auto_schema(methods=['get'],
+ manual_parameters=[
+ openapi.Parameter('test',
+ openapi.IN_QUERY,
+ "test manual param",
+ type=openapi.TYPE_BOOLEAN
+ ), ],
+ responses={
+ 200: openapi.Response('response description', VnfInfoSerializer()),
+ 500: ErrorSerializer()})
@api_view(http_method_names=['GET', 'PUT', 'DELETE'])
def access_vnf(request, *args, **kwargs):
+ requestSerializer = VnfInfoSerializer(data=request.data)
+ request_isValid = requestSerializer.is_valid()
vnf_inst_id = ignore_case_get(kwargs, "vnfInstId")
logger.info("Enter %s, method is %s, ", fun_name(), request.method)
logger.info("vnfInstId is %s, data is %s", vnf_inst_id, request.data)
- # ret, normal_status = None, None
try:
vnf = VnfRegModel.objects.filter(id=vnf_inst_id)
if not vnf:
err_msg = "Vnf(%s) does not exist." % vnf_inst_id
return Response(data={'error': err_msg}, status=status.HTTP_404_NOT_FOUND)
if request.method == 'GET':
- ret = {
+ resp = {
"vnfInstId": vnf_inst_id,
"ip": vnf[0].ip,
"port": vnf[0].port,
"username": vnf[0].username,
"password": vnf[0].password
}
+ responseSerializer = VnfInfoSerializer(data=resp)
+ if not responseSerializer.is_valid():
+ raise Exception(responseSerializer.errors)
+ ret = responseSerializer.data
normal_status = status.HTTP_200_OK
elif request.method == 'PUT':
- ip = ignore_case_get(request.data, "ip")
- port = ignore_case_get(request.data, "port")
- username = ignore_case_get(request.data, "username")
- password = ignore_case_get(request.data, "password")
+ if not request_isValid:
+ raise Exception(requestSerializer.errors)
+
+ requestData = requestSerializer.data
+ ip = ignore_case_get(requestData, "ip")
+ port = ignore_case_get(requestData, "port")
+ username = ignore_case_get(requestData, "username")
+ password = ignore_case_get(requestData, "password")
if ip:
vnf[0].ip = ip
if port:
ret = {}
normal_status = status.HTTP_204_NO_CONTENT
except Exception as e:
- logger.error(e.message)
- return Response(data={'error': e.message}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+ errorData = handler_exception(e)
+ return Response(data=errorData, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
return Response(data=ret, status=normal_status)
+@swagger_auto_schema(method='post',
+ request_body=VnfConfigSerializer(),
+ responses={
+ 202: NoneSerializer(),
+ 500: ErrorSerializer()})
@api_view(http_method_names=['POST'])
def vnf_config(request, *args, **kwargs):
logger.info("Enter %s, data is %s", fun_name(), request.data)
- vnf_inst_id = ignore_case_get(request.data, "vnfInstanceId")
+ requestSerializer = VnfConfigSerializer(data=request.data)
+ request_isValid = requestSerializer.is_valid()
try:
+ if not request_isValid:
+ raise Exception(requestSerializer.errors)
+
+ requestData = requestSerializer.data
+ vnf_inst_id = ignore_case_get(requestData, "vnfInstanceId")
vnf = VnfRegModel.objects.filter(id=vnf_inst_id)
if not vnf:
raise Exception("Vnf(%s) does not exist." % vnf_inst_id)
auth_type=restcall.rest_no_auth,
resource="v1/vnfconfig",
method="POST",
- content=json.dumps(request.data))
+ content=json.dumps(requestData))
if ret[0] != 0:
raise Exception("Failed to config Vnf(%s): %s" % (vnf_inst_id, ret[1]))
except Exception as e:
- logger.error(e.message)
- return Response(data={'error': e.message}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+ errorData = handler_exception(e)
+ return Response(data=errorData, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
return Response(data={}, status=status.HTTP_202_ACCEPTED)
Code Review / vfc / gvnfm / vnfmgr.git / blobdiff