如何使用自定义环境变量通过 subprocess.Popen 获取 PID？

Question

使用 Python，我如何使用修改后的环境变量运行子进程并获取其 PID？我假设 subprocess.Popen() 是在正确的轨道上......

在 shell (bash) 中，我会这样做：

MY_ENV_VAR=value ./program_name arg1 arg2 etc &

这会在后台运行program_name，传入“arg1”、“arg2”和“etc”，并使用修改后的环境变量“MY_ENV_VAR”，其值为“value”。程序program_name 要求将环境变量MY_ENV_VAR 设置为正确的值。

如何在 Python 中做同样的事情？我绝对需要进程的PID。 （我的目的是保持 python 脚本运行并检查program_name 在此期间正在做的一些事情，我需要进程 ID 以确保它仍在运行。）

我试过了：

proc = subprocess.Popen(['MY_ENV_VAR=value', './program_name', 'arg1', 'arg2', 'etc'])

当然，它希望第一项是程序，而不是环境变量。

也试过了：

environ = dict(os.environ)
environ['MY_ENV_VAR'] = 'value'
proc = subprocess.Popen(['./program_name', 'arg1', 'arg2', 'etc', env=environ])

关闭，我想，但没有雪茄。同样，这个：

environ = dict(os.environ)
environ['MY_ENV_VAR'] = 'value'
proc = subprocess.Popen(['echo', '$MY_ENV_VAR'], env=environ)

这在字面上与“$MY_ENV_VAR”相呼应，我想是因为没有外壳可以解释它。好的，所以我尝试上面的方法，但改用这一行：

proc = subprocess.Popen(['echo', '$MY_ENV_VAR'], env=environ, shell=True)

这很好，很漂亮，除了回显的值是空白的（显然不存在）。即使它确实有效，我也会得到 shell 的 PID，而不是我尝试启动的实际进程。

我需要使用自定义环境变量启动一个进程并获取它的 PID（不是 shell 的 PID）。想法？

Answer 1

您的上一个版本非常接近，但还不够。

您不希望 $MY_ENV_VAR 成为 echo 的参数。 echo 程序将在其环境中包含 MY_ENV_VAR，但 echo 不执行任何环境变量扩展。你需要在它到达 echo 之前通过 shell 扩展它。

这实际上可能与您的实际测试用例无关。您已经正在在所有测试中为子进程获取环境变量，只是echo 对该环境变量没有任何作用。如果您的实际程序只需要设置环境变量，那么您就完成了：

proc = subprocess.Popen(['./program_name', 'arg1', 'arg2', 'etc'], env=environ)

但是如果你的程序需要它被替换，比如echo，那么你必须将它替换到参数中在它们被传递给你的程序。

最简单的方法是只给 shell 一个命令行而不是参数列表：

proc = subprocess.Popen('echo "$MY_ENV_VAR"', env=environ, shell=True)

人们会告诉你，你永远不应该在subprocess 中使用命令字符串——但这样做的原因是你总是想防止 shell 以一种可能不安全/等的方式扩展变量等.在极少数情况下，当您想要 shell 执行其shell 的事情时，您想要一个命令字符串。

当然，如果您使用 shell，在大多数平台上，您最终会得到 shell 的 PID，而不是实际程序的 PID。如果没有做一些特定于平台的挖掘来枚举 shell 的孩子（或者将整个事情包装在一些简单的 sh 代码中，间接地给你孩子的 PID），没有办法解决这个问题。 shell 就是你正在运行的东西。

另一种选择是在 Python 中扩展变量，而不是让 shell 来做。那么你甚至需要一个shell：

proc = subprocess.Popen(['echo', os.path.expandvars('$MY_ENV_VAR')])

……或者，更简单地说：

proc = subprocess.Popen(['echo', os.environ['MY_ENV_VAR']])

Answer 2

这是一个吐出当前环境的程序。

#!/usr/bin/env python
##program_name
import os
for k,v in os.environ.iteritems():
    print k, '=', v

这是一个调用另一个程序，但首先改变环境的程序

#!/usr/bin/env python
import subprocess, os
newenv = os.environ.copy()
newenv['MY_ENV_VAR'] = 'value'
args = ['./program_name', 'arg1', 'arg2', 'etc']
proc = subprocess.Popen(args, env=newenv)
pid = proc.pid
proc.wait()
print 'PID =', pid