X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=catalog%2Fpackages%2Fbiz%2Fvnf_pkg_subscription.py;h=9876961ce3f7a2462aa39c50201930687c0c0ab5;hb=fe7fa3746a2c1085527b2355b59cab3dc37229e7;hp=d05936519d68251f8d0537e21bb7a2d73dbe6df8;hpb=011186aa2d7cb73592249f0ff7f472e75b9803fa;p=modeling%2Fetsicatalog.git diff --git a/catalog/packages/biz/vnf_pkg_subscription.py b/catalog/packages/biz/vnf_pkg_subscription.py index d059365..9876961 100644 --- a/catalog/packages/biz/vnf_pkg_subscription.py +++ b/catalog/packages/biz/vnf_pkg_subscription.py @@ -16,18 +16,18 @@ import ast import json import logging import os -import requests import uuid - from collections import Counter -from rest_framework import status + +import requests from requests.auth import HTTPBasicAuth +from rest_framework import status + from catalog.packages import const from catalog.pub.database.models import VnfPkgSubscriptionModel from catalog.pub.exceptions import VnfPkgSubscriptionException, \ VnfPkgDuplicateSubscriptionException, SubscriptionDoesNotExistsException from catalog.pub.utils.values import ignore_case_get -from catalog.pub.config.config import MSB_SERVICE_IP, MSB_SERVICE_PORT logger = logging.getLogger(__name__) @@ -45,6 +45,9 @@ def is_filter_type_equal(new_filter, existing_filter): class CreateSubscription(object): + """ + Create subscription info + """ def __init__(self, data): self.data = data self.filter = ignore_case_get(self.data, "filter", {}) @@ -59,6 +62,10 @@ class CreateSubscription(object): ignore_case_get(self.filter, "vnfProductsFromProviders", []) def check_callbackuri_connection(self): + """ + Check if the callback uri can access + :return: + """ logger.debug("SubscribeNotification-post::> Sending GET request " "to %s" % self.callback_uri) try: @@ -90,9 +97,13 @@ class CreateSubscription(object): ) def do_biz(self): + """ + Do business + :return: + """ self.subscription_id = str(uuid.uuid4()) - self.check_callbackuri_connection() self.check_valid_auth_info() + self.check_callbackuri_connection() self.check_valid() self.save_db() subscription = VnfPkgSubscriptionModel.objects.get( @@ -102,6 +113,10 @@ class CreateSubscription(object): return subscription.toDict() def check_valid_auth_info(self): + """ + Check if the Auth info is valid + :return: + """ logger.debug("SubscribeNotification--post::> Validating Auth " "details if provided") if self.authentication.get("paramsBasic", {}) and const.BASIC not in self.authentication.get("authType"): @@ -133,11 +148,15 @@ class CreateSubscription(object): links = json.loads(subscription.links) logger.error("Subscriptions has already exists with the same callbackUri and filter:%s" % links) raise VnfPkgDuplicateSubscriptionException( - "https://%s:%s/%s" % (MSB_SERVICE_IP, MSB_SERVICE_PORT, links["self"]["href"])) + "/%s" % (links["self"]["href"])) return True def save_db(self): + """ + Save the subscription info to DB + :return: + """ logger.debug("SubscribeNotification--post::> Saving the subscription " "%s to the database" % self.subscription_id) links = { @@ -160,7 +179,15 @@ class CreateSubscription(object): class QuerySubscription(object): + """ + The class for query subscription + """ def query_multi_subscriptions(self, params): + """ + Query subscriptions + :param params: + :return: + """ query_data = {} logger.debug("QuerySubscription--get--multi--subscriptions--biz::> Check " "for filter in query params %s" % params) @@ -177,6 +204,11 @@ class QuerySubscription(object): return [subscription.toDict() for subscription in subscriptions] def query_single_subscription(self, subscription_id): + """ + Query subscription by id + :param subscription_id: + :return: + """ logger.debug("QuerySingleSubscriptions--get--single--subscription--biz::> " "ID: %s" % subscription_id) @@ -189,6 +221,9 @@ class QuerySubscription(object): class TerminateSubscription(object): + """ + The class to terminate the subscription + """ def terminate(self, subscription_id): logger.debug("TerminateSubscriptions--delete--biz::> " "ID: %s" % subscription_id)