python修改windows環境變量

最近在學Django，看到代碼中有用os.environ修改 環境變量，但是在Windows中查看，卻沒有發現修改后的環境變量。遂在網上查找相關資料，找到這么一篇文章：

os.environ and os.putenv in Python

用Python Shell設置或獲取環境變量的方法：

一、設置系統環境變量

1、os.environ['環境變量名稱']='環境變量值' #其中key和value均為string類型

2、os.putenv('環境變量名稱', '環境變量值')

 

二、獲取系統環境變量

1、os.environ['環境變量名稱']

2、os.getenv('環境變量名稱')

 

以上方法，推薦用os.environ，因為使用os.putenv()並不會真正改變os.environ字典里面的環境變量，即某些平台無效，但是使用os.environ有一個潛在的隱患：在一些平台上，包括FreeBSD和Mac OS X，修改environ會導致內存泄露。詳情見Python API。

 

我們設置的環境變量只存在於當前的python shell中（設置成功后用print os.environ['環境變量名稱']或printos.getenv('環境變量名稱') 查看）。也就是說，比如Windows環境下，在"我的電腦"——“屬性”——“高級系統設置”——"高級"——"環境變量"中找不到剛才設置成功的環境變量。為什么會這樣呢，如何用Python真正設置環境變量？

 

如果你所在的開發環境是windows的操作系統，import _winreg模塊將環境變量寫入注冊表，再廣播WM_SETTINGCHANGE消息，可參考實例；如果你所在的開發環境是Linux的操作系統，使用linux命令，在bash_profile文件中添加環境變量后，使其生效即可。

以下是網址中給出的參考代碼：

import _winreg as winreg
import ctypes

ENV_HTTP_PROXY = u'http://87.254.212.121:8080'


class Registry(object):
    def __init__(self, key_location, key_path):
        self.reg_key = winreg.OpenKey(key_location, key_path, 0, winreg.KEY_ALL_ACCESS)

    def set_key(self, name, value):
        try:
            _, reg_type = winreg.QueryValueEx(self.reg_key, name)
        except WindowsError:
            # If the value does not exists yet, we (guess) use a string as the
            # reg_type
            reg_type = winreg.REG_SZ
        winreg.SetValueEx(self.reg_key, name, 0, reg_type, value)

    def delete_key(self, name):
        try:
            winreg.DeleteValue(self.reg_key, name)
        except WindowsError:
            # Ignores if the key value doesn't exists
            pass



class EnvironmentVariables(Registry):
    """
    Configures the HTTP_PROXY environment variable, it's used by the PIP proxy
    """

    def __init__(self):
        super(EnvironmentVariables, self).__init__(winreg.HKEY_LOCAL_MACHINE,
                                                   r'SYSTEM\CurrentControlSet\Control\Session Manager\Environment')

    def on(self):
        self.set_key('HTTP_PROXY', ENV_HTTP_PROXY)
        self.refresh()

    def off(self):
        self.delete_key('HTTP_PROXY')
        self.refresh()

    def refresh(self):
        HWND_BROADCAST = 0xFFFF
        WM_SETTINGCHANGE = 0x1A

        SMTO_ABORTIFHUNG = 0x0002

        result = ctypes.c_long()
        SendMessageTimeoutW = ctypes.windll.user32.SendMessageTimeoutW
        SendMessageTimeoutW(HWND_BROADCAST, WM_SETTINGCHANGE, 0, u'Environment', SMTO_ABORTIFHUNG, 5000, ctypes.byref(result));