如何在Python程序中获取生成标准输入的完整命令？

首先明确：管道（|）的本质是仅传递数据流，不会附带上游命令的任何元信息，所以无法直接从sys.stdin中获取上游命令。以下是几种可行的实现思路：

1. 利用进程信息遍历（Linux/macOS 专属） #

通过系统的进程树，找到与当前Python进程共享管道写端的上游进程，进而获取其完整命令行。需要依赖psutil库来简化进程信息的获取：

• 先安装依赖：

  pip install psutil
  

• 示例代码：

  import sys
  import psutil
  
  def get_upstream_command():
      current_proc = psutil.Process()
      parent_proc = current_proc.parent()
      # 遍历父进程（通常是shell）的所有子进程
      for child in parent_proc.children(recursive=False):
          try:
              # 检查子进程的stdout是否指向当前进程的stdin
              if child.stdout.fileno() == current_proc.stdin.fileno():
                  return ' '.join(child.cmdline())
          except (psutil.AccessDenied, psutil.NoSuchProcess, OSError):
              continue
      return None
  
  if __name__ == "__main__":
      if not sys.stdin.isatty():
          stdin_content = sys.stdin.read()
          upstream_cmd = get_upstream_command()
          print(f"读取到的内容: {stdin_content}")
          print(f"上游命令: {upstream_cmd}")
  

• 注意事项：
  • 仅适用于Linux/macOS，Windows平台的进程模型不同，此方法无效。
  • 若上游是复杂的管道链（比如echo Hello | grep H | python program.py），只能获取紧邻的上游进程（grep H），无法拿到完整的管道链。
  • 可能遇到权限问题，无法读取其他进程的文件描述符。

2. 让用户显式传递命令（跨平台最可靠） #

既然管道本身不传递命令信息，最稳妥的方式是让用户在调用时，将上游命令作为参数传给Python程序：

• 调用方式：

  echo Hello | python program.py "echo Hello"
  

• Python代码中获取：

  import sys
  
  if __name__ == "__main__":
      if not sys.stdin.isatty():
          stdin_content = sys.stdin.read()
          upstream_cmd = sys.argv[1] if len(sys.argv) > 1 else None
          print(f"读取到的内容: {stdin_content}")
          print(f"上游命令: {upstream_cmd}")
  

• 优点：
  • 完全跨平台，没有系统兼容性问题。
  • 可以准确获取完整的上游命令（包括复杂管道链）。
  • 实现简单，无需依赖第三方库。

3. 读取shell历史记录（不推荐） #

如果用户是在交互式shell中执行命令，可以尝试读取shell的历史文件（比如~/.bash_history、~/.zsh_history），但此方法局限性极大：

• 非交互式shell默认不会记录命令到历史文件。
• 历史文件的写入通常是延迟的，当前命令可能还没被写入。
• 无法确定历史记录中的哪条命令对应当前的管道调用。

内容的提问来源于stack exchange，提问作者Nauto the Boss