如何让Zsh始终显示Python虚拟环境名？解决重载Shell后虚拟环境名消失的问题

你的问题根源在于当前的Zsh提示符配置没有读取pyenv激活虚拟环境时设置的VIRTUAL_ENV环境变量——即使重载Shell后这个变量依然存在，但你的脚本没处理它，所以环境名就消失了。咱们来修改你的配置，让它能自动识别并显示虚拟环境名：

原理说明 #

当你用pyenv激活虚拟环境时，系统会设置VIRTUAL_ENV环境变量，其值是虚拟环境的完整路径（比如~/.pyenv/versions/my-ve-3.7.13）。我们可以通过Zsh的字符串截取功能${VIRTUAL_ENV:t}提取路径的最后一段，也就是你要的虚拟环境名。

修改后的完整配置脚本 #

把你原来的Zsh提示符配置替换成下面的代码：

setopt PROMPT_SUBST
autoload -Uz vcs_info
precmd() { vcs_info }
zstyle ':vcs_info:git:*' formats '(%b)'

MYPS1=''
MYPS1+='%F{green}'
MYPS1+='${${(%):-%n}:0:1}'
MYPS1+='@'
MYPS1+='${${(%):-%m}:(-4)}' # 获取主机名最后4个字符
MYPS1+=':'

# 新增：显示激活的Python虚拟环境名
if [[ -n "$VIRTUAL_ENV" ]]; then
  MYPS1+='%F{blue}('
  MYPS1+="${VIRTUAL_ENV:t}" # 提取虚拟环境名称
  MYPS1+=') '
fi

MYPS1+='%F{yellow}'
MYPS1+='%1~' # 只显示当前目录名或~（如果是家目录）
MYPS1+='%F{magenta}'
MYPS1+='${vcs_info_msg_0_}' # 显示Git分支（如果存在）
MYPS1+='%f'
MYPS1+='%# '
PS1=$MYPS1

关键修改点解释 #

• 我们在主机名之后、目录名之前加入了对VIRTUAL_ENV变量的判断：如果变量存在（即虚拟环境处于激活状态），就用蓝色字体显示带括号的虚拟环境名。
• ${VIRTUAL_ENV:t}是Zsh的特殊语法，t表示取路径的“尾部”（也就是最后一个目录名），刚好就是你的虚拟环境名称。

测试验证 #

1. 激活你的虚拟环境：pyenv activate my-ve-3.7.13
2. 重载Zsh配置：source ~/.zshrc
3. 此时你的提示符应该会变成类似这样：

u@m1:(my-ve-3.7.13) repo-name(github-branch)%

这样即使多次重载Shell，只要虚拟环境没退出，提示符就会一直显示环境名啦。

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