Can a python script persistently change a Windows environment variable? (elegantly)

后端 未结 5 1991
失恋的感觉
失恋的感觉 2020-12-16 02:34

Following on from my previous question, is it possible to make a Python script which persistently changes a Windows environment variable?

Changes to os.environ do n

5条回答
  •  心在旅途
    2020-12-16 02:46

    This link provides a solution that uses the built-in winreg library.

    (copypasta)

    import sys
    from subprocess import check_call
    if sys.hexversion > 0x03000000:
        import winreg
    else:
        import _winreg as winreg
    
    class Win32Environment:
        """Utility class to get/set windows environment variable"""
    
        def __init__(self, scope):
            assert scope in ('user', 'system')
            self.scope = scope
            if scope == 'user':
                self.root = winreg.HKEY_CURRENT_USER
                self.subkey = 'Environment'
            else:
                self.root = winreg.HKEY_LOCAL_MACHINE
                self.subkey = r'SYSTEM\CurrentControlSet\Control\Session Manager\Environment'
    
        def getenv(self, name):
            key = winreg.OpenKey(self.root, self.subkey, 0, winreg.KEY_READ)
            try:
                value, _ = winreg.QueryValueEx(key, name)
            except WindowsError:
                value = ''
            return value
    
        def setenv(self, name, value):
            # Note: for 'system' scope, you must run this as Administrator
            key = winreg.OpenKey(self.root, self.subkey, 0, winreg.KEY_ALL_ACCESS)
            winreg.SetValueEx(key, name, 0, winreg.REG_EXPAND_SZ, value)
            winreg.CloseKey(key)
            # For some strange reason, calling SendMessage from the current process
            # doesn't propagate environment changes at all.
            # TODO: handle CalledProcessError (for assert)
            check_call('''\
    "%s" -c "import win32api, win32con; assert win32api.SendMessage(win32con.HWND_BROADCAST, win32con.WM_SETTINGCHANGE, 0, 'Environment')"''' % sys.executable)
    

提交回复
热议问题