Cách đặt biến môi trường trong Python

Tôi cần đặt một số biến môi trường trong tập lệnh python và tôi muốn tất cả các tập lệnh khác được gọi từ python để xem các biến môi trường được đặt.

Nếu tôi làm

os.environ["DEBUSSY"] = 1`

Nó phàn nàn rằng đó 1phải là chuỗi.

Tôi cũng muốn biết làm thế nào để đọc các biến môi trường trong python (trong phần sau của tập lệnh) sau khi tôi đặt nó.

Biến môi trường phải là chuỗi, vì vậy sử dụng

os.environ["DEBUSSY"] = "1"

để đặt biến DEBUSSYcho chuỗi 1.

Để truy cập biến này sau, chỉ cần sử dụng:

print(os.environ["DEBUSSY"])

Các tiến trình con tự động kế thừa các biến môi trường của tiến trình cha - không yêu cầu hành động đặc biệt nào từ phía bạn.

Bạn có thể cần xem xét một số khía cạnh khác cho độ mạnh của mã;

khi bạn đang lưu trữ một biến có giá trị nguyên như một biến môi trường, hãy thử

os.environ['DEBUSSY'] = str(myintvariable)

sau đó để truy xuất, xem xét rằng để tránh lỗi, bạn nên thử

os.environ.get('DEBUSSY', 'Not Set')

có thể thay thế '-1' cho 'Chưa đặt'

vì vậy, để kết hợp tất cả lại với nhau

myintvariable = 1
os.environ['DEBUSSY'] = str(myintvariable)
strauss = int(os.environ.get('STRAUSS', '-1'))
# NB KeyError <=> strauss = os.environ['STRAUSS']
debussy = int(os.environ.get('DEBUSSY', '-1'))

print "%s %u, %s %u" % ('Strauss', strauss, 'Debussy', debussy)

os.environhoạt động như một từ điển python, vì vậy tất cả các hoạt động từ điển phổ biến có thể được thực hiện. Ngoài các hoạt động getvà setđược đề cập trong các câu trả lời khác, chúng tôi cũng có thể chỉ cần kiểm tra nếu một khóa tồn tại. Các khóa và giá trị nên được lưu trữ dưới dạng chuỗi .

Con trăn 3

Đối với python 3, từ điển sử dụng từ khóa trong thay vì has_key

>>> import os
>>> 'HOME' in os.environ  # Check an existing env. variable
True
...

Con trăn 2

>>> import os
>>> os.environ.has_key('HOME')  # Check an existing env. variable
True
>>> os.environ.has_key('FOO')   # Check for a non existing variable
False
>>> os.environ['FOO'] = '1'     # Set a new env. variable (String value)
>>> os.environ.has_key('FOO')
True
>>> os.environ.get('FOO')       # Retrieve the value
'1'

Có một điều quan trọng cần lưu ý khi sử dụng os.environ:

Mặc dù các tiến trình con kế thừa môi trường từ tiến trình cha, nhưng tôi đã gặp phải một vấn đề gần đây và nhận ra, nếu bạn có các tập lệnh khác cập nhật môi trường trong khi tập lệnh python của bạn đang chạy, hãy gọi os.environ lại sẽ không phản ánh các giá trị mới nhất .

Trích từ tài liệu :

Ánh xạ này được ghi lại lần đầu tiên mô-đun os được nhập, thường là trong quá trình khởi động Python như là một phần của quá trình xử lý trang web. Những thay đổi đối với môi trường được thực hiện sau thời gian này không được phản ánh trong os.envir, ngoại trừ những thay đổi được thực hiện bằng cách sửa đổi trực tiếp os.envir.

os.environ.data lưu trữ tất cả các biến môi trường, là một đối tượng dict, chứa tất cả các giá trị môi trường:

>>> type(os.environ.data)  # changed to _data since v3.2 (refer comment below)
<type 'dict'>

nếu tôi làm os.envir ["DEBUSSY"] = 1, nó phàn nàn rằng 1 phải là chuỗi.

Sau đó làm

os.environ["DEBUSSY"] = "1"

Tôi cũng muốn biết làm thế nào để đọc các biến môi trường trong python (trong phần sau của tập lệnh) sau khi tôi đặt nó.

Chỉ cần sử dụng os.environ["DEBUSSY"], như trong

some_value = os.environ["DEBUSSY"]

để đặt biến:

Mục Phương thức gán bằng phím:

import os    
os.environ['DEBUSSY'] = '1'  #Environ Variable must be string not Int

để có được hoặc kiểm tra xem nó có tồn tại hay không,

vì os.envir là một ví dụ bạn có thể thử cách đối tượng.

Cách 1:

os.environ.get('DEBUSSY') # this is error free method if not will return None by default

sẽ nhận được '1'như là giá trị trả lại

Cách 2:

os.environ['DEBUSSY'] # will throw an key error if not found!

Cách 3:

'DEBUSSY' in os.environ  # will return Boolean True/False

Cách 4:

os.environ.has_key('DEBUSSY') #last 2 methods are Boolean Return so can use for conditional statements

Bạn nên gán giá trị chuỗi cho biến môi trường.

os.environ["DEBUSSY"] = "1"

Nếu bạn muốn đọc hoặc in biến môi trường, chỉ cần sử dụng

print os.environ["DEBUSSY"]

Những thay đổi này sẽ chỉ có hiệu lực đối với quy trình hiện tại nơi nó được chỉ định, nó sẽ không thay đổi giá trị vĩnh viễn. Các tiến trình con sẽ tự động kế thừa môi trường của tiến trình cha.

Thế còn os.environ["DEBUSSY"] = '1'? Biến môi trường luôn là chuỗi.

Tôi đã cố gắng thêm các biến môi trường. Mục tiêu của tôi là lưu trữ một số thông tin người dùng cho các biến hệ thống để tôi có thể sử dụng các biến đó cho các giải pháp trong tương lai, như là một thay thế cho các tệp cấu hình. Tuy nhiên, phương pháp được mô tả trong đoạn mã dưới đây không giúp tôi chút nào.

import os
os.environ["variable_1"] = "value_1"
os.environ["variable_2"] = "value_2"
# To Verify above code
os.environ.get("variable_1")
os.environ.get("variable_2")

Khối mã đơn giản này hoạt động tốt, tuy nhiên, các biến này tồn tại bên trong các quy trình tương ứng sao cho bạn sẽ không tìm thấy chúng trong tab biến môi trường của cài đặt hệ thống windows. Khá nhiều mã trên không phục vụ mục đích của tôi. Vấn đề này được thảo luận ở đây: vấn đề lưu biến

os.environ.putenv(key, value)

Một nỗ lực không thành công. Vì vậy, cuối cùng, tôi đã quản lý để lưu biến thành công bên trong thanh ghi môi trường cửa sổ bằng cách bắt chước các lệnh shell của windows được gói bên trong lớp hệ thống của gói os. Các mã sau đây mô tả nỗ lực thành công này.

os.system("SETX {0} {1} /M".format(key, value))

Tôi hy vọng điều này sẽ hữu ích cho một số bạn.

Cần lưu ý rằng nếu bạn cố gắng đặt biến môi trường thành đánh giá bash, nó sẽ không lưu trữ những gì bạn mong đợi. Thí dụ:

from os import environ

environ["JAVA_HOME"] = "$(/usr/libexec/java_home)"

Điều này sẽ không đánh giá nó giống như trong một cái vỏ, vì vậy thay vì đi /Library/Java/JavaVirtualMachines/jdk1.8.0_144.jdk/Contents/Homenhư một con đường, bạn sẽ có được biểu thức nghĩa đen $(/usr/libexec/java_home).

Đảm bảo đánh giá nó trước khi đặt biến môi trường, như vậy:

from os import environ
from subprocess import Popen, PIPE

bash_variable = "$(/usr/libexec/java_home)"
capture = Popen(f"echo {bash_variable}", stdout=PIPE, shell=True)
std_out, std_err = capture.communicate()
return_code = capture.returncode

if return_code == 0:
    evaluated_env = std_out.decode().strip()
    environ["JAVA_HOME"] = evaluated_env
else:
    print(f"Error: Unable to find environment variable {bash_variable}")

Bạn có thể sử dụng os.environtừ điển để truy cập các biến môi trường của bạn.

Bây giờ, một vấn đề tôi gặp phải là nếu tôi cố sử dụng os.systemđể chạy một tệp bó đặt các biến môi trường của bạn (sử dụng lệnh SET trong tệp **. Bat *) thì nó sẽ không thực sự đặt chúng cho môi trường python của bạn (nhưng đối với quá trình con được tạo với os.systemhàm). Để thực sự có được các biến được đặt trong môi trường python, tôi sử dụng tập lệnh này:

import re
import system
import os

def setEnvBat(batFilePath, verbose = False):
    SetEnvPattern = re.compile("set (\w+)(?:=)(.*)$", re.MULTILINE)
    SetEnvFile = open(batFilePath, "r")
    SetEnvText = SetEnvFile.read()
    SetEnvMatchList = re.findall(SetEnvPattern, SetEnvText)

    for SetEnvMatch in SetEnvMatchList:
        VarName=SetEnvMatch[0]
        VarValue=SetEnvMatch[1]
        if verbose:
            print "%s=%s"%(VarName,VarValue)
        os.environ[VarName]=VarValue

Khi bạn chơi với các biến môi trường (thêm / sửa đổi / loại bỏ biến), một thực hành tốt là khôi phục trạng thái trước đó khi hoàn thành chức năng.

Bạn có thể cần một cái gì đó giống như trình modified_environquản lý bối cảnh mô tả trong câu hỏi này để khôi phục các biến môi trường.

Cách sử dụng cổ điển:

with modified_environ(DEBUSSY="1"):
    call_my_function()