Python sottoprocesso / Popen con un ambiente modificato

Credo che eseguire un comando esterno con un ambiente leggermente modificato sia un caso molto comune. È così che tendo a farlo:

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

Ho la sensazione che ci sia un modo migliore; sembra a posto?

Penso che os.environ.copy()sia meglio se non si intende modificare os.environ per il processo corrente:

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

Dipende da quale sia il problema. Se si tratta di clonare e modificare l'ambiente, una soluzione potrebbe essere:

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

Ma ciò dipende in qualche modo dal fatto che le variabili sostituite sono identificatori python validi, cosa che spesso accade (con quale frequenza si imbattono in nomi di variabili di ambiente che non sono alfanumerici + carattere di sottolineatura o variabili che iniziano con un numero?).

Altrimenti potresti scrivere qualcosa del tipo:

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

Nel caso molto strano (con quale frequenza usi codici di controllo o caratteri non ascii nei nomi delle variabili di ambiente?) Che le chiavi dell'ambiente bytesnon puoi (su python3) nemmeno usare quel costrutto.

Come puoi vedere, le tecniche (specialmente la prima) qui usate benefici sulle chiavi dell'ambiente normalmente sono validi identificatori python, e anche noti in anticipo (al momento della codifica), il secondo approccio ha dei problemi. Nei casi in cui non è così, dovresti probabilmente cercare un altro approccio .

potresti usare my_env.get("PATH", '')invece my_env["PATH"]nel caso in PATHqualche modo non definito nell'ambiente originale, ma a parte questo sembra a posto.

Con Python 3.5 puoi farlo in questo modo:

import os
import subprocess

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

subprocess.Popen(my_command, env=my_env)

Qui finiamo con una copia di os.environe PATHvalore sovrascritto .

È stato reso possibile da PEP 448 (Generalizzazioni di disimballaggio aggiuntive).

Un altro esempio. Se si dispone di un ambiente predefinito (ad es. os.environ) E un dict con cui si desidera sostituire le impostazioni predefinite, è possibile esprimerlo in questo modo:

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

Per impostare temporaneamente una variabile di ambiente senza dover copiare l'oggetto os.envrion ecc., Faccio questo:

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

Il parametro env accetta un dizionario. Puoi semplicemente prendere os.environ, aggiungere una chiave (la tua variabile desiderata) (a una copia del dict se necessario) e usarla come parametro per Popen.

So che è stato risposto per un po 'di tempo, ma ci sono alcuni punti che alcuni potrebbero voler sapere sull'uso di PYTHONPATH invece di PATH nella loro variabile di ambiente. Ho delineato una spiegazione dell'esecuzione di script Python con cronjobs che si occupa dell'ambiente modificato in un modo diverso ( trovato qui ). Ho pensato che sarebbe stato utile per coloro che, come me, avevano bisogno di qualcosa in più di questa risposta fornita.

In alcune circostanze potresti voler tramandare solo le variabili di ambiente di cui il tuo sottoprocesso ha bisogno, ma penso che tu abbia la giusta idea in generale (è così che lo faccio anche io).