Fix RANGE to HTTP_RANGE in request.META
[vfc/nfvo/catalog.git] / catalog / packages / views / ns_descriptor_views.py
index eb673ab..73388ee 100644 (file)
@@ -1,4 +1,4 @@
-# Copyright 2017 ZTE Corporation.
+# Copyright 2018 ZTE Corporation.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # limitations under the License.
 
 import logging
-import traceback
 
+from django.http import StreamingHttpResponse
 from drf_yasg.utils import no_body, swagger_auto_schema
 from rest_framework import status
 from rest_framework.decorators import api_view
 from rest_framework.response import Response
 
-from catalog.packages.biz.ns_descriptor import create, query_multiple, query_single
-from catalog.packages.serializers.create_nsd_info_request import \
-    CreateNsdInfoRequestSerializer
+from catalog.packages.biz.ns_descriptor import NsDescriptor
+from catalog.packages.serializers.create_nsd_info_request import CreateNsdInfoRequestSerializer
 from catalog.packages.serializers.nsd_info import NsdInfoSerializer
 from catalog.packages.serializers.nsd_infos import NsdInfosSerializer
+from catalog.packages.views.common import validate_data
 from catalog.pub.exceptions import CatalogException
+from .common import view_safe_call_with_log
 
 logger = logging.getLogger(__name__)
 
-"""
+
 @swagger_auto_schema(
+    method='GET',
+    operation_description="Query a NSD",
+    request_body=no_body,
     responses={
-        # status.HTTP_200_OK: Serializer(),
+        status.HTTP_200_OK: NsdInfoSerializer(),
+        status.HTTP_404_NOT_FOUND: 'NSDs do not exist',
         status.HTTP_500_INTERNAL_SERVER_ERROR: "Internal error"
     }
 )
-"""
-
-
 @swagger_auto_schema(
-    method='GET',
-    operation_description="Query an individual NS descriptor resource",
+    method='DELETE',
+    operation_description="Delete a NSD",
     request_body=no_body,
     responses={
-        status.HTTP_200_OK: NsdInfoSerializer(),
+        status.HTTP_204_NO_CONTENT: "No content",
         status.HTTP_500_INTERNAL_SERVER_ERROR: "Internal error"
     }
 )
-@api_view(http_method_names=['GET'])
-def ns_info_rd(request, nsdInfoId):
+@api_view(http_method_names=['GET', 'DELETE'])
+@view_safe_call_with_log(logger=logger)
+def ns_info_rd(request, **kwargs):
+    nsd_info_id = kwargs.get("nsdInfoId")
     if request.method == 'GET':
