codecoverage improvement
[dcaegen2/platform.git] / mod / distributorapi / distributor / onboarding_client.py
1 # ============LICENSE_START=======================================================
2 # Copyright (c) 2019-2022 AT&T Intellectual Property. All rights reserved.
3 # ================================================================================
4 # Licensed under the Apache License, Version 2.0 (the "License");
5 # you may not use this file except in compliance with the License.
6 # You may obtain a copy of the License at
7 #
8 #      http://www.apache.org/licenses/LICENSE-2.0
9 #
10 # Unless required by applicable law or agreed to in writing, software
11 # distributed under the License is distributed on an "AS IS" BASIS,
12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 # See the License for the specific language governing permissions and
14 # limitations under the License.
15 # ============LICENSE_END=========================================================
16 """Onboarding API client"""
17
18 from distributor.utils import urljoin, get_json
19 from distributor import errors
20
21
22 def get_component(onboarding_url, name, version):
23     url = urljoin(onboarding_url, "components", **{"name": name, "version": version})
24     result = get_json(url)["components"]
25
26     if result:
27         return get_json(result[0]["componentUrl"])
28     else:
29         raise errors.DistributorAPIResourceNotFound("Onboarding API: Component not found")
30
31
32 def get_components_indexed(onboarding_url, list_name_version):
33     return dict([((c[0], c[1]), get_component(onboarding_url, c[0], c[1])) for c in list_name_version])