在python脚本中设置环境变量

Question

我有一个 bash 脚本，用于设置环境变量并运行命令

LD_LIBRARY_PATH=my_path
sqsub -np $1 /homedir/anotherdir/executable

现在我想使用 python 而不是 bash，因为我想计算传递给命令的一些参数。

我已经尝试

putenv("LD_LIBRARY_PATH", "my_path")

并

call("export LD_LIBRARY_PATH=my_path")

遵循，

call("sqsub -np " + var1 + "/homedir/anotherdir/executable")

但程序总是放弃，因为未设置 LD_LIBRARY_PATH 。

我该如何解决这个问题？

感谢您的帮助！

（如果我在调用 python 脚本之前导出 LD_LIBRARY_PATH 一切正常，但我希望 python 确定路径并将环境变量设置为正确的值）

Answer 1

bash：

LD_LIBRARY_PATH=my_path
sqsub -np $1 /path/to/executable

类似，在Python中：

import os
import subprocess
import sys

os.environ['LD_LIBRARY_PATH'] = "my_path" # visible in this process + all children
subprocess.check_call(['sqsub', '-np', sys.argv[1], '/path/to/executable'],
                      env=dict(os.environ, SQSUB_VAR="visible in this subprocess"))

Answer 2

这里有很多好的答案，但您应该不惜一切代价避免使用 shell=True 将不受信任的变量传递给子进程，因为这是一个安全风险。变量可以转义到 shell 并运行任意命令！如果您无法避免它，至少使用 python3 的 shlex.quote() 来转义字符串（如果您有多个空格分隔的参数，请引用每个拆分而不是整个字符串）。

shell=False 始终是传递参数数组的默认值。

现在安全解决方案...

方法#1

更改您自己的进程环境 - 新环境将应用于python 本身和所有子进程。

os.environ['LD_LIBRARY_PATH'] = 'my_path'
command = ['sqsub', '-np', var1, '/homedir/anotherdir/executable']
subprocess.check_call(command)

方法#2

复制环境和通行证是给孩子的。您可以完全控制子环境，并且不会影响 python 自己的环境。

myenv = os.environ.copy()
myenv['LD_LIBRARY_PATH'] = 'my_path'
command = ['sqsub', '-np', var1, '/homedir/anotherdir/executable']
subprocess.check_call(command, env=myenv)

方法 #3

仅限 Unix：执行 env 设置环境变量。如果你有很多变量需要修改而不是 portabe，那就更麻烦了，但像 #2 一样，你保留对 python 和子环境的完全控制。

command = ['env', 'LD_LIBRARY_PATH=my_path', 'sqsub', '-np', var1, '/homedir/anotherdir/executable']
subprocess.check_call(command)

---

当然，如果 var1 包含多个空格分隔的参数，它们现在将作为带有空格的单个参数传递。要使用 shell=True 保留原始行为，您必须编写一个包含拆分字符串的命令数组：

command = ['sqsub', '-np'] + var1.split() + ['/homedir/anotherdir/executable']

Answer 3

您可以通过使用

os.environ['LD_LIBRARY_PATH'] = 'my_path'

shell（使用您的 os.environ）中的子进程来添加元素到您的环境中，方法是使用

subprocess.call('sqsub -np ' + var1 + '/homedir/anotherdir/executable', shell=True)