-        try:
-            data = query_single(nsdInfoId)
-            nsd_info = NsdInfoSerializer(data=data)
-            if not nsd_info.is_valid():
-                raise CatalogException
-            return Response(data=nsd_info.data, status=status.HTTP_200_OK)
-        except CatalogException:
-            logger.error(traceback.format_exc())
-            return Response(
-                data={'error': 'Query of an individual NS descriptor resource failed.'},
-                status=status.HTTP_500_INTERNAL_SERVER_ERROR
-            )
+        data = NsDescriptor().query_single(nsd_info_id)
+        nsd_info = validate_data(data, NsdInfoSerializer)
+        return Response(data=nsd_info.data, status=status.HTTP_200_OK)
+    if request.method == 'DELETE':
+        NsDescriptor().delete_single(nsd_info_id)
+        return Response(status=status.HTTP_204_NO_CONTENT)
 
 
 @swagger_auto_schema(
     method='POST',
-    operation_description="Create an individual NS descriptor resource",
+    operation_description="Create a NSD",
     request_body=CreateNsdInfoRequestSerializer(),
     responses={
         status.HTTP_201_CREATED: NsdInfoSerializer(),
@@ -76,7 +74,7 @@ def ns_info_rd(request, nsdInfoId):
 )
 @swagger_auto_schema(
     method='GET',
-    operation_description="Query multiple NS descriptor resources",
+    operation_description="Query multiple NSDs",
     request_body=no_body,
     responses={
         status.HTTP_200_OK: NsdInfosSerializer(),
@@ -84,31 +82,58 @@ def ns_info_rd(request, nsdInfoId):
     }
 )
 @api_view(http_method_names=['POST', 'GET'])
-def ns_descriptors_rc(request, *args, **kwargs):
+@view_safe_call_with_log(logger=logger)
+def ns_descriptors_rc(request):
     if request.method == 'POST':
-        try:
-            create_nsd_info_requst = CreateNsdInfoRequestSerializer(data=request.data)
-            if not create_nsd_info_requst.is_valid():
-                raise CatalogException
-            data = create(create_nsd_info_requst.data)
-            nsd_info = NsdInfoSerializer(data=data)
-            if not nsd_info.is_valid():
-                raise CatalogException
-            return Response(data=nsd_info.data, status=status.HTTP_201_CREATED)
-        except CatalogException:
-            logger.error(traceback.format_exc())
-            return Response(data={'error': 'Creating nsd info failed.'}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+        create_nsd_info_request = validate_data(request.data, CreateNsdInfoRequestSerializer)
+        data = NsDescriptor().create(create_nsd_info_request.data)
+        nsd_info = validate_data(data, NsdInfoSerializer)
+        return Response(data=nsd_info.data, status=status.HTTP_201_CREATED)
 
     if request.method == 'GET':
+        nsdId = request.query_params.get("nsdId", None)
+        data = NsDescriptor().query_multiple(nsdId)
+        nsd_infos = validate_data(data, NsdInfosSerializer)
+        return Response(data=nsd_infos.data, status=status.HTTP_200_OK)
+
+
+@swagger_auto_schema(
+    method='PUT',
+    operation_description="Upload NSD content",
+    request_body=no_body,
+    responses={
+        status.HTTP_204_NO_CONTENT: 'PNFD file',
+        status.HTTP_500_INTERNAL_SERVER_ERROR: "Internal error"
+    }
+)
+@swagger_auto_schema(
+    method='GET',
+    operation_description="Download NSD content",
+    request_body=no_body,
+    responses={
+        status.HTTP_204_NO_CONTENT: "No content",
+        status.HTTP_404_NOT_FOUND: 'NSD does not exist.',
+        status.HTTP_500_INTERNAL_SERVER_ERROR: "Internal error"
+    }
+)
+@api_view(http_method_names=['PUT', 'GET'])
+@view_safe_call_with_log(logger=logger)
+def nsd_content_ru(request, **kwargs):
+    nsd_info_id = kwargs.get("nsdInfoId")
+    if request.method == 'PUT':
+        files = request.FILES.getlist('file')
         try:
-            data = query_multiple()
-            nsd_infos = NsdInfosSerializer(data=data)
-            if not nsd_infos.is_valid():
-                raise CatalogException
-            return Response(data=nsd_infos.data, status=status.HTTP_200_OK)
-        except CatalogException:
-            logger.error(traceback.format_exc())
-            return Response(
-                data={'error': 'Query of multiple NS descriptor resources failed.'},
-                status=status.HTTP_500_INTERNAL_SERVER_ERROR
-            )
+            local_file_name = NsDescriptor().upload(nsd_info_id, files[0])
+            NsDescriptor().parse_nsd_and_save(nsd_info_id, local_file_name)
+            return Response(data=None, status=status.HTTP_204_NO_CONTENT)
+        except CatalogException as e:
+            NsDescriptor().handle_upload_failed(nsd_info_id)
+            raise e
+        except Exception as e:
+            NsDescriptor().handle_upload_failed(nsd_info_id)
+            raise e
+
+    if request.method == 'GET':
+        file_range = request.META.get('HTTP_RANGE')
+        file_iterator = NsDescriptor().download(nsd_info_id, file_range)
+        return StreamingHttpResponse(file_iterator, status=status.HTTP_200_OK)