improve code coverage rate (scale vnf to level) after vfclcm upgraded from python2...
[vfc/gvnfm/vnflcm.git] / lcm / lcm / nf / views / subscriptions_view.py
1 # Copyright (C) 2018 Verizon. All Rights Reserved\r
2 #\r
3 # Licensed under the Apache License, Version 2.0 (the "License");\r
4 # you may not use this file except in compliance with the License.\r
5 # You may obtain a copy of the License at\r
6 #\r
7 #         http://www.apache.org/licenses/LICENSE-2.0\r
8 #\r
9 # Unless required by applicable law or agreed to in writing, software\r
10 # distributed under the License is distributed on an "AS IS" BASIS,\r
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
12 # See the License for the specific language governing permissions and\r
13 # limitations under the License.\r
14 \r
15 import ast\r
16 import json\r
17 import logging\r
18 \r
19 from drf_yasg.utils import swagger_auto_schema\r
20 from rest_framework import status\r
21 from rest_framework.response import Response\r
22 from rest_framework.views import APIView\r
23 \r
24 from lcm.nf.serializers.lccn_subscription_request import LccnSubscriptionRequestSerializer\r
25 from lcm.nf.serializers.lccn_subscription import LccnSubscriptionSerializer\r
26 from lcm.nf.serializers.lccn_subscriptions import LccnSubscriptionsSerializer\r
27 from lcm.nf.serializers.response import ProblemDetailsSerializer\r
28 from lcm.pub.exceptions import NFLCMException, NFLCMExceptionBadRequest\r
29 from lcm.nf.biz.create_subscription import CreateSubscription\r
30 from lcm.nf.biz.query_subscription import QuerySubscription\r
31 from lcm.nf.biz.delete_subscription import DeleteSubscription\r
32 from .common import view_safe_call_with_log\r
33 \r
34 logger = logging.getLogger(__name__)\r
35 VALID_FILTERS = ["operationTypes", "operationStates", "notificationTypes", "vnfInstanceId"]\r
36 \r
37 \r
38 def get_problem_details_serializer(status_code, error_message):\r
39     problem_details = {\r
40         "status": status_code,\r
41         "detail": error_message\r
42     }\r
43     problem_details_serializer = ProblemDetailsSerializer(data=problem_details)\r
44     problem_details_serializer.is_valid()\r
45     return problem_details_serializer\r
46 \r
47 \r
48 class SubscriptionsView(APIView):\r
49     @swagger_auto_schema(\r
50         request_body=LccnSubscriptionRequestSerializer(),\r
51         responses={\r
52             status.HTTP_201_CREATED: LccnSubscriptionSerializer(),\r
53             status.HTTP_303_SEE_OTHER: ProblemDetailsSerializer(),\r
54             status.HTTP_400_BAD_REQUEST: ProblemDetailsSerializer(),\r
55             status.HTTP_500_INTERNAL_SERVER_ERROR: ProblemDetailsSerializer()\r
56         }\r
57     )\r
58     @view_safe_call_with_log(logger=logger)\r
59     def post(self, request):\r
60         logger.debug("SubscribeNotification--post::> %s" % request.data)\r
61 \r
62         lccn_subscription_request_serializer = LccnSubscriptionRequestSerializer(data=request.data)\r
63         if not lccn_subscription_request_serializer.is_valid():\r
64             raise NFLCMExceptionBadRequest(lccn_subscription_request_serializer.errors)\r
65         subscription = CreateSubscription(\r
66             request.data).do_biz()\r
67         lccn_notifications_filter = {\r
68             "notificationTypes": ast.literal_eval(subscription.notification_types),\r
69             "operationTypes": ast.literal_eval(subscription.operation_types),\r
70             "operationStates": ast.literal_eval(subscription.operation_states),\r
71             "vnfInstanceSubscriptionFilter": json.loads(subscription.vnf_instance_filter)\r
72         }\r
73         subscription_data = {\r
74             "id": subscription.subscription_id,\r
75             "callbackUri": subscription.callback_uri,\r
76             "_links": json.loads(subscription.links),\r
77             "filter": lccn_notifications_filter\r
78         }\r
79         sub_resp_serializer = LccnSubscriptionSerializer(data=subscription_data)\r
80         if not sub_resp_serializer.is_valid():\r
81             raise NFLCMException(sub_resp_serializer.errors)\r
82         return Response(data=subscription_data, status=status.HTTP_201_CREATED)\r
83 \r
84     @swagger_auto_schema(\r
85         responses={\r
86             status.HTTP_200_OK: LccnSubscriptionsSerializer(),\r
87             status.HTTP_400_BAD_REQUEST: ProblemDetailsSerializer(),\r
88             status.HTTP_500_INTERNAL_SERVER_ERROR: ProblemDetailsSerializer()\r
89         }\r
90     )\r
91     @view_safe_call_with_log(logger=logger)\r
92     def get(self, request):\r
93         logger.debug("SubscribeNotification--get::> %s" % request.query_params)\r
94 \r
95         if request.query_params and not set(request.query_params).issubset(set(VALID_FILTERS)):\r
96             problem_details_serializer = get_problem_details_serializer(\r
97                 status.HTTP_400_BAD_REQUEST,\r
98                 "Not a valid filter"\r
99             )\r
100             return Response(data=problem_details_serializer.data,\r
101                             status=status.HTTP_400_BAD_REQUEST)\r
102         resp_data = QuerySubscription(request.query_params).query_multi_subscriptions()\r
103 \r
104         subscriptions_serializer = LccnSubscriptionsSerializer(data=resp_data)\r
105         if not subscriptions_serializer.is_valid():\r
106             raise NFLCMException(subscriptions_serializer.errors)\r
107 \r
108         logger.debug("SubscribeNotification--get::> Remove default fields")\r
109         return Response(data=resp_data, status=status.HTTP_200_OK)\r
110 \r
111 \r
112 class SubscriptionDetailView(APIView):\r
113     @swagger_auto_schema(\r
114         responses={\r
115             status.HTTP_200_OK: LccnSubscriptionSerializer(),\r
116             status.HTTP_404_NOT_FOUND: ProblemDetailsSerializer(),\r
117             status.HTTP_500_INTERNAL_SERVER_ERROR: ProblemDetailsSerializer()\r
118         }\r
119     )\r
120     @view_safe_call_with_log(logger=logger)\r
121     def get(self, request, subscriptionid):\r
122         logger.debug("SubscriptionDetailView--get::> %s" % subscriptionid)\r
123 \r
124         resp_data = QuerySubscription(\r
125             subscription_id=subscriptionid\r
126         ).query_single_subscription()\r
127 \r
128         subscription_serializer = LccnSubscriptionSerializer(data=resp_data)\r
129         if not subscription_serializer.is_valid():\r
130             raise NFLCMException(subscription_serializer.errors)\r
131 \r
132         return Response(data=resp_data, status=status.HTTP_200_OK)\r
133 \r
134     @swagger_auto_schema(\r
135         responses={\r
136             status.HTTP_204_NO_CONTENT: "",\r
137             status.HTTP_404_NOT_FOUND: ProblemDetailsSerializer(),\r
138             status.HTTP_500_INTERNAL_SERVER_ERROR: ProblemDetailsSerializer()\r
139         }\r
140     )\r
141     @view_safe_call_with_log(logger=logger)\r
142     def delete(self, request, subscriptionid):\r
143         logger.debug("SubscriptionDetailView--delete::> %s" % subscriptionid)\r
144 \r
145         DeleteSubscription(\r
146             subscription_id=subscriptionid\r
147         ).delete_single_subscription()\r
148 \r
149         return Response(status=status.HTTP_204_NO_CONTENT)\r