Sous-processus Python / Popen avec un environnement modifié

Je crois que l'exécution d'une commande externe avec un environnement légèrement modifié est un cas très courant. Voilà comment j'ai tendance à le faire:

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

J'ai l'intuition qu'il y a une meilleure façon; ça a l'air bien?

Je pense que os.environ.copy()c'est mieux si vous n'avez pas l'intention de modifier os.environ pour le processus actuel:

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

Cela dépend du problème. Pour cloner et modifier l'environnement, une solution pourrait être:

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

Mais cela dépend quelque peu du fait que les variables remplacées sont des identificateurs python valides, ce qu'elles sont le plus souvent (à quelle fréquence rencontrez-vous des noms de variables d'environnement qui ne sont pas alphanumériques + trait de soulignement ou des variables qui commencent par un nombre?).

Sinon, vous pourriez écrire quelque chose comme:

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

Dans le cas très étrange (à quelle fréquence utilisez-vous des codes de contrôle ou des caractères non-ascii dans les noms de variables d'environnement?) Que les clés de l'environnement ne sont bytespas vous (sur python3) même utiliser cette construction.

Comme vous pouvez voir les techniques (en particulier la première) utilisées ici, les avantages sur les clés de l'environnement sont normalement des identifiants python valides, et également connus à l'avance (au moment du codage), la deuxième approche a des problèmes. Dans les cas où ce n'est pas le cas, vous devriez probablement chercher une autre approche .

vous pouvez utiliser my_env.get("PATH", '')au lieu de my_env["PATH"]dans le cas où PATHnon défini dans l'environnement d'origine, mais à part cela, cela semble bien.

Avec Python 3.5, vous pouvez le faire de cette façon:

import os
import subprocess

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

subprocess.Popen(my_command, env=my_env)

Ici, nous nous retrouvons avec une copie de os.environet une PATHvaleur remplacée .

Il a été rendu possible par le PEP 448 (Générations de déballage supplémentaires).

Un autre exemple. Si vous avez un environnement par défaut (c.-à-d. os.environ) Et un dict avec lequel vous souhaitez remplacer les valeurs par défaut, vous pouvez l'exprimer comme ceci:

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

Pour définir temporairement une variable d'environnement sans avoir à copier l'objet os.envrion, etc., je fais ceci:

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

Le paramètre env accepte un dictionnaire. Vous pouvez simplement prendre os.environ, ajouter une clé (votre variable souhaitée) (à une copie du dict si vous le devez) et l'utiliser comme paramètre pour Popen.

Je sais que cela a été répondu depuis un certain temps, mais il y a certains points que certains voudront peut-être savoir sur l'utilisation de PYTHONPATH au lieu de PATH dans leur variable d'environnement. J'ai décrit une explication de l'exécution de scripts python avec cronjobs qui traite l'environnement modifié d'une manière différente ( trouvée ici ). Je pensais que ce serait quelque chose de bon pour ceux qui, comme moi, avaient besoin d'un peu plus que cette réponse.

Dans certaines circonstances, vous souhaiterez peut-être transmettre uniquement les variables d'environnement dont votre sous-processus a besoin, mais je pense que vous avez la bonne idée en général (c'est comme ça que je le fais aussi).