lcm/lcm/nf/biz/query_subscription.py

   1 # Copyright (C) 2018 Verizon. All Rights Reserved
   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
  19 from lcm.pub.database.models import SubscriptionModel
  20 from lcm.pub.exceptions import NFLCMException
  21
  22 logger = logging.getLogger(__name__)
  23 ROOT_FILTERS = {
  24     'operationTypes': 'operation_types',
  25     'operationStates': 'operation_states',
  26     'notificationTypes': 'notification_types'
  27 }
  28 VNF_INSTANCE_FILTERS = {
  29     "vnfInstanceId": "vnf_instance_filter"
  30 }
  31
  32
  33 class QuerySubscription:
  34     def __init__(self, data, subscription_id=''):
  35         self.subscription_id = subscription_id
  36         self.params = data
  37
  38     def query_single_subscription(self):
  39         subscription = SubscriptionModel.objects.filter(subscription_id=self.subscription_id)
  40         if not subscription.exists():
  41             raise NFLCMException('Subscription(%s) does not exist' % self.subscription_id)
  42         return self.fill_resp_data(subscription)
  43
  44     def query_multi_subscriptions(self):
  45         query_data = {}
  46         logger.debug("QueryMultiSubscriptions--get--biz::> Check for filters in query params" % self.params)
  47         for query, value in list(self.params.items()):
  48             if query in ROOT_FILTERS:
  49                 query_data[ROOT_FILTERS[query] + '__icontains'] = value
  50         for query, value in list(self.params.items()):
  51             if query in VNF_INSTANCE_FILTERS:
  52                 query_data[VNF_INSTANCE_FILTERS[query] + '__icontains'] = value
  53         # Query the database with filters if the request has fields in request params, else fetch all records
  54         if query_data:
  55             subscriptions = SubscriptionModel.objects.filter(**query_data)
  56         else:
  57             subscriptions = SubscriptionModel.objects.all()
  58         if not subscriptions.exists():
  59             raise NFLCMException('Subscriptions do not exist')
  60         return [self.fill_resp_data(subscription) for subscription in subscriptions]
  61
  62     def fill_resp_data(self, subscription):
  63         subscription_filter = {
  64             "notificationTypes": ast.literal_eval(subscription.notification_types),
  65             "operationTypes": ast.literal_eval(subscription.operation_types),
  66             "operationStates": ast.literal_eval(subscription.operation_states),
  67             "vnfInstanceSubscriptionFilter": json.loads(subscription.vnf_instance_filter)
  68         }
  69         resp_data = {
  70             'id': subscription.subscription_id,
  71             'callbackUri': subscription.callback_uri,
  72             'filter': subscription_filter,
  73             '_links': json.loads(subscription.links)
  74         }
  75         return resp_data