mod/distributorapi/distributor/utils.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 """Utility functions"""
  17
  18 from urllib.parse import quote
  19 import requests as reqs
  20 from distributor import errors
  21
  22
  23 def urljoin(base, *trailing, **query_params):
  24     base = base[0:-1] if base[-1] == "/" else base
  25     full = [base] + list(trailing)
  26     url = "/".join(full)
  27
  28     if query_params:
  29         qp = ["{0}={1}".format(quote(k), quote(str(v))) for k, v in query_params.items()]
  30         qp = "&".join(qp)
  31         return "?".join([url, qp])
  32     else:
  33         return url
  34
  35
  36 def get_json(url):
  37     resp = reqs.get(url)
  38
  39     try:
  40         resp.raise_for_status()
  41         return resp.json()
  42     except Exception as e:
  43         raise errors.DistributorAPIError(e)