Python subprocess 模块与Shell交互的进阶技巧
在Python系统管理或自动化任务中，经常需要与Shell交互。本文探讨如何使用subprocess模块高效地与Shell进行多次交互，如同在终端中一样灵活地执行命令并获取结果。
许多开发者尝试使用subprocess.Popen模拟Shell交互，却常遇到代码卡死或无法获取预期输出的问题。例如，直接用subprocess.Popen启动Shell进程（如/bin/bash），然后循环输入命令，往往得不到正确响应。
根本原因在于，Shell命令通常以换行符(\n)作为结束标志。仅发送命令字符串，Shell无法识别命令结束，导致命令执行失败，程序陷入阻塞。
立即学习“Python免费学习笔记（深入）”；
让我们分析一段典型的错误代码：
import subprocess
process = subprocess.Popen('/bin/bash', stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
while True:
cmd = input("$ ")
if cmd == "exit":
break
cmd = cmd.encode('utf-8')
print('cmd',cmd)
process.stdin.write(cmd)
process.stdin.flush()
output = process.stdout.readline().decode('utf-8')
print(output.strip())
这段代码的缺陷在于缺少在发送命令后添加换行符。这导致Shell无法正确解析命令，程序卡住。
解决方法是在发送命令前，将命令字符串编码后添加换行符b'\n'：
import subprocess
process = subprocess.Popen('/bin/bash', stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
while True:
cmd = input("$ ")
if cmd == "exit":
break
cmd = cmd.encode('utf-8') + b'\n'
print('cmd',cmd)
process.stdin.write(cmd)
process.stdin.flush()
output = process.stdout.readline().decode('utf-8')
print(output.strip())
通过添加b'\n'，Shell能够正确识别每个命令的结束，实现与Shell的交互式操作，达到预期效果。 Python脚本可以像在终端一样，与Shell进行多次交互，输入命令并获取输出结果。
