summaryrefslogtreecommitdiffstats
path: root/onap_data_provider/resources/tenant_resource.py
diff options
context:
space:
mode:
Diffstat (limited to 'onap_data_provider/resources/tenant_resource.py')
-rw-r--r--onap_data_provider/resources/tenant_resource.py85
1 files changed, 85 insertions, 0 deletions
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