python - 子进程运行结束后如何获取环境变量？

Question

我正在寻找一种方法来执行此操作，以便我可以将其传递给另一个子进程的环境。

Answer 1

这是一个简单的函数，它在子进程中运行命令，然后将其环境提取到当前进程中。

它基于 Fnord 的版本，没有临时文件，并带有一个标记行以将 SET 命令与进程本身的任何输出区分开来。它不是防弹的，但它适合我的目的。

def setenv(cmd):
    cmd = cmd + ' && echo ~~~~START_ENVIRONMENT_HERE~~~~ && set'

    env = (subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE)
                     .stdout
                     .read()
                     .decode('utf-8')
                     .splitlines())

    record = False
    for e in env:
        if record:
            e = e.strip().split('=')
            os.environ[e[0]] = e[1]
        elif e.strip() == '~~~~START_ENVIRONMENT_HERE~~~~': 
            record = True