如何编写Python程序修改Shell状态？参考Anaconda环境激活效果

首先要明确：Python脚本作为Shell的子进程，无法直接修改父Shell的环境变量（Shell的提示符由PS1环境变量控制），所以直接运行python setstate.py无法实现需求，必须通过让父Shell执行Python输出的Shell命令来完成。

可行实现方案 #

1. 编写Python脚本输出Shell命令

写一个setstate.py脚本，功能是生成修改PS1的Shell命令，让父Shell通过eval执行：

import sys

def set_state(state_name):
    # 输出兼容bash/zsh的PS1修改命令，把自定义状态前缀加到原有提示符前
    print(f'export PS1="({state_name}) $PS1"')

if __name__ == "__main__":
    if len(sys.argv) != 2:
        print("Usage: python setstate.py <state_name>")
        sys.exit(1)
    target_state = sys.argv[1]
    set_state(target_state)
    print(f'Set state to {target_state}')

2. 在Shell中执行生效

在终端里运行以下命令，就能让当前Shell的提示符变成(custom_state) ~/Desktop $的格式：

eval $(python setstate.py custom_state)

3. 封装成Shell函数（更便捷）

如果想像conda那样直接输入命令激活，可以把以下函数加到你的Shell配置文件（~/.bashrc或~/.zshrc）里：

setstate() {
    if [ $# -ne 1 ]; then
        echo "Usage: setstate <state_name>"
        return 1
    fi
    eval $(python /path/to/setstate.py "$1")
}

保存后重启Shell，之后直接输入setstate custom_state就能一键修改提示符。

原理说明 #

类似conda激活环境的效果，本质是conda通过Shell脚本直接在当前Shell进程中修改PS1环境变量；而Python作为子进程无法直接操作父进程的环境，所以通过输出Shell命令，让父Shell执行的方式间接实现修改。

注意事项 #

• 这种修改是临时生效的，关闭当前Shell会话后会恢复原来的提示符；如果需要永久修改，可直接把export PS1="(custom_state) $PS1"加到Shell配置文件中。
• 不同Shell的PS1格式可能有差异，上述方案兼容bash和zsh主流Shell。

内容的提问来源于stack exchange，提问作者Maxim Kozlov