Load a UCP client bundle in python3

load_env.py

"""
load_env is the functional equivalent of loading a UCP client bundle
"""
def load_env(client_bundle="/client-bundle"):
    logging.info("Loading UCP client bundle...")
    try:
        # Open the extracted, mounted client_bundle directory and use its
        # contents
        bundle_env = client_bundle + "/env.sh"
        with open(bundle_env) as f:
            content = f.readlines()
    except FileNotFoundError as e:
        logging.error("Unable to load UCP client bundle: {0}".format(e))
    for line in content:
        if line.startswith("export DOCKER_") or line.startswith("export KUBECONFIG"):
            logging.debug("Applying Docker environment variable: {}".format(line))
            env_vars = line.split(" ")[1].split("=")
            os.environ["{0}".format(env_vars[0])] = "{0}".format(env_vars[1].strip())
        # Apply kubectl set-* commands
        if line.startswith("kubectl"):
            if "kubectl config" in line:
                try:
                    # Run each given line
                    run('{}'.format(line.replace("$PWD", bundle_dir).strip()), output=False)
                except RuntimeError as err:
                    logging.error("Unable to set kubectl config: {}".format(e))
    # Apply the final DOCKER_CERT_PATH environment var
    os.environ["DOCKER_CERT_PATH"] = client_bundle