PowerShell Global:Prompt函数输出捕获异常与旧提示符兼容问题咨询

问题描述 #

当前$function:Global:Prompt函数逻辑如下：通过Write-Host输出带绿色ANSI序列的Python虚拟环境前缀，再调用旧提示符生成剩余内容：

Write-Host -NoNewline -ForegroundColor Green "($_PYTHON_VENV_PROMPT_PREFIX) "
_OLD_VIRTUAL_PROMPT

终端中能正常显示完整提示符：

(.venv)  ~/projects/test-project/untitled :: git(main)             3.9.13 00:11:51

但遇到两个问题：

1. 执行$function:Global:Prompt.Invoke() > output.txt时，输出文件缺失绿色(.venv)部分的ANSI序列，仅保留旧提示符的内容。
2. 使用自定义MyCaptureOutput函数捕获输出时，(.venv)被追加到结果末尾，顺序完全错乱。

核心原因 #

• Write-Host默认直接写入控制台宿主，而非PowerShell标准输出流，因此>重定向无法捕获这部分内容。
• 自定义捕获函数中，Write-Host的控制台写入操作与_OLD_VIRTUAL_PROMPT的标准输出返回存在顺序异步问题，导致内容位置颠倒。

解决方案 #

方法1：修改Prompt函数，统一输出到标准输出 #

将Write-Host替换为手动构造ANSI颜色码的字符串返回，让所有内容进入标准输出流，重定向时可完整捕获：

function Global:Prompt {
  # 手动拼接绿色ANSI序列前缀
  $greenVenvPrefix = "`e[32m($_PYTHON_VENV_PROMPT_PREFIX) `e[0m"
  # 合并旧提示符内容并返回
  $greenVenvPrefix + $(_OLD_VIRTUAL_PROMPT)
}

修改后执行$function:Global:Prompt.Invoke() > output.txt，文件中将包含完整的带ANSI序列的提示符内容。

方法2：改进捕获函数，同步合并两类输出 #

如果无法修改原Prompt函数，可通过分别捕获控制台写入和标准输出，再按顺序合并的方式解决：

function CaptureFullPrompt {
  # 捕获Write-Host写入控制台的内容
  $consoleBuffer = New-Object System.Text.StringBuilder
  $stringWriter = New-Object System.IO.StringWriter($consoleBuffer)
  $originalConsoleOut = [Console]::Out
  [Console]::SetOut($stringWriter)

  # 捕获标准输出流的内容（旧提示符返回值）
  $streamOutput = & { $function:Global:Prompt.Invoke() }

  # 恢复原控制台输出流
  [Console]::SetOut($originalConsoleOut)

  # 按正确顺序合并两类输出
  return $consoleBuffer.ToString() + $streamOutput
}

# 调用测试
CaptureFullPrompt

方法3：临时使用内置转录功能捕获 #

如果仅需临时获取完整输出，可使用PowerShell内置的Start-Transcript：

Start-Transcript -Path "full_prompt.txt" -NoClobber
$function:Global:Prompt.Invoke()
Stop-Transcript

转录文件会包含完整的带ANSI序列的提示符，但需自行过滤转录自带的日志头信息。

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