Python subprocess模块与Shell交互：如何避免代码卡死并正确获取输出？-Python教程-PHP中文网

Python subprocess模块与Shell交互：如何避免代码卡死并正确获取输出？

心靈之曲

发布： 2025-03-11 10:12:38

原创

422人浏览过

python subprocess模块与shell交互：如何避免代码卡死并正确获取输出？

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

以上就是Python subprocess模块与Shell交互：如何避免代码卡死并正确获取输出？的详细内容，更多请关注php中文网其它相关文章！