From 66e44262b8eb996c06670dcededd899dd1cbd7dc Mon Sep 17 00:00:00 2001
From: Michal Jagiello <michal.jagiello@t-mobile.pl>
Date: Tue, 30 Nov 2021 08:25:09 +0000
Subject: Data provider release

Change-Id: Ia041a07152e8dabd87de05992d3670cbdc1ddaae
Issue-ID: INT-2010
Signed-off-by: Michal Jagiello <michal.jagiello@t-mobile.pl>
---
 onap_data_provider/resources/tenant_resource.py | 85 +++++++++++++++++++++++++
 1 file changed, 85 insertions(+)
 create mode 100644 onap_data_provider/resources/tenant_resource.py

(limited to 'onap_data_provider/resources/tenant_resource.py')

diff --git a/onap_data_provider/resources/tenant_resource.py b/onap_data_provider/resources/tenant_resource.py
new file mode 100644
index 0000000..13d003f
--- /dev/null
+++ b/onap_data_provider/resources/tenant_resource.py
@@ -0,0 +1,85 @@
+"""Tenant 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, Optional
+
+from onapsdk.aai.cloud_infrastructure import CloudRegion, Tenant  # type: ignore
+
+from .resource import Resource
+from onapsdk.exceptions import ResourceNotFound  # type: ignore
+
+
+class TenantResource(Resource):
+    """Tenant resource class.
+
+    Creates tenant.
+    """
+
+    def __init__(self, data: Dict[str, Any], cloud_region: CloudRegion) -> None:
+        """Tenant resource initialization.
+
+        Args:
+            data (Dict[str, Any]): Data needed to create tenant
+            cloud_region (CloudRegion): Cloud region for which tenant is going to be created
+
+        """
+        super().__init__(data)
+        self.cloud_region: CloudRegion = cloud_region
+        self._tenant: Optional[Tenant] = None
+
+    def create(self) -> None:
+        """Create tenant resource.
+
+        Add tenant to provided cloud region
+
+        """
+        if not self.exists:
+            self.cloud_region.add_tenant(
+                tenant_id=self.data["tenant-id"],
+                tenant_name=self.data["tenant-name"],
+                tenant_context=self.data.get("tenant-context"),
+            )
+
+    @property
+    def exists(self) -> bool:
+        """Determine if resource already exists or not.
+
+        Returns:
+            bool: True if object exists, False otherwise
+
+        """
+        return self.tenant is not None
+
+    @property
+    def tenant(self) -> Tenant:
+        """Tenant property.
+
+        Returns:
+            Tenant: Tenant object
+
+        """
+        if not self._tenant:
+            try:
+                self._tenant = self.cloud_region.get_tenant(self.data["tenant-id"])
+            except ResourceNotFound:
+                logging.error(
+                    "Tenant %s does not exist in %s cloud region",
+                    self.data["tenant-id"],
+                    self.cloud_region.cloud_region_id,
+                )
+                return None
+        return self._tenant
-- 
cgit 1.2.3-korg