Python中如何模拟Shell交互避免命令执行卡死?

Python中如何模拟Shell交互避免命令执行卡死?

python模拟Shell交互,避免命令执行卡死

本文介绍如何利用Python的subprocess模块模拟Shell交互式操作,解决subprocess.Popen导致命令执行卡死的问题。

许多Python开发者希望在程序中实现与Shell类似的交互体验,例如执行一系列命令并实时获取反馈。然而,直接使用subprocess.Popen循环输入命令,常常导致程序阻塞。根本原因在于Shell命令输入处理不当。

一些初始代码尝试通过subprocess.Popen启动bash进程,然后循环读取用户输入并发送命令。然而,这些代码通常忽略一个关键步骤:在发送命令后未添加换行符。Shell解释器需要换行符来识别命令的结束。

立即学习Python免费学习笔记(深入)”;

改进后的代码在发送命令前添加了换行符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())  process.stdin.close() # 添加了关闭stdin,确保进程正常结束 process.wait() # 等待子进程结束

通过在cmd后添加b’n’,确保每个命令都被正确发送到bash进程并执行,从而实现与Shell类似的交互式操作。 这有效地解决了程序卡死的问题,允许用户在python程序中与bash进行多次交互。 此外,我们还添加了process.stdin.close()和process.wait()来确保子进程的正确关闭和等待,提高程序的健壮性。

© 版权声明
THE END
喜欢就支持一下吧
点赞9 分享