lcm/ns/views/sol/subscriptions_view.py

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