自定义ZSH提示符：显示pipenv虚拟环境、路径及Git分支

现有配置修改方案 #

你不需要额外安装组件，直接利用虚拟环境激活时自动设置的VIRTUAL_ENV环境变量即可实现需求，修改~/.zshrc中的对应配置为以下内容：

# Load version control information
autoload -Uz vcs_info

precmd() { 
  vcs_info
  # 处理虚拟环境显示逻辑
  if [[ -n "$VIRTUAL_ENV" ]]; then
    # 提取虚拟环境目录名作为展示名称，可根据需求调整格式
    virtualenv_info="($(basename "$VIRTUAL_ENV"))"
  else
    virtualenv_info=""
  fi
}

# Format the vcs_info_msg_0_ variable
zstyle ':vcs_info:git:*' formats '(%b)'

setopt PROMPT_SUBST
# 按需求调整提示符结构，这里移除了你原有配置中的用户名%n，需要可以自行加回
PROMPT='${virtualenv_info} ${PWD/#$HOME/~} ${vcs_info_msg_0_} > '

修改完成后执行source ~/.zshrc重载配置即可生效。

如果你的pipenv虚拟环境名称自带随机hash后缀，想只展示项目名，可以把virtualenv_info的赋值行替换为：

virtualenv_info="($(basename "$VIRTUAL_ENV" | cut -d'-' -f1))"

可以根据自己的命名规则调整分割规则。

其他可选实现方案 #

• 如果你使用oh-my-zsh框架，可以直接开启内置的virtualenv插件，插件会自动处理虚拟环境名称的提取逻辑，支持直接在主题的PROMPT配置中调用对应变量使用，不需要手动写判断逻辑。
• 不建议通过调用pipenv --venv等外部命令的方式获取虚拟环境信息，会增加提示符加载延迟，不如直接读取环境变量的效率高。

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