最近写在一个 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)
