Как установить переменные среды в Python

Мне нужно установить некоторые переменные окружения в скрипте python, и я хочу, чтобы все остальные скрипты, которые вызываются из python, видели набор переменных окружения.

Если я сделаю

os.environ["DEBUSSY"] = 1`

он жалуется, говоря, что 1должен быть строкой.

Я также хочу знать, как читать переменные окружения в python (в более поздней части скрипта), как только я его установлю.

Переменные среды должны быть строками, поэтому используйте

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

установить переменную DEBUSSYв строку 1.

Чтобы получить доступ к этой переменной позже, просто используйте:

print(os.environ["DEBUSSY"])

Дочерние процессы автоматически наследуют переменные среды родительского процесса - никаких специальных действий с вашей стороны не требуется.

Возможно, вам придется рассмотреть некоторые дополнительные аспекты надежности кода;

когда вы сохраняете целочисленную переменную как переменную окружения, попробуйте

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

затем для извлечения учтите, что во избежание ошибок следует попробовать

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

возможно, заменить «-1» на «Не установлено»

так что, чтобы положить все это вместе

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.environведет себя как словарь Python, так что все общие словарные операции могут быть выполнены. В дополнении к getи setоперациям , упомянутых в других ответах, мы можем просто проверить , если ключ существует. Ключи и значения должны храниться в виде строк .

Python 3

Для python 3 словари используют ключевое слово in вместо 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'

Есть одна важная вещь, которую стоит отметить при использовании os.environ :

Хотя дочерние процессы наследуют среду от родительского процесса, я недавно столкнулся с проблемой и выяснил, что если у вас есть другие сценарии, обновляющие среду во время работы сценария python, os.environповторный вызов не будет отражать последние значения .

Выдержка из документов :

Это сопоставление фиксируется при первом импорте модуля os, обычно во время запуска Python, как часть обработки site.py. Изменения в среде, сделанные после этого времени, не отражаются в os.environ, за исключением изменений, внесенных путем непосредственного изменения os.environ.

os.environ.data который хранит все переменные окружения, является объектом dict, который содержит все значения окружения:

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

если я делаю os.environ ["DEBUSSY"] = 1, он жалуется, говоря, что 1 должен быть строкой.

Тогда делай

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

Я также хочу знать, как читать переменные окружения в python (в более поздней части скрипта), как только я его установлю.

Просто используйте os.environ["DEBUSSY"], как в

some_value = os.environ["DEBUSSY"]

установить переменную:

Метод назначения элемента по ключу:

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

получить или проверить, существует ли он или нет,

с тех пор - это экземпляр, вы можете попробовать объектный путь.

Способ 1:

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

получите '1' качестве возвращаемого значения

Способ 2:

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

Способ 3:

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

Способ 4:

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

Вы должны присвоить строковое значение переменной среды.

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

Если вы хотите прочитать или распечатать переменную окружения, просто используйте

print os.environ["DEBUSSY"]

Эти изменения вступят в силу только для текущего процесса, которому он был назначен, и не изменит значение навсегда. Дочерние процессы автоматически наследуют среду родительского процесса.

Как насчет os.environ["DEBUSSY"] = '1'? Переменные среды всегда являются строками.

Я пытался добавить переменные среды. Моя цель состояла в том, чтобы сохранить некоторую пользовательскую информацию в системных переменных, чтобы я мог использовать эти переменные для будущих решений в качестве альтернативы файлам конфигурации. Однако метод, описанный в приведенном ниже коде, мне совсем не помог.

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

Этот простой блок кода работает хорошо, однако эти переменные существуют внутри соответствующих процессов, так что вы не найдете их на вкладке переменных среды системных настроек Windows. Приведенный выше код не служил моей цели. Эта проблема обсуждается здесь: проблема сохранения переменной

os.environ.putenv(key, value)

Еще одна неудачная попытка. Итак, наконец, мне удалось успешно сохранить переменную в регистре среды окна, имитируя команды оболочки Windows, заключенные в системный класс пакета os. Следующий код описывает эту успешную попытку.

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

Я надеюсь, что это будет полезно для некоторых из вас.

Следует отметить, что если вы попытаетесь установить переменную окружения для оценки bash, она не будет хранить то, что вы ожидаете. Пример:

from os import environ

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

Это не будет оцениваться как в оболочке, поэтому вместо /Library/Java/JavaVirtualMachines/jdk1.8.0_144.jdk/Contents/Homeпути вы получите буквальное выражение $(/usr/libexec/java_home).

Обязательно оцените его перед установкой переменной среды, например так:

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

Вы можете использовать os.environсловарь для доступа к переменным среды.

Теперь проблема, с которой я столкнулся, заключается в том, что если бы я попытался использовать os.systemдля запуска командного файла, который устанавливает переменные среды (используя команду SET в файле **. Bat *), он не установил бы их для вашей среды Python (но для дочерний процесс, который создается с помощью os.systemфункции). Чтобы получить переменные, установленные в среде Python, я использую этот скрипт:

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

Когда вы играете с переменными среды (добавляете / модифицируете / удаляете переменные), хорошей практикой является восстановление предыдущего состояния при завершении функции.

Вам может понадобиться что-то вроде modified_environконтекстного менеджера, описанного в этом вопросе для восстановления переменных среды.

Классическое использование:

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