2019-08-26 如何用python设置环境变量

在使用linux时，经常需要配置环境变量，有时候甚至需要在几套环境变量之间切换，这时候如果可以用python脚本来管理就会非常方便。

如果修改的环境变量在进程中生效就可以了，只需要使用os.environ设置：

import os

os.environ['JAVA_HOME'] = '/path/to/java_home'

但是当我们希望修改的环境变量在shell中生效时就不行了。
一种可行的方案是把设置环境变量的命令写到输出里，再使用eval执行。例如：

print('export JAVA_HOME=/path/to/java_home')

然后执行：

eval $(./setpath.py)

使用source也可以

source <(./setpath.py)

这样就可以通过脚本来修改环境变量，如果给脚本加上参数，就可以控制多套环境变量的切换了。