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 import traceback\r
  19 \r
  20 from drf_yasg.utils import swagger_auto_schema\r
  21 from lcm.nf.biz.create_subscription import CreateSubscription\r
  22 from lcm.nf.biz.query_subscription import QuerySubscription\r
  23 from rest_framework import status\r
  24 from rest_framework.response import Response\r
  25 from rest_framework.views import APIView\r
  26 \r
  27 from lcm.nf.serializers.lccn_subscription_request import LccnSubscriptionRequestSerializer\r
  28 from lcm.nf.serializers.lccn_subscription import LccnSubscriptionSerializer\r
  29 from lcm.nf.serializers.lccn_subscriptions import LccnSubscriptionsSerializer\r
  30 from lcm.nf.serializers.response import ProblemDetailsSerializer\r
  31 from lcm.pub.exceptions import NFLCMException\r
  32 \r
  33 logger = logging.getLogger(__name__)\r
  34 VALID_FILTERS = ["operationTypes", "operationStates", "notificationTypes", "vnfInstanceId"]\r
  35 \r
  36 \r
  37 def get_problem_details_serializer(status_code, error_message):\r
  38     problem_details = {\r
  39         "status": status_code,\r
  40         "detail": error_message\r
  41     }\r
  42     problem_details_serializer = ProblemDetailsSerializer(data=problem_details)\r
  43     problem_details_serializer.is_valid()\r
  44     return problem_details_serializer\r
  45 \r
  46 \r
  47 class SubscriptionsView(APIView):\r
  48     @swagger_auto_schema(\r
  49         request_body=LccnSubscriptionRequestSerializer(),\r
  50         responses={\r
  51             status.HTTP_201_CREATED: LccnSubscriptionSerializer(),\r
  52             status.HTTP_303_SEE_OTHER: ProblemDetailsSerializer(),\r
  53             status.HTTP_500_INTERNAL_SERVER_ERROR: ProblemDetailsSerializer()\r
  54         }\r
  55     )\r
  56     def post(self, request):\r
  57         logger.debug("SubscribeNotification--post::> %s" % request.data)\r
  58         try:\r
  59             lccn_subscription_request_serializer = LccnSubscriptionRequestSerializer(data=request.data)\r
  60             if not lccn_subscription_request_serializer.is_valid():\r
  61                 raise NFLCMException(lccn_subscription_request_serializer.errors)\r
  62             subscription = CreateSubscription(\r
  63                 lccn_subscription_request_serializer.data).do_biz()\r
  64             lccn_notifications_filter = {\r
  65                 "notificationTypes": ast.literal_eval(subscription.notification_types),\r
  66                 "operationTypes": ast.literal_eval(subscription.operation_types),\r
  67                 "operationStates": ast.literal_eval(subscription.operation_states),\r
  68                 "vnfInstanceSubscriptionFilter": json.loads(subscription.vnf_instance_filter)\r
  69             }\r
  70             subscription_data = {\r
  71                 "id": subscription.subscription_id,\r
  72                 "callbackUri": subscription.callback_uri,\r
  73                 "_links": json.loads(subscription.links),\r
  74                 "filter": lccn_notifications_filter\r
  75             }\r
  76             sub_resp_serializer = LccnSubscriptionSerializer(data=subscription_data)\r
  77             if not sub_resp_serializer.is_valid():\r
  78                 raise NFLCMException(sub_resp_serializer.errors)\r
  79             return Response(data=sub_resp_serializer.data, status=status.HTTP_201_CREATED)\r
  80         except NFLCMException as e:\r
  81             logger.error(e.message)\r
  82             if "exists" in e.message:\r
  83                 return Response(data={'error': '%s' % e.message}, status=status.HTTP_303_SEE_OTHER)\r
  84             return Response(data={'error': '%s' % e.message}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)\r
  85         except Exception as e:\r
  86             logger.error(e.message)\r
  87             logger.error(traceback.format_exc())\r
  88             return Response(data={'error': e.message}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)\r
  89 \r
  90     @swagger_auto_schema(\r
  91         responses={\r
  92             status.HTTP_200_OK: LccnSubscriptionsSerializer(),\r
  93             status.HTTP_400_BAD_REQUEST: ProblemDetailsSerializer(),\r
  94             status.HTTP_500_INTERNAL_SERVER_ERROR: ProblemDetailsSerializer()\r
  95         }\r
  96     )\r
  97     def get(self, request):\r
  98         logger.debug("SubscribeNotification--get::> %s" % request.query_params)\r
  99         try:\r
 100             if request.query_params and not set(request.query_params).issubset(set(VALID_FILTERS)):\r
 101                 problem_details_serializer = get_problem_details_serializer(status.HTTP_400_BAD_REQUEST, "Not a valid filter")\r
 102                 return Response(data=problem_details_serializer.data, status=status.HTTP_400_BAD_REQUEST)\r
 103             resp_data = QuerySubscription(request.query_params).query_multi_subscriptions()\r
 104 \r
 105             subscriptions_serializer = LccnSubscriptionsSerializer(data=resp_data)\r
 106             if not subscriptions_serializer.is_valid():\r
 107                 raise NFLCMException(subscriptions_serializer.errors)\r
 108 \r
 109             logger.debug("SubscribeNotification--get::> Remove default fields if exclude_default" +\r
 110                          " is specified")\r
 111             return Response(data=subscriptions_serializer.data, status=status.HTTP_200_OK)\r
 112         except NFLCMException as e:\r
 113             logger.error(e.message)\r
 114             problem_details_serializer = get_problem_details_serializer(status.HTTP_500_INTERNAL_SERVER_ERROR, traceback.format_exc())\r
 115             return Response(data=problem_details_serializer.data, status=status.HTTP_500_INTERNAL_SERVER_ERROR)\r
 116 \r
 117         except Exception as e:\r
 118             logger.error(e.message)\r
 119             logger.error(traceback.format_exc())\r
 120             problem_details_serializer = get_problem_details_serializer(status.HTTP_500_INTERNAL_SERVER_ERROR, traceback.format_exc())\r
 121             return Response(data=problem_details_serializer.data, status=status.HTTP_500_INTERNAL_SERVER_ERROR)\r