Python 永久修改环境变量

最近写在一个 windows 上快速切换 python 环境的软件（PySwitch）的时候遇到一个问题。就是用户下载后台，需要用户手动为软件添加环境变量，以便在任何地方都可以调起软件。后来一想，手动添加环境变量太麻烦了，为什么不在安装软件初始化的时候自动去添加环境变量呢。想到就动手...

于是就百度了一圈，查到可以通过 os.environ 中的 os.environ.setdefault 方法可以修改系统环境变量。经过实验使用后发现，os.environ 只能影响到当前运行程序，无法保存到系统中。无耐只好放弃...

本着不抛弃不放弃的理念，想到 windos 是可以通过 dos 命令 来改变环境变量的，所以想到了一个曲线救国的方法，通过 python 来运行 dos 命令来修改系统环境变量。通过查找资料，发现可以通过 setx 和 reg add（修改注册表） 两命令来修改系统环境变量。

STEX 和 REG ADD 的优劣

• stex ：stex 命令功能强大，使用较为简单，但 setx 有截断 1024 字符设置，所以超过 1024 字符无法使用 stex 命令
• reg add ：没有字符截断，但使用上较 setx 繁琐

setx 设置环境变量

# 用户环境变量
setx    key  value

# 系统环境变量
setx /M    key  value

reg add 设置环境变量

# 用户环境变量注册表位置
set regPath= HKEY_CURRENT_USER\Environment

# 系统环境变量注册表位置
set regPath= HKEY_LOCAL_MACHINE\SYSTEM\CurrentControlSet\Control\Session" "Manager\Environment

reg add %regPath% /v %key% /t REG_EXPAND_SZ /d "%oldValue%;%value%" /f

在 Python 中使用

SET_ENV=r'''
@echo
set %{key}%={value}
echo %{key}%

if {user}==sys (
    setx /M｛key｝ "%{key}%"
) else (
    setx {key} "%{key}%"
)
'''


ADD_ENV=r'''
@echo off

if {user}==sys (
    set regPath= HKEY_LOCAL_MACHINE\SYSTEM\CurrentControlSet\Control\Session" "Manager\Environment
) else (
    set regPath= HKEY_CURRENT_USER\Environment
)

set key={key}
set value={value}
:: 判断是否存在该路径
reg query %regPath% /v %key% 1>nul 2>nul
if %ERRORLEVEL%==0 (
    :: 取值
    For /f "tokens=3* delims= " %%i in ('Reg Query %regPath% /v %key% ') do (
        if "%%j"=="" (Set oldValue=%%i) else (Set oldValue=%%i %%j)
    )
) else Set oldValue="."

:: 备份注册表
reg export %regPath% %~dp0%~n0.reg
:: 写入环境变量
if "%oldValue%"=="." (
    reg add %regPath% /v %key% /t REG_EXPAND_SZ /d "%value%" /f
) else (
    reg add %regPath% /v %key% /t REG_EXPAND_SZ /d "%oldValue%;%value%" /f
)
'''

# 连续运行 bat 命令
def runbat(self, bat):
    tmp_bat_file = 'tmp.bat'
    with open(tmp_bat_file, 'w') as f:
        f.writelines(bat)
    self.runcmd(tmp_bat_file)


# 添加环境变量
def _append_env(self, is_sys, key, value):
    # 运行设置环境变量命令
    user = 'sys' if is_sys else 'me'
    cmds = ADD_ENV.format(user=user, key=key, value=value)
    self.runbat(cmds)


# 设置环境变量
def _set_env(self, is_sys, key, value):
    user = 'sys' if is_sys else 'me'
    cmds = SET_ENV.format(user=user, key=key, value=value)
    self.runbat(cmds)