it-swarm.com.ru

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

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

Если я делаю os.environ["DEBUSSY"] = 1, он жалуется, говоря, что 1 должен быть строкой. Я также хочу знать, как читать переменные окружения в python (в более поздней части скрипта), как только я его установлю.

477
prahallada

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

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

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

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

print(os.environ["DEBUSSY"])

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

579
Sven Marnach

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

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

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)
95
Mark

os.environ ведет себя как словарь python, поэтому можно выполнять все обычные словарные операции. В дополнение к операциям get и set, упомянутым в других ответах, мы также можем просто проверить, существует ли ключ

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'

Python 3

Для python 3 словари используют ключевое слово in вместо has_key

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

При использовании 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'>
42
sisanared

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

Тогда делай

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

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

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

some_value = os.environ["DEBUSSY"]
18
Jim Brissom

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

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

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

print os.environ["DEBUSSY"]

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

enter image description here

16
RejeeshChandran

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

14
ThiefMaster

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

метод назначения элемента с использованием ключа:

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

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

так как os.environ это экземпляр, который вы можете попробовать способом объекта.

Метод 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
10
Mohideen bin Mohammed

Вы можете использовать словарь 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
10
Ricardo Alejos

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

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

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

with modified_environ(DEBUSSY="1"):
    call_my_function()
8
Laurent LAPORTE