サブプロセスから環境を取得する方法は？

StackOverflow https://stackoverflow.com/questions/1214496

質問

Pythonプログラムを介してプロセスを呼び出したいのですが、このプロセスには別のプロセスによって設定される特定の環境変数が必要です。最初のプロセス環境変数を取得して2番目に渡すにはどうすればよいですか？

これはプログラムの外観です：

import subprocess

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

しかし、処理するプロセスは同じ環境を共有しません。これらのプログラムは私のものではないことに注意してください（1つ目は大きくてlyい.batファイルで、2つ目は独自のソフトです）。 ）。

N.B .:私はWindowsを使用していますが、クロスプラットフォームソリューションを好みます（ただし、Unixのようなものでは私の問題は起こりません...）

役に立ちましたか？

解決

明らかにWindowsにいるので、Windowsの回答が必要です。

ラッパーバッチファイルを作成します。 ＆quot; run_program.bat＆quot ;、両方のプログラムを実行します：

@echo off
call proc1.bat
proc2

スクリプトが実行され、環境変数が設定されます。両方のスクリプトは同じインタープリター（cmd.exeインスタンス）で実行されるため、prog2の実行時に変数prog1.batが設定する が設定されます。

それほどきれいではありませんが、動作します。

（Unixの人は、bashスクリプトで同じことを行うことができます：＆quot; source file.sh＆quot;。）

他のヒント

ラッパースクリプトを作成せずに、バッチまたは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)

おっしゃるように、プロセスは環境を共有しません。したがって、Pythonだけでなく、任意のプログラミング言語でも、文字通り尋ねることはできません。

できるできることは、環境変数をファイル、またはパイプに入れて、

  • 親プロセスにそれらを読み取らせ、proc2が作成される前にproc2に渡すか、
  • proc2にそれらを読み込ませ、ローカルに設定する

後者には、proc2からの協力が必要です。前者は、proc2が開始される前に変数が既知になる必要があります。

2つのことが頭に浮かぶ：（1）プロセスを同じプロセスに何らかの方法で組み合わせてプロセスを同じ環境に共有させるか、（2）最初のプロセスに関連する環境変数を含む出力を生成させるそれを読んで、2番目のプロセスの環境を構築します。あなたが望んでいるように、サブプロセスから環境を取得する方法はないと私は思います（100％確信はありませんが）。

Python標準モジュールのマルチプロセッシングには、pickleを渡すことができるキューシステムがあります。プロセスを介して渡される-ableオブジェクト。また、プロセスはos.pipeを使用してメッセージ（ピクルスオブジェクト）を交換できます。リソース（例：データベース接続）とハンドル（例：ファイルハンドル）はピクルできないことに注意してください。

このリンクは面白いと思うかもしれません： マルチプロセスを使用したプロセス間の通信

言及する価値のあるマルチプロセッシングに関するPyMOTwも： マルチプロセッシングの基本

つづりが申し訳ありません

環境は親プロセスから継承されます。サブプロセス（子）ではなく、メインスクリプトで必要な環境を設定します。

ライセンス： CC-BY-SA と 帰属
所属していません StackOverflow
scroll top