español - subprocess python 3 example

¿Cómo obtener el medio ambiente de un subproceso?

Question

Quiero llamar a un proceso a través de un programa python, sin embargo, este proceso necesita algunas variables de entorno específicas que se configuran mediante otro proceso. ¿Cómo puedo obtener las primeras variables de entorno de proceso para pasarlas a la segunda?

Así es como se ve el programa:

import subprocess

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

Pero el proceso no comparte el mismo entorno. Tenga en cuenta que estos programas no son míos (el primero es un archivo .bat grande y feo y el segundo un software propietario), por lo que no puedo modificarlos (está bien, puedo extraer todo lo que necesito del archivo .bat, pero es muy atractivo). ).

NB: Estoy usando Windows, pero prefiero una solución multiplataforma (pero mi problema no sucedería en un sistema similar a Unix ...)

Answer 1

Como aparentemente estás en Windows, necesitas una respuesta de Windows.

Crear un archivo por lotes de contenedor, por ejemplo. "run_program.bat", y ejecute ambos programas:

@echo off
call proc1.bat
proc2

El script se ejecutará y establecerá sus variables de entorno. Ambos scripts se ejecutan en el mismo intérprete (instancia cmd.exe), por lo que las variables prog1.bat se establecerán cuando se ejecute prog2.

No es terriblemente bonito, pero funcionará.

(Gente de Unix, puede hacer lo mismo en un script de bash: "archivo de origen.sh").

Answer 2

Como dices, los procesos no comparten el entorno, por lo que lo que pides literalmente no es posible, no solo en Python, sino en cualquier lenguaje de programación.

Lo que puede hacer es poner las variables de entorno en un archivo, o en una tubería, y cualquiera

• hacer que el proceso padre los lea, y pasarlos a proc2 antes de crear proc2, o
• Haga que proc2 los lea y los establezca localmente.

Este último requeriría la cooperación de proc2; el primero requiere que las variables se conozcan antes de iniciar proc2.

Answer 3

Dos cosas vienen a la mente: (1) hacer que los procesos compartan el mismo entorno, combinándolos de alguna manera en el mismo proceso, o (2) hacer que el primer proceso produzca resultados que contengan las variables de entorno relevantes, de esa manera Python puede leerlo y Construir el entorno para el segundo proceso. Creo (aunque no estoy 100% seguro) que no hay ninguna forma de obtener el entorno de un subproceso como espera.

Answer 4

El entorno se hereda del proceso padre. Establezca el entorno que necesita en el script principal, no un subproceso (hijo).

Answer 5

Este es un ejemplo de cómo puede extraer variables de entorno de un archivo por lotes o cmd sin crear un script de envoltura. Disfrutar.

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

Entonces, para responder a su pregunta, crearía un archivo .py que haga lo siguiente:

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

Answer 6

El multiprocesamiento del módulo estándar de Python tiene un sistema de colas que le permite pasar objetos que se pueden pasar a través de procesos. También los procesos pueden intercambiar mensajes (un objeto encurtido) utilizando os.pipe. Recuerde que los recursos (p. Ej., Conexión a la base de datos) y el manejador (p. Ej.: Mangos de archivo) no se pueden decapar.

Puede encontrar este enlace interesante: Comunicación entre procesos con multiprocesamiento.

También el PyMOTw sobre multiprocesamiento vale la pena mencionar: aspectos básicos de multiprocesamiento

lo siento por mi ortografía