如何在精简的PowerShell提示符中显示Python虚拟环境？

当Python虚拟环境激活时，系统会自动设置$env:VIRTUAL_ENV环境变量，我们可以基于这个变量将虚拟环境名称前置到精简提示符中，修改后的prompt函数如下：

function prompt {
    $venvSegment = if ($env:VIRTUAL_ENV) {
        "($(Split-Path -Path $env:VIRTUAL_ENV -Leaf)) "
    } else {
        ""
    }
    $venvSegment + 'PS ' + $(Get-Location | Split-Path -Leaf) + "> "
}

说明 #

• 激活虚拟环境时，$env:VIRTUAL_ENV会指向虚拟环境的目录路径，通过Split-Path -Leaf提取路径的最后一段，就是虚拟环境的名称，格式化为(venv)的形式。
• 未激活虚拟环境时，虚拟环境段为空字符串，提示符保持你原本的精简样式：PS Document> 。

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