tools/utils.py

   1 # Copyright 2019 Nokia
   2 #
   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
   6 #
   7 #     http://www.apache.org/licenses/LICENSE-2.0
   8 #
   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.
  14
  15 import os
  16
  17
  18 def apply_jenkins_auth(url):
  19     validate_environ(['JENKINS_USERNAME', 'JENKINS_TOKEN'])
  20     protocol, address = url.split('://')
  21     url = '{protocol}://{user}:{token}@{address}'.format(**dict(protocol=protocol,
  22                                                                 user=os.environ['JENKINS_USERNAME'],
  23                                                                 token=os.environ['JENKINS_TOKEN'],
  24                                                                 address=address))
  25     return url
  26
  27
  28 def validate_environ(expected_vars):
  29     for env in expected_vars:
  30         if env not in os.environ:
  31             raise Exception('{} must be defined in environment'.format(env))