Subprocesso Python / Popen com um ambiente modificado

Acredito que executar um comando externo com um ambiente ligeiramente modificado é um caso muito comum. É assim que eu faço:

import subprocess, os
my_env = os.environ
my_env["PATH"] = "/usr/sbin:/sbin:" + my_env["PATH"]
subprocess.Popen(my_command, env=my_env)

Tenho a sensação de que há uma maneira melhor; parece bem?

Eu acho que os.environ.copy()é melhor se você não pretende modificar o os.environ para o processo atual:

import subprocess, os
my_env = os.environ.copy()
my_env["PATH"] = "/usr/sbin:/sbin:" + my_env["PATH"]
subprocess.Popen(my_command, env=my_env)

Isso depende de qual é o problema. Se for para clonar e modificar o ambiente, uma solução pode ser:

subprocess.Popen(my_command, env=dict(os.environ, PATH="path"))

Mas isso depende um pouco do fato de as variáveis ​​substituídas serem identificadores python válidos, o que geralmente é (com que frequência você encontra nomes de variáveis ​​de ambiente que não são alfanuméricos + sublinhado ou variáveis ​​que começam com um número?).

Caso contrário, você poderá escrever algo como:

subprocess.Popen(my_command, env=dict(os.environ, 
                                      **{"Not valid python name":"value"}))

No caso muito estranho (com que frequência você usa códigos de controle ou caracteres não-ascii em nomes de variáveis ​​de ambiente?) Que as chaves do ambiente são as que bytesvocê não pode (no python3) usar essa construção.

Como você pode ver as técnicas (especialmente a primeira) usadas aqui, os benefícios nas chaves do ambiente são normalmente identificadores python válidos e também conhecidos antecipadamente (no momento da codificação), a segunda abordagem tem problemas. Nos casos em que esse não é o caso, você provavelmente deve procurar outra abordagem .

você pode usar em my_env.get("PATH", '')vez de my_env["PATH"], caso PATHnão esteja definido de alguma forma no ambiente original, mas que não seja o que parece ser bom.

Com o Python 3.5, você pode fazer o seguinte:

import os
import subprocess

my_env = {**os.environ, 'PATH': '/usr/sbin:/sbin:' + os.environ['PATH']}

subprocess.Popen(my_command, env=my_env)

Aqui terminamos com uma cópia os.environe um PATHvalor substituído .

Foi possível pelo PEP 448 (Generalizações adicionais de desempacotamento).

Outro exemplo. Se você possui um ambiente padrão (por exemplo os.environ) e um ditado com o qual deseja substituir os padrões, você pode expressá-lo assim:

my_env = {**os.environ, **dict_with_env_variables}

Para definir temporariamente uma variável de ambiente sem precisar copiar o objeto os.envrion, etc, faço isso:

process = subprocess.Popen(['env', 'RSYNC_PASSWORD=foobar', 'rsync', \
'rsync://[email protected]::'], stdout=subprocess.PIPE)

O parâmetro env aceita um dicionário. Você pode simplesmente pegar o os.environ, adicionar uma chave (sua variável desejada) (a uma cópia do ditado, se necessário) e usar como parâmetro para Popen.

Sei que isso já foi respondido há algum tempo, mas há alguns pontos que alguns podem querer saber sobre o uso de PYTHONPATH em vez de PATH em sua variável de ambiente. Descrevi uma explicação da execução de scripts python com cronjobs que lida com o ambiente modificado de uma maneira diferente ( encontrada aqui ). Achei que seria de alguma utilidade para quem, como eu, precisava de um pouco mais do que a resposta fornecida.

Em certas circunstâncias, você pode querer passar apenas as variáveis ​​de ambiente de que seu subprocesso precisa, mas acho que você teve a idéia certa em geral (é assim que eu também faço).