1 # Copyright (C) 2021 Wind River Systems, Inc.
3 # Licensed under the Apache License, Version 2.0 (the "License");
4 # you may not use this file except in compliance with the License.
5 # You may obtain a copy of the License at
7 # http://www.apache.org/licenses/LICENSE-2.0
9 # Unless required by applicable law or agreed to in writing, software
10 # distributed under the License is distributed on an "AS IS" BASIS,
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
19 from datetime import datetime
22 from o2common.service import unit_of_work
23 from o2common.config import config
24 from o2common.views.view import gen_filter, check_filter
25 from o2common.views.pagination_view import Pagination
26 from o2common.views.route_exception import BadRequestException
28 from o2ims.domain import ocloud
29 from o2ims.views.ocloud_dto import SubscriptionDTO
30 from o2ims.domain.subscription_obj import Subscription
32 from o2common.helper import o2logging
33 logger = o2logging.get_logger(__name__)
36 def oclouds(uow: unit_of_work.AbstractUnitOfWork):
38 li = uow.oclouds.list()
39 return [r.serialize() for r in li]
42 def ocloud_one(ocloudid: str, uow: unit_of_work.AbstractUnitOfWork):
44 first = uow.oclouds.get(ocloudid)
45 return first.serialize() if first is not None else None
48 def resource_types(uow: unit_of_work.AbstractUnitOfWork, **kwargs):
49 pagination = Pagination(**kwargs)
50 query_kwargs = pagination.get_pagination()
51 args = gen_filter(ocloud.ResourceType,
52 kwargs['filter']) if 'filter' in kwargs else []
54 li = uow.resource_types.list_with_count(*args, **query_kwargs)
55 return pagination.get_result(li)
58 def resource_type_one(resourceTypeId: str,
59 uow: unit_of_work.AbstractUnitOfWork):
61 first = uow.resource_types.get(resourceTypeId)
62 return first.serialize() if first is not None else None
65 def resource_pools(uow: unit_of_work.AbstractUnitOfWork, **kwargs):
66 pagination = Pagination(**kwargs)
67 query_kwargs = pagination.get_pagination()
68 args = gen_filter(ocloud.ResourcePool,
69 kwargs['filter']) if 'filter' in kwargs else []
71 li = uow.resource_pools.list_with_count(*args, **query_kwargs)
72 return pagination.get_result(li)
75 def resource_pool_one(resourcePoolId: str,
76 uow: unit_of_work.AbstractUnitOfWork):
78 first = uow.resource_pools.get(resourcePoolId)
79 return first.serialize() if first is not None else None
82 def resources(resourcePoolId: str, uow: unit_of_work.AbstractUnitOfWork,
84 pagination = Pagination(**kwargs)
85 # filter key should be the same with database name
86 query_kwargs = pagination.get_pagination()
87 if 'resourceTypeName' in kwargs:
88 resource_type_name = kwargs['resourceTypeName']
90 # res_types = uow.resource_types.list()
92 # restype.resourceTypeId for restype in res_types
93 # if resourceTypeName == restype.name]
94 # restype_id = '' if len(restype_ids) == 0 else restype_ids[0]
95 res_type = uow.resource_types.get_by_name(resource_type_name)
96 restype_id = '' if res_type is None else res_type.resourceTypeId
97 query_kwargs['resourceTypeId'] = restype_id
99 ocloud.Resource, kwargs['filter']) if 'filter' in kwargs else []
100 args.append(ocloud.Resource.resourcePoolId == resourcePoolId)
101 # args.append(ocloud.Resource.parentId == None)
103 if 'parentId' in kwargs:
104 query_kwargs['parentId'] = kwargs['parentId']
106 query_kwargs['sort'] = kwargs['sort']
109 ret = uow.resources.list_with_count(
110 resourcePoolId, *args, **query_kwargs)
112 return pagination.get_result(ret)
115 def resource_one(resourceId: str, uow: unit_of_work.AbstractUnitOfWork):
117 first = uow.resources.get(resourceId)
118 return first.serialize() if first is not None else None
121 def deployment_managers(uow: unit_of_work.AbstractUnitOfWork, **kwargs):
122 pagination = Pagination(**kwargs)
123 query_kwargs = pagination.get_pagination()
124 args = gen_filter(ocloud.DeploymentManager,
125 kwargs['filter']) if 'filter' in kwargs else []
127 li = uow.deployment_managers.list_with_count(*args, **query_kwargs)
128 return pagination.get_result(li)
131 def deployment_manager_one(deploymentManagerId: str,
132 uow: unit_of_work.AbstractUnitOfWork,
134 ocloud.DeploymentManagerProfileDefault):
135 profile = profile.lower()
137 first = uow.deployment_managers.get(deploymentManagerId)
140 result = first.serialize()
144 profile_data = result.pop("profile", None)
145 result['profileName'] = profile
146 profiles = config.get_dms_support_profiles()
147 if profile not in profiles:
150 if ocloud.DeploymentManagerProfileDefault == profile \
151 or ocloud.DeploymentManagerProfileSOL018 == profile:
152 result['serviceUri'] = \
153 profile_data['cluster_api_endpoint']
154 result['profileData'] = profile_data
155 elif ocloud.DeploymentManagerProfileSOL018HelmCLI == profile:
156 result['serviceUri'] = \
157 profile_data['cluster_api_endpoint']
159 helmcli_profile = dict()
160 helmcli_profile["helmcli_host_with_port"], helmcli_profile[
161 "helmcli_username"], helmcli_profile["helmcli_password"] = \
162 config.get_helmcli_access()
163 helmcli_profile["helmcli_kubeconfig"] = _gen_kube_config(
164 deploymentManagerId, profile_data)
165 result['profileData'] = helmcli_profile
172 def _gen_kube_config(dmId: str, kubeconfig: dict) -> dict:
174 data = config.gen_k8s_config_dict(
175 kubeconfig.pop('cluster_api_endpoint', None),
176 kubeconfig.pop('cluster_ca_cert', None),
177 kubeconfig.pop('admin_user', None),
178 kubeconfig.pop('admin_client_cert', None),
179 kubeconfig.pop('admin_client_key', None),
182 # Generate a random key for tmp kube config file
183 # letters = string.ascii_uppercase
184 # random_key = ''.join(random.choice(letters) for i in range(10))
187 # Get datetime of now as tag of the tmp file
188 current_time = datetime.now().strftime("%Y%m%d%H%M%S")
189 tmp_file_name = 'kubeconfig_' + name_key + "_" + current_time
190 kube_config_name = 'kubeconfig_' + name_key + '.config'
192 # write down the yaml file of kubectl into tmp folder
193 with open('/tmp/' + tmp_file_name, 'w') as file:
194 yaml.dump(data, file)
196 # generate the kube config file if not exist or update the file if it
198 if not os.path.exists('/configs/' + kube_config_name) or not \
199 filecmp.cmp('/tmp/'+tmp_file_name, '/configs/'+kube_config_name):
200 shutil.move(os.path.join('/tmp', tmp_file_name),
201 os.path.join('/configs', kube_config_name))
203 return '/configs/'+kube_config_name
206 def subscriptions(uow: unit_of_work.AbstractUnitOfWork, **kwargs):
207 pagination = Pagination(**kwargs)
208 query_kwargs = pagination.get_pagination()
209 args = gen_filter(Subscription,
210 kwargs['filter']) if 'filter' in kwargs else []
212 li = uow.subscriptions.list_with_count(*args, **query_kwargs)
213 return pagination.get_result(li)
216 def subscription_one(subscriptionId: str,
217 uow: unit_of_work.AbstractUnitOfWork):
219 first = uow.subscriptions.get(subscriptionId)
220 return first.serialize() if first is not None else None
223 def subscription_create(subscriptionDto: SubscriptionDTO.subscription_create,
224 uow: unit_of_work.AbstractUnitOfWork):
225 filter = subscriptionDto.get('filter', '')
226 consumer_subs_id = subscriptionDto.get('consumerSubscriptionId', '')
228 check_filter(ocloud.Resource, filter)
230 sub_uuid = str(uuid.uuid4())
231 subscription = Subscription(
232 sub_uuid, subscriptionDto['callback'],
233 consumer_subs_id, filter)
236 args.append(getattr(Subscription, 'callback')
237 == subscriptionDto['callback'])
238 args.append(getattr(Subscription, 'filter') == filter)
239 args.append(getattr(Subscription,
240 'consumerSubscriptionId') == consumer_subs_id)
241 count, _ = uow.alarm_subscriptions.list_with_count(*args)
243 raise BadRequestException("The value of parameters is duplicated")
244 uow.subscriptions.add(subscription)
246 first = uow.subscriptions.get(sub_uuid)
247 return first.serialize()
250 def subscription_delete(subscriptionId: str,
251 uow: unit_of_work.AbstractUnitOfWork):
253 uow.subscriptions.delete(subscriptionId)