mgr/mgr/vnfreg/views.py

   1 # Copyright 2017 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
  15 import logging
  16 import json
  17 import traceback
  18
  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
  25
  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, \
  31     VnfConfigSerializer
  32
  33 logger = logging.getLogger(__name__)
  34
  35
  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
  42
  43
  44 class vnfmgr_addvnf(APIView):
  45     @swagger_auto_schema(request_body=VnfInfoSerializer(),
  46                          responses={
  47                              status.HTTP_201_CREATED: ResponseSerializer(),
  48                              status.HTTP_500_INTERNAL_SERVER_ERROR: 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()
  53         try:
  54             if not request_isValid:
  55                 raise Exception(requestSerializer.errors)
  56
  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)
  61             VnfRegModel(
  62                 id=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()
  67
  68             responseSerializer = ResponseSerializer(data={"vnfInstId": vnf_inst_id})
  69             isValid = responseSerializer.is_valid()
  70             if not isValid:
  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)
  75
  76         return Response(data=responseSerializer.data, status=status.HTTP_201_CREATED)
  77
  78
  79 @swagger_auto_schema(method='put',
  80                      request_body=VnfInfoSerializer(),
  81                      responses={
  82                          status.HTTP_202_ACCEPTED: NoneSerializer(),
  83                          status.HTTP_500_INTERNAL_SERVER_ERROR: ErrorSerializer()})
  84 @swagger_auto_schema(method='delete',
  85                      responses={
  86                          status.HTTP_204_NO_CONTENT: NoneSerializer(),
  87                          status.HTTP_500_INTERNAL_SERVER_ERROR: ErrorSerializer()})
  88 @swagger_auto_schema(methods=['get'],
  89                      manual_parameters=[
  90                          openapi.Parameter('test',
  91                                            openapi.IN_QUERY,
  92                                            "test manual param",
  93                                            type=openapi.TYPE_BOOLEAN
  94                                            ), ],
  95                      responses={
  96                          status.HTTP_200_OK: openapi.Response('response description', VnfInfoSerializer()),
  97                          status.HTTP_500_INTERNAL_SERVER_ERROR: 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)
 105     try:
 106         vnf = VnfRegModel.objects.filter(id=vnf_inst_id)
 107         if not vnf:
 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':
 111             resp = {
 112                 "vnfInstId": vnf_inst_id,
 113                 "ip": vnf[0].ip,
 114                 "port": vnf[0].port,
 115                 "username": vnf[0].username,
 116                 "password": vnf[0].password
 117             }
 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)
 126
 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")
 132             if ip:
 133                 vnf[0].ip = ip
 134             if port:
 135                 vnf[0].port = port
 136             if username:
 137                 vnf[0].username = username
 138             if password:
 139                 vnf[0].password = password
 140             vnf[0].save()
 141             ret = {}
 142             normal_status = status.HTTP_202_ACCEPTED
 143         else:
 144             vnf.delete()
 145             ret = {}
 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)
 151
 152
 153 @swagger_auto_schema(method='post',
 154                      request_body=VnfConfigSerializer(),
 155                      responses={
 156                          status.HTTP_202_ACCEPTED: NoneSerializer(),
 157                          status.HTTP_500_INTERNAL_SERVER_ERROR: 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()
 163     try:
 164         if not request_isValid:
 165             raise Exception(requestSerializer.errors)
 166
 167         requestData = requestSerializer.data
 168         vnf_inst_id = ignore_case_get(requestData, "vnfInstanceId")
 169         vnf = VnfRegModel.objects.filter(id=vnf_inst_id)
 170         if not vnf:
 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",
 178             method="POST",
 179             content=json.dumps(requestData))
 180         if ret[0] != 0:
 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)