Data provider release

Change-Id: Ia041a07152e8dabd87de05992d3670cbdc1ddaae
Issue-ID: INT-2010
Signed-off-by: Michal Jagiello <michal.jagiello@t-mobile.pl>

1 files changed, 189 insertions, 0 deletions

diff --git a/onap_data_provider/resources/customer_resource.py b/onap_data_provider/resources/customer_resource.py
new file mode 100644
index 0000000..2bbb1ef
--- /dev/null
+++ b/onap_data_provider/resources/customer_resource.py
@@ -0,0 +1,189 @@
+"""Customer resource module."""
+"""
+   Copyright 2021 Deutsche Telekom AG
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.
+"""
+
+import logging
+from typing import Any, Dict
+
+from onapsdk.aai.business import Customer, ServiceSubscription  # type: ignore
+from onapsdk.aai.cloud_infrastructure import CloudRegion, Tenant  # type: ignore
+
+from onapsdk.sdc.service import Service  # type: ignore
+
+from .resource import Resource
+from onapsdk.exceptions import ResourceNotFound  # type: ignore
+
+
+class CustomerResource(Resource):
+    """Customer resource class.
+
+    Creates customer.
+    """
+
+    def __init__(self, data: Dict[str, Any]) -> None:
+        """Initialize customer resource.
+
+        Args:
+            data (Dict[str, Any]): Data needed to create resource.
+
+        """
+        super().__init__(data)
+        self._customer: Customer = None
+
+    def create(self) -> None:
+        """Create customer resource.
+
+        Create customer resource and all related resources.
+
+        """
+        logging.debug("Create Customer %s", self.data["global-customer-id"])
+        if not self.exists:
+            self._customer = Customer.create(
+                global_customer_id=self.data["global-customer-id"],
+                subscriber_name=self.data["subscriber-name"],
+                subscriber_type=self.data["subscriber-type"],
+            )
+
+        for service_subscription in self.data.get("service-subscriptions", []):
+            resource = CustomerResource.ServiceSubscriptionResource(
+                service_subscription, self._customer
+            )
+            resource.create()
+
+    @property
+    def exists(self) -> bool:
+        """Determine if resource already exists or not.
+
+        Returns:
+            bool: True if object exists, False otherwise
+
+        """
+        return self.customer is not None
+
+    @property
+    def customer(self) -> Customer:
+        """Access to customer property.
+
+        Customer property containing Customer object.
+
+        Returns:
+            Customer: Customer object
+
+        """
+        if not self._customer:
+            try:
+                self._customer = Customer.get_by_global_customer_id(
+                    self.data["global-customer-id"]
+                )
+            except ResourceNotFound:
+                logging.error(
+                    "Customer %s does not exist",
+                    self.data["global-customer-id"],
+                )
+                return None
+        return self._customer
+
+    class ServiceSubscriptionResource(Resource):
+        """Service subscription class.
+
+        Creates service subscription.
+        """
+
+        def __init__(self, data: Dict[str, str], customer: Customer) -> None:
+            """Initialize service subscription resource.
+
+            Args:
+                data (Dict[str, str]): Data needed to create resource.
+                customer (Customer): Related Customer object.
+
+            """
+            super().__init__(data)
+            self._service_subscription: ServiceSubscription = None
+            self._customer: Customer = customer
+
+        def create(self) -> None:
+            """Create Service subscription resource.
+
+            Create service subscription resource belonging to a customer.
+
+            """
+            logging.debug("Create ServiceSubscription %s", self.data["service-type"])
+            if not self.exists:
+                self._service_subscription = self._customer.subscribe_service(
+                    Service(self.data["service-type"])
+                )
+
+            for tenant_cloud_region_data in self.data.get("tenants", []):
+                try:
+                    cloud_region: CloudRegion = CloudRegion.get_by_id(
+                        tenant_cloud_region_data["cloud-owner"],
+                        tenant_cloud_region_data["cloud-region-id"],
+                    )
+                except ResourceNotFound:
+                    logging.error(
+                        f"Cloud region {tenant_cloud_region_data['cloud-owner']} {tenant_cloud_region_data['cloud-region-id']} does not exists"
+                    )
+                    continue
+                try:
+                    tenant: Tenant = cloud_region.get_tenant(
+                        tenant_cloud_region_data["tenant-id"]
+                    )
+                except ResourceNotFound:
+                    logging.error(
+                        f"Tenant {tenant_cloud_region_data['tenant-id']} does not exist"
+                    )
+                    continue
+
+                self.service_subscription.link_to_cloud_region_and_tenant(
+                    cloud_region, tenant
+                )
+                logging.debug(
+                    f"Service subscription linked to {tenant.name} tenant and {cloud_region.cloud_region_id} cloud region"
+                )
+
+        @property
+        def exists(self) -> bool:
+            """Determine if resource already exists or not.
+
+            Returns:
+                bool: True if object exists, False otherwise
+
+            """
+            return self.service_subscription is not None
+
+        @property
+        def service_subscription(self) -> ServiceSubscription:
+            """Get ServiceSubscription instance.
+
+            Get ServiceSubscription instance.
+
+            Returns:
+                ServiceSubscription: Created `ServiceSubscription` subclass instance.
+            """
+            if not self._service_subscription:
+                try:
+                    self._service_subscription = (
+                        self._customer.get_service_subscription_by_service_type(
+                            self.data["service-type"]
+                        )
+                    )
+                except ResourceNotFound:
+                    logging.error(
+                        "Service type %s does not exist",
+                        self.data["service-type"],
+                    )
+                    return None
+            return self._service_subscription