1 # Copyright (C) 2018 Verizon. All Rights Reserved
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
7 # http://www.apache.org/licenses/LICENSE-2.0
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.
19 from lcm.pub.database.models import SubscriptionModel
20 from lcm.pub.exceptions import NFLCMException
22 logger = logging.getLogger(__name__)
24 'operationTypes': 'operation_types',
25 'operationStates': 'operation_states',
26 'notificationTypes': 'notification_types'
28 VNF_INSTANCE_FILTERS = {
29 "vnfInstanceId": "vnf_instance_filter"
33 class QuerySubscription:
34 def __init__(self, data, subscription_id=''):
35 self.subscription_id = subscription_id
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)
44 def query_multi_subscriptions(self):
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
55 subscriptions = SubscriptionModel.objects.filter(**query_data)
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]
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)
70 'id': subscription.subscription_id,
71 'callbackUri': subscription.callback_uri,
72 'filter': subscription_filter,
73 '_links': json.loads(subscription.links)