Jak ustawić zmienne środowiskowe w Pythonie

Muszę ustawić niektóre zmienne środowiskowe w skrypcie Pythona i chcę, aby wszystkie inne skrypty wywoływane z Pythona widziały ustawione zmienne środowiskowe.

Jeśli zrobię

os.environ["DEBUSSY"] = 1`

narzeka, że 1musi to być sznurek.

Chcę również wiedzieć, jak odczytać zmienne środowiskowe w pythonie (w dalszej części skryptu) po ustawieniu.

Zmienne środowiskowe muszą być łańcuchami, więc używaj

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

aby ustawić zmienną DEBUSSYna ciąg 1.

Aby uzyskać dostęp do tej zmiennej później, po prostu użyj:

print(os.environ["DEBUSSY"])

Procesy potomne automatycznie dziedziczą zmienne środowiskowe procesu macierzystego - nie jest wymagane żadne specjalne działanie z Twojej strony.

Konieczne może być rozważenie dalszych aspektów niezawodności kodu;

gdy przechowujesz zmienną o wartości całkowitej jako zmienną środowiskową, spróbuj

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

następnie w celu pobrania weź pod uwagę, że aby uniknąć błędów, powinieneś spróbować

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

ewentualnie zamień „-1” na „Not Set”

podsumowując to wszystko

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.environzachowuje się jak słownik Pythona, więc można wykonywać wszystkie typowe operacje słownikowe. Oprócz operacji geti setwymienionych w innych odpowiedziach możemy również po prostu sprawdzić, czy istnieje klucz. Klucze i wartości powinny być przechowywane jako ciągi znaków .

Python 3

W Pythonie 3 słowniki używają słowa kluczowego in zamiast has_key

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

Python 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'

Należy pamiętać o jednej ważnej rzeczy os.environ:

Chociaż procesy potomne dziedziczą środowisko po procesie nadrzędnym, niedawno natknąłem się na problem i zorientowałem się, że jeśli inne skrypty aktualizują środowisko podczas działania skryptu python, wywołanie os.environponownie nie będzie odzwierciedlało najnowszych wartości .

Fragment dokumentów :

To odwzorowanie jest przechwytywane przy pierwszym imporcie modułu os, zazwyczaj podczas uruchamiania Pythona w ramach przetwarzania site.py. Zmiany w środowisku wprowadzone po tym czasie nie są odzwierciedlane w os.environ, z wyjątkiem zmian dokonanych przez bezpośrednią modyfikację os.environ.

os.environ.data który przechowuje wszystkie zmienne środowiskowe, jest obiektem dict, który zawiera wszystkie wartości środowiskowe:

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

jeśli zrobię os.environ ["DEBUSSY"] = 1, narzeka, że ​​1 musi być ciągiem znaków.

Więc zrób

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

Chcę również wiedzieć, jak odczytać zmienne środowiskowe w pythonie (w dalszej części skryptu) po ustawieniu.

Po prostu użyj os.environ["DEBUSSY"], jak w

some_value = os.environ["DEBUSSY"]

ustawić zmienną:

Metoda przypisania przedmiotu za pomocą klucza:

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

uzyskać lub sprawdzić, czy istnieje, czy nie,

ponieważ os.environ jest instancją, możesz wypróbować sposób obiektowy.

Metoda 1:

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

otrzyma '1'jako wartość zwracaną

Metoda 2:

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

Metoda 3:

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

Metoda 4:

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

Powinieneś przypisać wartość ciągu do zmiennej środowiskowej.

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

Jeśli chcesz odczytać lub wydrukować zmienną środowiskową, po prostu użyj

print os.environ["DEBUSSY"]

Zmiany będą obowiązywać tylko w bieżącym procesie, do którego zostały przypisane, nie spowoduje trwałej zmiany wartości. Procesy potomne automatycznie odziedziczą środowisko procesu macierzystego.

Co os.environ["DEBUSSY"] = '1'? Zmienne środowiskowe są zawsze łańcuchami.

Próbowałem dodać zmienne środowiskowe. Moim celem było przechowywanie niektórych informacji o użytkownikach w zmiennych systemowych, dzięki czemu mogę używać tych zmiennych w przyszłych rozwiązaniach, jako alternatywę dla plików konfiguracyjnych. Jednak metoda opisana w poniższym kodzie wcale mi nie pomogła.

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")

Ten prosty blok kodu działa dobrze, jednak te zmienne istnieją w odpowiednich procesach, tak że nie znajdziesz ich w zakładce zmiennych środowiskowych w ustawieniach systemu Windows. Prawie powyższy kod nie służył mojemu celowi. Ten problem jest omawiany tutaj: problem ze składowaniem zmiennej

os.environ.putenv(key, value)

Kolejna nieudana próba. Wreszcie udało mi się pomyślnie zapisać zmienną w rejestrze środowiska Windows, naśladując polecenia powłoki systemu Windows zawarte w klasie systemowej pakietu OS. Poniższy kod opisuje tę udaną próbę.

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

Mam nadzieję, że będzie to pomocne dla niektórych z was.

Należy zauważyć, że jeśli spróbujesz ustawić zmienną środowiskową na ocenę bash, nie zapiszesz tego, czego oczekujesz. Przykład:

from os import environ

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

Nie będzie to oceniać tak jak w powłoce, więc zamiast uzyskać /Library/Java/JavaVirtualMachines/jdk1.8.0_144.jdk/Contents/Homeścieżkę, otrzymasz dosłowne wyrażenie $(/usr/libexec/java_home).

Pamiętaj, aby to ocenić przed ustawieniem zmiennej środowiskowej, na przykład:

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}")

Możesz użyć os.environsłownika, aby uzyskać dostęp do zmiennych środowiskowych.

Problem, który miałem, polega na tym, że gdybym spróbował os.systemuruchomić plik wsadowy, który ustawia zmienne środowiskowe (używając polecenia SET w pliku **. Bat *), tak naprawdę nie ustawiłby ich dla twojego środowiska python (ale dla proces potomny utworzony za pomocą os.systemfunkcji). Aby faktycznie uzyskać zmienne ustawione w środowisku python, używam tego skryptu:

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

Podczas gry ze zmiennymi środowiskowymi (dodawaj / modyfikuj / usuwaj zmienne) dobrą praktyką jest przywrócenie poprzedniego stanu po zakończeniu funkcji.

Możesz potrzebować czegoś takiego, jak modified_environmenedżer kontekstu opisany w tym pytaniu, aby przywrócić zmienne środowiskowe.

Klasyczne zastosowanie:

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