Как получить среду из подпроцесса в Python

Я хочу вызвать процесс через программу 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

Поскольку вы, видимо, в Windows, вам нужен ответ Windows.

Создайте пакетный файл обертки, например. "run_program.bat" и запустите обе программы:

@echo off
call proc1.bat
proc2

Запустится script и установите его переменные среды. Оба сценария работают в одном интерпретаторе (экземпляр cmd.exe), поэтому переменные prog1.bat устанавливаются при выполнении prog2.

Не очень красиво, но это сработает.

(Unix-люди, вы можете сделать то же самое в bash script: "source file.sh".)

Ответ 2

Вот пример того, как вы можете извлекать переменные среды из пакетного или cmd файла без создания обертки script. Наслаждайтесь.

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)

Ответ 3

Как вы говорите, процессы не обмениваются средой, поэтому то, что вы в буквальном смысле задаете, невозможно, не только на Python, но и на любом языке программирования.

Что вы можете сделать, это поместить переменные среды в файл или в трубку, а также

  • попросите родительский процесс прочитать их и передать их proc2 перед созданием proc2 или
  • прочитайте их proc2 и установите их локально

Последнее потребует сотрудничества от proc2; первый требует, чтобы переменные стали известны до начала proc2.

Ответ 4

Среда наследуется от родительского процесса. Задайте нужную среду в главном script, а не подпроцессе (дочернем).

Ответ 5

Две вещи spring для ума: (1) заставить процессы совместно использовать одну и ту же среду, путем объединения их каким-то образом в один и тот же процесс или (2) иметь первый процесс, производящий вывод, содержащий соответствующие переменные среды, таким образом Python может прочитать его и построить среду для второго процесса. Я думаю (хотя я не уверен на 100%), что нет способа получить среду из подпроцесса, как вы надеетесь сделать.

Ответ 6

Многопроцессорный модуль стандартного модуля Python имеет систему очередей, которая позволяет передавать передаваемый способ размножения через процессы. Также процессы могут обмениваться сообщениями (маринованным объектом) с помощью os.pipe. Помните, что ресурсы (например, подключение к базе данных) и дескриптор (например, файлы) не могут быть маринованными.

Вы можете найти эту ссылку интересной: Связь между процессами с многопроцессорностью

Также PyMOTw о многопроцессорности стоит упомянуть: основы многопроцессорности

извините за мое правописание