无root权限时在Bash提示符中格式化Python虚拟环境名称

通常我们会直接修改/bin/activate文件里的PS1变量来调整虚拟环境的提示符，但这个文件是只读状态无法编辑。可以通过配置~/.bash_profile（或~/.bashrc，依系统而定）来实现相同需求，无需改动activate文件。

原/bin/activate中的相关逻辑 #

原文件里通过以下代码修改提示符：

if [ -z "${VIRTUAL_ENV_DISABLE_PROMPT:-}" ] ; then
    _OLD_VIRTUAL_PS1="${PS1:-}"
    PS1="(python3.9.13_) ${PS1:-}"
    export PS1
fi

这段代码会在虚拟环境激活时，给原有提示符加上(python3.9.13_)前缀，前提是VIRTUAL_ENV_DISABLE_PROMPT变量为空。

实现步骤 #

1. 打开你的~/.bash_profile文件，添加以下代码：

# 禁用虚拟环境默认的提示符修改逻辑
export VIRTUAL_ENV_DISABLE_PROMPT=1

# 自定义虚拟环境提示符函数
function set_virtualenv_prompt {
    # 判断是否处于激活的虚拟环境中
    if [ -n "$VIRTUAL_ENV" ]; then
        # 从虚拟环境路径中提取环境名称（也可以直接写死成你需要的内容，比如venv_name="python3.9.13_"）
        venv_name=$(basename "$VIRTUAL_ENV")
        # 保存原有提示符（仅第一次激活时保存）
        if [ -z "${_OLD_VIRTUAL_PS1:-}" ]; then
            _OLD_VIRTUAL_PS1="${PS1:-}"
        fi
        # 设置带前缀的新提示符
        PS1="($venv_name) ${_OLD_VIRTUAL_PS1:-}"
    else
        # 退出虚拟环境时恢复原有提示符
        if [ -n "${_OLD_VIRTUAL_PS1:-}" ]; then
            PS1="${_OLD_VIRTUAL_PS1:-}"
            unset _OLD_VIRTUAL_PS1
        fi
    fi
}

# 让bash每次显示提示符前自动执行上述函数
PROMPT_COMMAND="set_virtualenv_prompt; $PROMPT_COMMAND"

2. 保存文件后，执行以下命令让配置生效：

source ~/.bash_profile

3. 激活你的虚拟环境，就能看到自定义的提示符了。如果需要固定前缀，把代码里的venv_name=$(basename "$VIRTUAL_ENV")替换成venv_name="python3.9.13_"即可。

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