如何使用 python 在 GitHub actions 中设置环境变量

Question

我想执行一个 python 脚本来在 GitHub actions 中设置一些环境变量。我想稍后在 GitHub 操作步骤中使用这些环境变量。我的 python 脚本如下所示：

new_ver = get_version_from_commit(commit_msg)
if new_ver:
    if new_ver == "false":
        os.environ["SHOULD_PUSH"] = "0"
        print("Not pushing the image to k8s")
        exit(0)
    else:
        new_tag = app_name + ":" + str(new_ver)
        os.environ["DOCKER_IMAGE_TAG"] = new_tag
        os.environ["SHOULD_PUSH"] = "1"
        print("New tag: " + new_tag)
        exit(0)

执行上述 python 脚本后，我的 GitHub 操作文件的一部分如下所示：

- name: Print env var
  run: echo ${{ env.DOCKER_IMAGE_TAG }}
- name: Build and push
  id: docker_build
  uses: docker/build-push-action@v2
  with:
     push: true
     tags: ${{ secrets.DOCKER_REGISTRY }}/${{ env.DOCKER_IMAGE_TAG }}

但使用 os.environ 不会将环境变量暴露在 python 进程之外。我怎样才能解决这个问题 ？

Answer 1

您不能直接设置环境变量。相反，您需要将环境变量写入一个文件，您可以通过$GITHUB_ENV.

在一个简单的工作流程步骤中，您可以将其附加到文件中，如下所示（来自文档）：

echo "{name}={value}" >> $GITHUB_ENV

在Python中，你可以这样做：

import os

env_file = os.getenv('GITHUB_ENV')

with open(env_file, "a") as myfile:
    myfile.write("MY_VAR=MY_VALUE")

给定这个 python 脚本，您可以设置并使用新的环境变量，如下所示：

- run: python write-env.py
- run: echo ${{ env.MY_VAR }}