Quá trình con Python / Popen với môi trường được sửa đổi

Tôi tin rằng việc chạy một lệnh bên ngoài với môi trường được sửa đổi một chút là trường hợp rất phổ biến. Đó là cách tôi có xu hướng làm điều đó:

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

Tôi đã có một cảm giác ruột rằng có một cách tốt hơn; trông nó có ổn không?

Tôi nghĩ os.environ.copy()sẽ tốt hơn nếu bạn không có ý định sửa đổi os.envir cho quy trình hiện tại:

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

Điều đó phụ thuộc vào vấn đề là gì. Nếu nó nhân bản và sửa đổi môi trường, một giải pháp có thể là:

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

Nhưng điều đó phần nào phụ thuộc vào việc các biến được thay thế là các định danh python hợp lệ, mà chúng thường là nhất (bạn có thường chạy vào các tên biến môi trường không phải là chữ và số + gạch dưới hoặc các biến bắt đầu bằng một số không?).

Nếu không, bạn sẽ có thể viết một cái gì đó như:

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

Trong trường hợp rất kỳ lạ (bạn có thường xuyên sử dụng mã điều khiển hoặc ký tự không phải mã ascii trong tên biến môi trường không?) Rằng các khóa của môi trường là bytesbạn không thể (trên python3) thậm chí sử dụng cấu trúc đó.

Như bạn có thể thấy các kỹ thuật (đặc biệt là kỹ thuật đầu tiên) được sử dụng ở đây mang lại lợi ích cho các khóa của môi trường thông thường là các định danh python hợp lệ và cũng được biết trước (tại thời điểm mã hóa), cách tiếp cận thứ hai có vấn đề. Trong trường hợp đó không phải là trường hợp bạn có thể nên tìm cách tiếp cận khác .

bạn có thể sử dụng my_env.get("PATH", '')thay vì my_env["PATH"]trong trường hợp PATHbằng cách nào đó không được xác định trong môi trường ban đầu, nhưng khác hơn là nó trông ổn.

Với Python 3.5, bạn có thể làm theo cách này:

import os
import subprocess

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

subprocess.Popen(my_command, env=my_env)

Ở đây chúng tôi kết thúc với một bản sao của os.environ và PATHgiá trị ghi đè .

Nó đã được thực hiện bởi PEP 448 (Tổng quát hóa giải nén bổ sung).

Một vi dụ khac. Nếu bạn có một môi trường mặc định (nghĩa là os.environ) và một lệnh bạn muốn ghi đè mặc định, bạn có thể diễn đạt nó như sau:

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

Để tạm thời đặt biến môi trường mà không phải sao chép đối tượng os.envrion, v.v., tôi làm điều này:

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

Tham số env chấp nhận một từ điển. Bạn có thể chỉ cần lấy os.envir, thêm một khóa (biến mong muốn của bạn) (vào một bản sao của dict nếu bạn phải) vào đó và sử dụng nó làm tham số Popen.

Tôi biết điều này đã được trả lời một thời gian, nhưng có một số điểm mà một số người có thể muốn biết về việc sử dụng PYTHONPATH thay vì PATH trong biến môi trường của họ. Tôi đã phác thảo một lời giải thích về việc chạy các kịch bản python với các cronjobs liên quan đến môi trường đã sửa đổi theo một cách khác ( tìm thấy ở đây ). Nghĩ rằng nó sẽ có ích cho những người, giống như tôi, chỉ cần nhiều hơn một chút so với câu trả lời này được cung cấp.

Trong một số trường hợp nhất định, bạn có thể chỉ muốn chuyển xuống các biến môi trường mà quy trình con của bạn cần, nhưng tôi nghĩ rằng bạn đã có ý tưởng chung nói chung (đó là cách tôi cũng làm như vậy).