code refactor for genericparser
[modeling/etsicatalog.git] / genericparser / packages / views / vnf_package_views.py
1 # Copyright 2018 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
17 from django.http import StreamingHttpResponse
18 from drf_yasg.utils import swagger_auto_schema, no_body
19 from rest_framework import status
20 from rest_framework.decorators import api_view
21 from rest_framework.response import Response
22 from genericparser.packages.serializers.upload_vnf_pkg_from_uri_req import UploadVnfPackageFromUriRequestSerializer
23 from genericparser.packages.serializers.create_vnf_pkg_info_req import CreateVnfPkgInfoRequestSerializer
24 from genericparser.packages.serializers.vnf_pkg_info import VnfPkgInfoSerializer
25 from genericparser.packages.serializers.vnf_pkg_infos import VnfPkgInfosSerializer
26 from genericparser.packages.biz.vnf_package import VnfPackage
27 from genericparser.packages.biz.vnf_package import VnfPkgUploadThread
28 from genericparser.packages.biz.vnf_package import parse_vnfd_and_save
29 from genericparser.packages.biz.vnf_package import handle_upload_failed
30 from .common import validate_data
31 from .common import view_safe_call_with_log
32
33 logger = logging.getLogger(__name__)
34
35
36 @swagger_auto_schema(
37     method="GET",
38     operation_description="Query multiple VNF package resource",
39     request_body=no_body,
40     responses={
41         status.HTTP_200_OK: VnfPkgInfosSerializer(),
42         status.HTTP_500_INTERNAL_SERVER_ERROR: "Internal error"
43     }
44 )
45 @swagger_auto_schema(
46     method="POST",
47     operation_description="Create an individual VNF package resource",
48     request_body=CreateVnfPkgInfoRequestSerializer,
49     responses={
50         status.HTTP_201_CREATED: VnfPkgInfoSerializer(),
51         status.HTTP_500_INTERNAL_SERVER_ERROR: "Internal error"
52     }
53 )
54 @api_view(http_method_names=["GET", "POST"])
55 @view_safe_call_with_log(logger=logger)
56 def vnf_packages_rc(request):
57     if request.method == 'GET':
58         logger.debug("Query VNF packages> %s" % request.data)
59         data = VnfPackage().query_multiple()
60         vnf_pkg_infos = validate_data(data, VnfPkgInfosSerializer)
61         return Response(data=vnf_pkg_infos.data, status=status.HTTP_200_OK)
62
63     if request.method == 'POST':
64         logger.debug("Create VNF package> %s" % request.data)
65         create_vnf_pkg_info_request = validate_data(request.data,
66                                                     CreateVnfPkgInfoRequestSerializer)
67         data = VnfPackage().create_vnf_pkg(create_vnf_pkg_info_request.data)
68         vnf_pkg_info = validate_data(data, VnfPkgInfoSerializer)
69         return Response(data=vnf_pkg_info.data, status=status.HTTP_201_CREATED)
70
71
72 @swagger_auto_schema(
73     method='PUT',
74     operation_description="Upload VNF package content",
75     request_body=no_body,
76     responses={
77         status.HTTP_202_ACCEPTED: "Successfully",
78         status.HTTP_500_INTERNAL_SERVER_ERROR: "Internal error"
79     }
80 )
81 @swagger_auto_schema(
82     method="GET",
83     operation_description="Fetch VNF package content",
84     request_body=no_body,
85     responses={
86         status.HTTP_200_OK: VnfPkgInfosSerializer(),
87         status.HTTP_404_NOT_FOUND: "VNF package does not exist",
88         status.HTTP_500_INTERNAL_SERVER_ERROR: "Internal error"
89     }
90 )
91 @api_view(http_method_names=["PUT", "GET"])
92 @view_safe_call_with_log(logger=logger)
93 def package_content_ru(request, **kwargs):
94     vnf_pkg_id = kwargs.get("vnfPkgId")
95     if request.method == "PUT":
96         logger.debug("Upload VNF package %s" % vnf_pkg_id)
97         files = request.FILES.getlist('file')
98         try:
99             local_file_name = VnfPackage().upload(vnf_pkg_id, files[0])
100             parse_vnfd_and_save(vnf_pkg_id, local_file_name)
101             return Response(None, status=status.HTTP_202_ACCEPTED)
102         except Exception as e:
103             handle_upload_failed(vnf_pkg_id)
104             raise e
105
106     if request.method == "GET":
107         file_range = request.META.get('HTTP_RANGE')
108         file_iterator = VnfPackage().download(vnf_pkg_id, file_range)
109         return StreamingHttpResponse(file_iterator, status=status.HTTP_200_OK)
110
111
112 @swagger_auto_schema(
113     method='POST',
114     operation_description="Upload VNF package content from uri",
115     request_body=UploadVnfPackageFromUriRequestSerializer,
116     responses={
117         status.HTTP_202_ACCEPTED: "Successfully",
118         status.HTTP_500_INTERNAL_SERVER_ERROR: "Internal error"
119     }
120 )
121 @api_view(http_method_names=['POST'])
122 @view_safe_call_with_log(logger=logger)
123 def upload_from_uri_c(request, **kwargs):
124     vnf_pkg_id = kwargs.get("vnfPkgId")
125     try:
126         upload_vnf_from_uri_request = validate_data(request.data,
127                                                     UploadVnfPackageFromUriRequestSerializer)
128         VnfPkgUploadThread(upload_vnf_from_uri_request.data, vnf_pkg_id).start()
129         return Response(None, status=status.HTTP_202_ACCEPTED)
130     except Exception as e:
131         handle_upload_failed(vnf_pkg_id)
132         raise e
133
134
135 @swagger_auto_schema(
136     method='GET',
137     operation_description="Query an individual VNF package resource",
138     request_body=no_body,
139     responses={
140         status.HTTP_200_OK: VnfPkgInfoSerializer(),
141         status.HTTP_404_NOT_FOUND: "VNF package does not exist",
142         status.HTTP_500_INTERNAL_SERVER_ERROR: "Internal error"
143     }
144 )
145 @swagger_auto_schema(
146     method='DELETE',
147     operation_description="Delete an individual VNF package resource",
148     request_body=no_body,
149     responses={
150         status.HTTP_204_NO_CONTENT: "No content",
151         status.HTTP_500_INTERNAL_SERVER_ERROR: "Internal error"
152     }
153 )
154 @api_view(http_method_names=['GET', 'DELETE'])
155 @view_safe_call_with_log(logger=logger)
156 def vnf_package_rd(request, **kwargs):
157     vnf_pkg_id = kwargs.get("vnfPkgId")
158     if request.method == 'GET':
159         logger.debug("Query an individual VNF package> %s" % request.data)
160         data = VnfPackage().query_single(vnf_pkg_id)
161         vnf_pkg_info = validate_data(data, VnfPkgInfoSerializer)
162         return Response(data=vnf_pkg_info.data, status=status.HTTP_200_OK)
163
164     if request.method == 'DELETE':
165         logger.debug("Delete an individual VNF package> %s" % request.data)
166         VnfPackage().delete_vnf_pkg(vnf_pkg_id)
167         return Response(data=None, status=status.HTTP_204_NO_CONTENT)