在python中设置调用bash脚本的环境变量[重复]

Question

我有一个如下所示的 bash 脚本：

python myPythonScript.py

python myOtherScript.py $VarFromFirstScript

而myPythonScript.py 看起来像这样：

print("Running some code...")
VarFromFirstScript = someFunc()
print("Now I do other stuff")

问题是，我如何将变量 VarFromFirstScript 返回到名为 myPythonScript.py 的 bash 脚本。

我尝试了os.environ['VarFromFirstScript'] = VarFromFirstScript，但这不起作用（我认为这意味着 python 环境与调用 bash 脚本的环境不同）。

Answer 1

您不能将环境变量传播到父进程。但是您可以打印变量，然后从您的 shell 将其分配回变量名：

VarFromFirstScript=$(python myOtherScript.py $VarFromFirstScript)

您不得在代码中打印任何其他内容，或使用stderr

sys.stderr.write("Running some code...\n")
VarFromFirstScript = someFunc()
sys.stdout.write(VarFromFirstScript)

另一种方法是创建一个包含要设置的变量的文件，并使其由您的 shell 解析（您可以创建一个父 shell 将 source 的 shell）

import shlex
with open("shell_to_source.sh","w") as f:
   f.write("VarFromFirstScript={}\n".format(shlex.quote(VarFromFirstScript))

（shlex.quote 允许避免来自 python 的代码注入，由 Charles Duffy 提供）

那么在调用python之后：

source ./shell_to_source.sh

Answer 2

您只能将环境变量从父进程传递给子进程。

创建子进程时，环境块复制到子进程 - 子进程有一个副本，因此子进程中的任何更改只会影响子进程的副本（以及它的任何其他子进程）创建）。

要与父级通信，最简单的方法是在我们捕获stdout 的bash 中使用命令替换：

Bash 脚本：

#!/bin/bash
var=$(python myPythonScript.py)
echo "Value in bash: $var"

Python 脚本：

print("Hollow world!")

示例运行：

$ bash gash.sh
Value in bash: Hollow world!

您在 python 中有其他打印语句，您需要过滤出您需要的数据，可能通过使用众所周知的前缀标记数据。

如果你在 python 中有很多 print 语句，那么这个解决方案是不可扩展的，所以你可能需要使用进程替换，像这样：

Bash 脚本：

#!/bin/bash

while read -r line
do
    if [[ $line = ++++* ]]
    then
        # Strip out the marker
        var=${line#++++}
    else
        echo "$line"
    fi
done < <(python myPythonScript.py)

echo "Value in bash: $var"

Python 脚本：

def someFunc():
    return "Hollow World"

print("Running some code...")

VarFromFirstScript = someFunc()
# Prefix our data with a well-known marker
print("++++" + VarFromFirstScript)

print("Now I do other stuff")

示例运行：

$ bash gash.sh
Running some code...
Now I do other stuff
Value in bash: Hollow World

Answer 3

我会获取您的脚本，这是最常用的方法。这会在当前 shell 下执行脚本，而不是加载另一个脚本。因为这使用了您设置的相同 shell 环境变量，所以在它退出时将可以访问它。 . /path/to/script.sh 或 source /path/to/script.sh 都可以工作，. 在 source 有时不工作的地方工作。