如何从 python 脚本中获取由 bash 脚本设置的环境变量？

Question

我确实有一个 setenv.sh 脚本，它在其他几个脚本的开头被调用，其中大多数是 bash 脚本。显然这个脚本确实设置了一些环境变量，这些变量是这些脚本以后使用的。

现在，问题是我想在一些 python 脚本中实现相同的行为，我发现如果你运行 setenv，python 脚本的环境不会更新。

因为我不想创建另一个脚本，它首先调用setenv.sh 和其他这个 myscript.py --- 我正在寻找一种方法来说服 python 加载这些变量（解析 seteve.h 不是选项...因为它更复杂）。

Answer 1

这是一个可以做到这一点的小库：

https://github.com/aneilbaboo/shellvars-py

Answer 2

最简单的解决方案显然是你不想要的，那就是为每个 python 脚本创建一个新的脚本文件。

但是，您可以通过调用 python 脚本本身来做大致等效的操作。当然，您需要在第二次调用时通知它不要这样做，否则您将得到无限（尾）递归。

下面的小“模块”（您可以直接导入，但您应该在启动时立即执行，然后再执行其他操作）将检查环境变量 SETENV 是否已设置，如果已设置，它将在获取SETENV 命名的文件后重新发出python 命令（尽其所能，因此如果它不仅仅是一个简单的脚本执行，它可能会出错）。它缺乏大量的错误检查，不应该被认为是生产就绪的；而是概念验证：

# file env_set.py
import os
import sys

if sys.argv[0] and "SETENV" in os.environ:
  setenv = os.environ["SETENV"]
  del os.environ["SETENV"]
  os.execvp("bash", ["bash", "-c",
        "source " + setenv + "; exec python " + sys.argv[0] + ' "${@}"',
        "--"] + sys.argv[1:])

还有一个小测试：

# file test_env_set.py
import env_set

import os
import sys
for name in sys.argv[1:]:
  if name in os.environ:
    print(name + "=" + os.environ[name])
  else:
    print("Undefined: " + name)

# file setenv.sh
export the_answer=42

$ python test_env_set.py SETENV the_answer
Undefined: SETENV
Undefined: the_answer
$ SETENV=setenv.sh python test_env_set.py SETENV the_answer
Undefined: SETENV
the_answer=42

Answer 3

也许尝试在导出之前和之后运行env，然后比较结果。像这样的

$ pwd
/tmp/test
$ cat setenv.sh
#!/bin/bash
export test=1234
$ cat test.sh
#!/bin/bash
source /tmp/test/setenv.sh
echo $test
$ ./test.sh
1234
$ python test.py
test=1234
$ cat test.py
#/usr/bin/env python
import os, subprocess
p=subprocess.Popen('env',stdout = subprocess.PIPE, stderr = subprocess.PIPE, shell = True)
oldEnv=p.communicate()[0]
p=subprocess.Popen('source /tmp/test/setenv.sh ; env',stdout = subprocess.PIPE, stderr = subprocess.PIPE, shell = True)
newEnv=p.communicate()[0]
for newStr in newEnv.split('\n'):
    flag = True
    for oldStr in oldEnv.split('\n'):
        if newStr == oldStr:
            #not exported by setenv.sh
            flag = False
            break
    if flag:
        #exported by setenv.sh
        print newStr