From d88b50e139aebf12e3c4ae6d00e45a9e0b680c72 Mon Sep 17 00:00:00 2001
From: "ying.yunlong" <ying.yunlong@zte.com.cn>
Date: Wed, 7 Feb 2018 11:27:46 +0800
Subject: Add vfc-ztevnfmdriver notify auto-swagger

Change-Id: I52cf1f75d3c1f554d9f04c5461d7cdd1267727c3
Issue-ID: VFC-736
Signed-off-by: ying.yunlong <ying.yunlong@zte.com.cn>
---
 zte/vmanager/driver/interfaces/serializers.py | 15 +++++++++++---
 zte/vmanager/driver/interfaces/views.py       | 29 ++++++++++++++++++---------
 2 files changed, 31 insertions(+), 13 deletions(-)

(limited to 'zte/vmanager')

diff --git a/zte/vmanager/driver/interfaces/serializers.py b/zte/vmanager/driver/interfaces/serializers.py
index ed6658fe..bb7fc238 100644
--- a/zte/vmanager/driver/interfaces/serializers.py
+++ b/zte/vmanager/driver/interfaces/serializers.py
@@ -54,7 +54,10 @@ class VmlistSerializer(serializers.Serializer):
         max_length=255,
         required=False,
         allow_null=True)
-    VMIDlist = VMIDlistSerializer(help_text='VMIDlist', required=False, many=True)
+    VMIDlist = VMIDlistSerializer(
+        help_text='VMIDlist',
+        required=False,
+        many=True)
 
 
 class NotifyReqSerializer(serializers.Serializer):
@@ -88,7 +91,10 @@ class NotifyReqSerializer(serializers.Serializer):
         max_length=255,
         required=False,
         allow_null=True)
-    vmlist = VmlistSerializer(help_text='vmlist', required=False, many=True)
+    vmlist = VmlistSerializer(
+        help_text='vmlist',
+        required=False,
+        many=True)
     extension = serializers.DictField(
         help_text="extension",
         child=serializers.DictField(allow_null=True),
@@ -176,7 +182,10 @@ class HealReqSerializer(serializers.Serializer):
         max_length=255,
         required=False,
         allow_null=True)
-    affectedvm = AffectedvmSerializer(help_text='affectedvm', required=True, many=True)
+    affectedvm = AffectedvmSerializer(
+        help_text='affectedvm',
+        required=True,
+        many=True)
 
 
 class InstScaleHealRespSerializer(serializers.Serializer):
diff --git a/zte/vmanager/driver/interfaces/views.py b/zte/vmanager/driver/interfaces/views.py
index 76b010f1..0866301e 100644
--- a/zte/vmanager/driver/interfaces/views.py
+++ b/zte/vmanager/driver/interfaces/views.py
@@ -313,11 +313,19 @@ def grantvnf(request, *args, **kwargs):
 
 class Notify(APIView):
     @swagger_auto_schema(
-        request_body=NotifyReqSerializer()
+        request_body=NotifyReqSerializer(),
+        responses={
+            status.HTTP_200_OK: 'Successfully',
+            status.HTTP_500_INTERNAL_SERVER_ERROR: 'Internal error'
+        }
     )
     def post(self, request):
         try:
             logger.info("[%s]req_data = %s", fun_name(), request.data)
+            notifyReqSerializer = NotifyReqSerializer(data=request.data)
+            if not notifyReqSerializer.is_valid():
+                raise Exception(notifyReqSerializer.errors)
+
             notify_param_map = {
                 "NFVOID": "",
                 "VNFMID": "VNFMID",
@@ -331,7 +339,7 @@ class Notify(APIView):
                 "VMIDlist": "",
                 "VMUUID": ""
             }
-            data = mapping_conv(notify_param_map, request.data)
+            data = mapping_conv(notify_param_map, notifyReqSerializer.data)
             logger.info("[%s]data = %s", fun_name(), data)
 
             data["status"] = "result"
@@ -341,7 +349,7 @@ class Notify(APIView):
             data["affectedVirtualStorage"] = []
             data["affectedCp"] = []
 
-            extension = ignorcase_get(request.data, "extension")
+            extension = ignorcase_get(notifyReqSerializer.data, "extension")
             openo_notification = ignorcase_get(extension, "openo_notification")
             if openo_notification:
                 affectedvnfcs = ignorcase_get(openo_notification, "affectedVnfc")
@@ -349,10 +357,10 @@ class Notify(APIView):
                 affectedcps = ignorcase_get(openo_notification, "affectedCp")
                 vnfdmodule = ignorcase_get(openo_notification, "vnfdmodule")
             else:
-                affectedvnfcs = ignorcase_get(ignorcase_get(request.data, "extension"), "affectedvnfc")
-                affectedvls = ignorcase_get(ignorcase_get(request.data, "extension"), "affectedvl")
-                affectedcps = ignorcase_get(ignorcase_get(request.data, "extension"), "affectedcp")
-                vnfdmodule = ignorcase_get(ignorcase_get(request.data, "extension"), "vnfdmodule")
+                affectedvnfcs = ignorcase_get(ignorcase_get(notifyReqSerializer.data, "extension"), "affectedvnfc")
+                affectedvls = ignorcase_get(ignorcase_get(notifyReqSerializer.data, "extension"), "affectedvl")
+                affectedcps = ignorcase_get(ignorcase_get(notifyReqSerializer.data, "extension"), "affectedcp")
+                vnfdmodule = ignorcase_get(ignorcase_get(notifyReqSerializer.data, "extension"), "vnfdmodule")
 
             data["vnfdmodule"] = vnfdmodule
 
@@ -389,12 +397,13 @@ class Notify(APIView):
 
             logger.info("[%s]data = %s", fun_name(), ret)
             if ret[0] != 0:
-                return Response(data={'error': ret[1]}, status=ret[2])
+                raise Exception(ret[1])
+
             return Response(data=None, status=ret[2])
         except Exception as e:
-            logger.error("Error occurred in LCM notification.")
+            logger.error("Error occurred in LCM notification,error: %s", e.message)
             logger.error(traceback.format_exc())
-            raise e
+            return Response(data={'error': 'notify expection'}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
 
 
 class Scale(APIView):
-- 
cgit 1.2.3-korg