X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=o2ims%2Fviews%2Focloud_view.py;h=bc1109774834fd74f7a1ef2d9211f77e9757e063;hb=c6a78454000fc8ad2699207063c1c7a87652bbd7;hp=7355f71b74cd9e6861b02c956b1bd19872aa6f51;hpb=defe8209b3628593c186487857fe02586d7e1503;p=pti%2Fo2.git diff --git a/o2ims/views/ocloud_view.py b/o2ims/views/ocloud_view.py index 7355f71..bc11097 100644 --- a/o2ims/views/ocloud_view.py +++ b/o2ims/views/ocloud_view.py @@ -12,9 +12,21 @@ # See the License for the specific language governing permissions and # limitations under the License. +import filecmp +import os.path import uuid +import yaml +from datetime import datetime +import shutil from o2common.service import unit_of_work +from o2common.config import config +from o2common.views.view import gen_filter, check_filter +from o2common.views.pagination_view import Pagination +from o2common.views.route_exception import BadRequestException, \ + NotFoundException + +from o2ims.domain import ocloud from o2ims.views.ocloud_dto import SubscriptionDTO from o2ims.domain.subscription_obj import Subscription @@ -34,10 +46,14 @@ def ocloud_one(ocloudid: str, uow: unit_of_work.AbstractUnitOfWork): return first.serialize() if first is not None else None -def resource_types(uow: unit_of_work.AbstractUnitOfWork): +def resource_types(uow: unit_of_work.AbstractUnitOfWork, **kwargs): + pagination = Pagination(**kwargs) + query_kwargs = pagination.get_pagination() + args = gen_filter(ocloud.ResourceType, + kwargs['filter']) if 'filter' in kwargs else [] with uow: - li = uow.resource_types.list() - return [r.serialize() for r in li] + li = uow.resource_types.list_with_count(*args, **query_kwargs) + return pagination.get_result(li) def resource_type_one(resourceTypeId: str, @@ -47,10 +63,14 @@ def resource_type_one(resourceTypeId: str, return first.serialize() if first is not None else None -def resource_pools(uow: unit_of_work.AbstractUnitOfWork): +def resource_pools(uow: unit_of_work.AbstractUnitOfWork, **kwargs): + pagination = Pagination(**kwargs) + query_kwargs = pagination.get_pagination() + args = gen_filter(ocloud.ResourcePool, + kwargs['filter']) if 'filter' in kwargs else [] with uow: - li = uow.resource_pools.list() - return [r.serialize() for r in li] + li = uow.resource_pools.list_with_count(*args, **query_kwargs) + return pagination.get_result(li) def resource_pool_one(resourcePoolId: str, @@ -62,8 +82,9 @@ def resource_pool_one(resourcePoolId: str, def resources(resourcePoolId: str, uow: unit_of_work.AbstractUnitOfWork, **kwargs): - - filter_kwargs = {} # filter key should be the same with database name + pagination = Pagination(**kwargs) + # filter key should be the same with database name + query_kwargs = pagination.get_pagination() if 'resourceTypeName' in kwargs: resource_type_name = kwargs['resourceTypeName'] with uow: @@ -74,16 +95,22 @@ def resources(resourcePoolId: str, uow: unit_of_work.AbstractUnitOfWork, # restype_id = '' if len(restype_ids) == 0 else restype_ids[0] res_type = uow.resource_types.get_by_name(resource_type_name) restype_id = '' if res_type is None else res_type.resourceTypeId - filter_kwargs['resourceTypeId'] = restype_id + query_kwargs['resourceTypeId'] = restype_id + args = gen_filter( + ocloud.Resource, kwargs['filter']) if 'filter' in kwargs else [] + args.append(ocloud.Resource.resourcePoolId == resourcePoolId) + # args.append(ocloud.Resource.parentId == None) - # li = uow.resources.list(resourcePoolId) - # return [r.serialize() for r in li if r.resourceTypeId == restype_id] if 'parentId' in kwargs: - filter_kwargs['parentId'] = kwargs['parentId'] + query_kwargs['parentId'] = kwargs['parentId'] + if 'sort' in kwargs: + query_kwargs['sort'] = kwargs['sort'] with uow: - li = uow.resources.list(resourcePoolId, **filter_kwargs) - return [r.serialize() for r in li] + ret = uow.resources.list_with_count( + resourcePoolId, *args, **query_kwargs) + + return pagination.get_result(ret) def resource_one(resourceId: str, uow: unit_of_work.AbstractUnitOfWork): @@ -92,23 +119,102 @@ def resource_one(resourceId: str, uow: unit_of_work.AbstractUnitOfWork): return first.serialize() if first is not None else None -def deployment_managers(uow: unit_of_work.AbstractUnitOfWork): +def deployment_managers(uow: unit_of_work.AbstractUnitOfWork, **kwargs): + pagination = Pagination(**kwargs) + query_kwargs = pagination.get_pagination() + args = gen_filter(ocloud.DeploymentManager, + kwargs['filter']) if 'filter' in kwargs else [] with uow: - li = uow.deployment_managers.list() - return [r.serialize() for r in li] + li = uow.deployment_managers.list_with_count(*args, **query_kwargs) + return pagination.get_result(li) def deployment_manager_one(deploymentManagerId: str, - uow: unit_of_work.AbstractUnitOfWork): + uow: unit_of_work.AbstractUnitOfWork, + profile: str = + ocloud.DeploymentManagerProfileDefault): + profile = profile.lower() with uow: first = uow.deployment_managers.get(deploymentManagerId) - return first.serialize() if first is not None else None - - -def subscriptions(uow: unit_of_work.AbstractUnitOfWork): + if first is None: + return first + result = first.serialize() + if result is None: + return None + + profile_data = result.pop("profile", None) + profiles = config.get_dms_support_profiles() + if profile not in profiles: + return "" + + extensions = { + 'profileName': profile + } + if ocloud.DeploymentManagerProfileDefault == profile \ + or ocloud.DeploymentManagerProfileSOL018 == profile: + result['serviceUri'] = \ + profile_data['cluster_api_endpoint'] + extensions['profileData'] = profile_data + elif ocloud.DeploymentManagerProfileSOL018HelmCLI == profile: + result['serviceUri'] = \ + profile_data['cluster_api_endpoint'] + + helmcli_profile = dict() + helmcli_profile["helmcli_host_with_port"], helmcli_profile[ + "helmcli_username"], helmcli_profile["helmcli_password"] = \ + config.get_helmcli_access() + helmcli_profile["helmcli_kubeconfig"] = _gen_kube_config( + deploymentManagerId, profile_data) + extensions['profileData'] = helmcli_profile + else: + return "" + + result['extensions'] = extensions + return result + + +def _gen_kube_config(dmId: str, kubeconfig: dict) -> dict: + + data = config.gen_k8s_config_dict( + kubeconfig.pop('cluster_api_endpoint', None), + kubeconfig.pop('cluster_ca_cert', None), + kubeconfig.pop('admin_user', None), + kubeconfig.pop('admin_client_cert', None), + kubeconfig.pop('admin_client_key', None), + ) + + # Generate a random key for tmp kube config file + # letters = string.ascii_uppercase + # random_key = ''.join(random.choice(letters) for i in range(10)) + name_key = dmId[:8] + + # Get datetime of now as tag of the tmp file + current_time = datetime.now().strftime("%Y%m%d%H%M%S") + tmp_file_name = 'kubeconfig_' + name_key + "_" + current_time + kube_config_name = 'kubeconfig_' + name_key + '.config' + + # write down the yaml file of kubectl into tmp folder + with open('/tmp/' + tmp_file_name, 'w') as file: + yaml.dump(data, file) + + # generate the kube config file if not exist or update the file if it + # changes + if not os.path.exists('/configs/' + kube_config_name) or not \ + filecmp.cmp('/tmp/'+tmp_file_name, '/configs/'+kube_config_name): + shutil.move(os.path.join('/tmp', tmp_file_name), + os.path.join('/configs', kube_config_name)) + + return '/configs/'+kube_config_name + + +def subscriptions(uow: unit_of_work.AbstractUnitOfWork, **kwargs): + pagination = Pagination(**kwargs) + query_kwargs = pagination.get_pagination() + args = gen_filter(Subscription, + kwargs['filter']) if 'filter' in kwargs else [] with uow: - li = uow.subscriptions.list() - return [r.serialize() for r in li] + li = uow.subscriptions.list_with_count(*args, **query_kwargs) + return pagination.get_result(li) def subscription_one(subscriptionId: str, @@ -118,23 +224,40 @@ def subscription_one(subscriptionId: str, return first.serialize() if first is not None else None -def subscription_create(subscriptionDto: SubscriptionDTO.subscription, +def subscription_create(subscriptionDto: SubscriptionDTO.subscription_create, uow: unit_of_work.AbstractUnitOfWork): + filter = subscriptionDto.get('filter', '') + consumer_subs_id = subscriptionDto.get('consumerSubscriptionId', '') + + check_filter(ocloud.Resource, filter) sub_uuid = str(uuid.uuid4()) subscription = Subscription( sub_uuid, subscriptionDto['callback'], - subscriptionDto['consumerSubscriptionId'], - subscriptionDto['filter']) + consumer_subs_id, filter) with uow: + args = list() + args.append(getattr(Subscription, 'callback') + == subscriptionDto['callback']) + args.append(getattr(Subscription, 'filter') == filter) + args.append(getattr(Subscription, + 'consumerSubscriptionId') == consumer_subs_id) + count, _ = uow.subscriptions.list_with_count(*args) + if count > 0: + raise BadRequestException("The value of parameters is duplicated") uow.subscriptions.add(subscription) uow.commit() - return {"subscriptionId": sub_uuid} + first = uow.subscriptions.get(sub_uuid) + return first.serialize() def subscription_delete(subscriptionId: str, uow: unit_of_work.AbstractUnitOfWork): with uow: + first = uow.subscriptions.get(subscriptionId) + if not first: + raise NotFoundException( + "Subscription {} not found.".format(subscriptionId)) uow.subscriptions.delete(subscriptionId) uow.commit() return True