Python subprocess 模块与Shell交互的进阶技巧

在Python系统管理或自动化任务中，经常需要与Shell交互。本文探讨如何使用subprocess模块高效地与Shell进行多次交互，如同在终端中一样灵活地执行命令并获取结果。

许多开发者尝试使用subprocess.Popen模拟Shell交互，却常遇到代码卡死或无法获取预期输出的问题。例如，直接用subprocess.Popen启动Shell进程（如/bin/bash），然后循环输入命令，往往得不到正确响应。

根本原因在于，Shell命令通常以换行符(\n)作为结束标志。仅发送命令字符串，Shell无法识别命令结束，导致命令执行失败，程序陷入阻塞。

立即学习“Python免费学习笔记（深入）”；

python学习

Python 编程学习助手 skill，面向零基础用户。 当用户表达以下意图时触发： - 学习 Python 基础概念（变量、数据类型、循环、函数、类等） - 请求 Python 练习题或测验 - 请求审查或调试 Python 代码 - 请求 Python 项目实战指导（爬虫、数据分析、Web 开发等入门项目） - 泛化表述如"教我 Python""我想学编程""Python 怎么写""帮我学 Python" 此 skill

下载

让我们分析一段典型的错误代码：

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进行多次交互，输入命令并获取输出结果。