如何从子进程获取环境变量?

27

我想通过python程序调用一个进程,但是这个进程需要一些特定的环境变量,这些变量是由另一个进程设置的。我该如何获取第一个进程的环境变量,并将它们传递给第二个进程?

以下是程序的样子:

import subprocess

subprocess.call(['proc1']) # this set env. variables for proc2
subprocess.call(['proc2']) # this must have env. variables set by proc1 to work

但是这两个进程没有共享相同的环境。请注意,这些程序不是我的(第一个是一个又大又丑的.bat文件,第二个是专有软件),所以我不能修改它们(好吧,我可以从.bat文件中提取所有我需要的内容,但这很麻烦)。

N.B.:我正在使用Windows,但我更喜欢跨平台的解决方案(但我的问题在类Unix系统上不会发生...)


1
一个.bat文件?如果你在Windows上运行,你应该清楚地说明。 - Glenn Maynard
7个回答

31

以下是一个示例,演示如何从批处理或CMD文件中提取环境变量,而无需创建包装脚本。享受吧。

from __future__ import print_function
import sys
import subprocess
import itertools

def validate_pair(ob):
    try:
        if not (len(ob) == 2):
            print("Unexpected result:", ob, file=sys.stderr)
            raise ValueError
    except:
        return False
    return True

def consume(iter):
    try:
        while True: next(iter)
    except StopIteration:
        pass

def get_environment_from_batch_command(env_cmd, initial=None):
    """
    Take a command (either a single command or list of arguments)
    and return the environment created after running that command.
    Note that if the command must be a batch file or .cmd file, or the
    changes to the environment will not be captured.

    If initial is supplied, it is used as the initial environment passed
    to the child process.
    """
    if not isinstance(env_cmd, (list, tuple)):
        env_cmd = [env_cmd]
    # construct the command that will alter the environment
    env_cmd = subprocess.list2cmdline(env_cmd)
    # create a tag so we can tell in the output when the proc is done
    tag = 'Done running command'
    # construct a cmd.exe command to do accomplish this
    cmd = 'cmd.exe /s /c "{env_cmd} && echo "{tag}" && set"'.format(**vars())
    # launch the process
    proc = subprocess.Popen(cmd, stdout=subprocess.PIPE, env=initial)
    # parse the output sent to stdout
    lines = proc.stdout
    # consume whatever output occurs until the tag is reached
    consume(itertools.takewhile(lambda l: tag not in l, lines))
    # define a way to handle each KEY=VALUE line
    handle_line = lambda l: l.rstrip().split('=',1)
    # parse key/values into pairs
    pairs = map(handle_line, lines)
    # make sure the pairs are valid
    valid_pairs = filter(validate_pair, pairs)
    # construct a dictionary of the pairs
    result = dict(valid_pairs)
    # let the process finish
    proc.communicate()
    return result

回答您的问题,您需要创建一个.py文件,该文件执行以下操作:

env = get_environment_from_batch_command('proc1')
subprocess.Popen('proc2', env=env)

一次编辑建议,当前的实现不支持非 ASCII 代码页。在这种情况下,推荐的修复方法是在 cmd 构造中的 {env_cmd} 前面加上 chcp 65001 > NULL && - Jason R. Coombs
1
这是此函数的更新版本 https://github.com/PySide/pyside2-setup/blob/master/utils.py#L379 - Anton Kochkov
3
这不是跨平台的,正如 OP 所要求的。 - nmz787
2
是的,通常情况下不可能从运行中的进程中提取环境,正如Martin v. Löwis' 的回答所述。本答案重点解决OP手头的问题。假设如果您在该平台上有cmd.exe,那么这应该可以跨平台使用,如果您正在处理proc1的批处理文件。 - Jason R. Coombs

5
正如您所说,进程不共享环境 - 所以您字面上的要求是不可能的,在Python或任何编程语言中都不可能。
您可以将环境变量放在文件或管道中,并执行以下操作:
- 父进程读取它们,并在创建proc2之前将它们传递给proc2。 - proc2读取它们,并在本地设置它们。
后者需要proc2的配合;前者需要在启动proc2之前使变量变为已知。

在Linux下,如果你是root用户,你可以检查/proc/<pid>/environ并解析它...这并非“不可能”。 - 0x6adb015
1
@0x6abd015:然而,这不是OP所问的:他不想让proc2找出proc1的环境(你可以用proc文件系统做什么),而是他想让proc1为proc2设置环境 - 我仍然认为这是不可能的。 - Martin v. Löwis

2

根据您的情况,在Windows系统下,您需要一个Windows答案。

创建一个包装批处理文件,例如“run_program.bat”,并运行这两个程序:

@echo off
call proc1.bat
proc2

脚本将运行并设置其环境变量。这两个脚本在同一个解释器(cmd.exe实例)中运行,因此当执行prog2时,prog1.bat设置的变量将被设置。
不是非常漂亮,但它会起作用。
(Unix用户可以在bash脚本中执行相同的操作:“source file.sh”。)

1
我正在尝试绕过使用sourch blah; now_do_second_thing - nmz787

2
您可以使用psutil中的Process来获取该进程的环境变量。
如果您想自己实现,可以参考psutil的内部实现。它适用于一些操作系统。
当前支持的操作系统有:
  • AIX
  • FreeBSD、OpenBSD、NetBSD
  • Linux
  • macOS
  • Sun Solaris
  • Windows
例如:在Linux平台上,您可以在文件/proc/7877/environ中找到一个pid为7877的进程的环境变量,只需以rt模式打开即可读取。
当然,最好的方法是:
import os
from typing import Dict
from psutil import Process

process = Process(pid=os.getpid())
process_env: Dict = process.environ()

print(process_env)

您可以在源代码中找到其他平台的实现。

希望我能帮到您。


3
根据 https://psutil.readthedocs.io/en/latest/#processes,``environ()``方法可能不包含启动后环境的更改,因此这种方法无效。 - drodri
@drodri 我刚在Windows 10上进行了测试,它确实包含了更改。我认为那个警告是针对其他操作系统的。 - Carl Walsh

1

0
有两件事情需要考虑:(1)通过某种方式将进程合并到同一个进程中,使它们共享相同的环境;或者(2)让第一个进程生成包含相关环境变量的输出,这样Python就可以读取它并为第二个进程构建环境。我认为(尽管我不是100%确定),没有办法像你希望的那样从子进程中获取环境。

-1

环境是从父进程继承的。在主脚本中设置所需的环境,而不是在子进程中。


这并没有回答问题。如果您有需要设置冲突的环境变量的程序,那么也有很好的理由不这样做。 - nmz787

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接