Subproses Python / Popen dengan lingkungan yang dimodifikasi

Saya percaya bahwa menjalankan perintah eksternal dengan lingkungan yang sedikit dimodifikasi adalah kasus yang sangat umum. Begitulah cara saya cenderung melakukannya:

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

Saya punya firasat bahwa ada cara yang lebih baik; apakah ini terlihat baik-baik saja?

Saya pikir os.environ.copy()lebih baik jika Anda tidak bermaksud memodifikasi os.environ untuk proses saat ini:

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

Itu tergantung pada apa masalahnya. Jika ingin mengkloning dan memodifikasi lingkungan, salah satu solusinya adalah:

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

Tapi itu agak tergantung pada variabel yang diganti adalah pengidentifikasi python yang valid, yang paling sering adalah (seberapa sering Anda mengalami nama variabel lingkungan yang bukan garis bawah + alfanumerik atau variabel yang dimulai dengan angka?).

Kalau tidak, Anda bisa menulis sesuatu seperti:

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

Dalam kasus yang sangat aneh (seberapa sering Anda menggunakan kode kontrol atau karakter non-ascii dalam nama variabel lingkungan?) Bahwa kunci-kunci lingkungan bytesAnda tidak dapat (pada python3) bahkan menggunakan konstruksi itu.

Seperti yang Anda lihat teknik (terutama yang pertama) yang digunakan di sini manfaat pada kunci lingkungan biasanya adalah pengidentifikasi python yang valid, dan juga dikenal sebelumnya (pada waktu pengkodean), pendekatan kedua memiliki masalah. Dalam kasus di mana bukan itu masalahnya Anda mungkin harus mencari pendekatan lain .

Anda dapat menggunakannya my_env.get("PATH", '')sebagai pengganti my_env["PATH"]jika dalam kasus PATHentah bagaimana tidak didefinisikan di lingkungan asli, tetapi selain itu terlihat baik-baik saja.

Dengan Python 3.5 Anda bisa melakukannya dengan cara ini:

import os
import subprocess

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

subprocess.Popen(my_command, env=my_env)

Di sini kita berakhir dengan salinan os.environdan PATHnilai yang diganti .

Itu dimungkinkan oleh PEP 448 (Generalisasi Pembongkaran Tambahan).

Contoh lain. Jika Anda memiliki lingkungan default (yaitu os.environ), dan dikt Anda ingin mengganti default dengan, Anda dapat mengekspresikannya seperti ini:

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

Untuk sementara mengatur variabel lingkungan tanpa harus menyalin objek os.envrion dll, saya melakukan ini:

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

Parameter env menerima kamus. Anda cukup mengambil os.environ, menambahkan kunci (variabel yang Anda inginkan) (ke salinan dikt jika Anda harus) untuk itu dan menggunakannya sebagai parameter untuk Popen.

Saya tahu ini telah dijawab untuk beberapa waktu, tetapi ada beberapa poin yang beberapa mungkin ingin tahu tentang menggunakan PYTHONPATH daripada PATH dalam variabel lingkungan mereka. Saya telah menjabarkan penjelasan tentang menjalankan skrip python dengan cronjobs yang berhubungan dengan lingkungan yang dimodifikasi dengan cara yang berbeda ( ditemukan di sini ). Kupikir itu akan bermanfaat bagi mereka yang, seperti aku, membutuhkan sedikit lebih banyak daripada jawaban yang diberikan.

Dalam keadaan tertentu Anda mungkin hanya ingin mewariskan variabel lingkungan yang dibutuhkan subproses Anda, tetapi saya pikir Anda memiliki ide yang tepat secara umum (itulah cara saya melakukannya juga).