我一直在使用 Python 的
subprocess
模块,我想用 python 中的 bash 进行“交互式会话”。我希望能够从 Python 读取 bash 输出/写入命令,就像在终端模拟器上一样。我想代码示例可以更好地解释它:
>>> proc = subprocess.Popen(['/bin/bash'])
>>> proc.communicate()
('user@machine:~/','')
>>> proc.communicate('ls\n')
('file1 file2 file3','')
(显然,这样不行。)这样的事情可能吗?如何实现?
非常感谢
尝试这个例子:
import subprocess
proc = subprocess.Popen(['/bin/bash'], stdin=subprocess.PIPE, stdout=subprocess.PIPE)
stdout = proc.communicate('ls -lash')
print stdout
您必须阅读更多有关 stdin、stdout 和 stderr 的内容。这看起来像是很好的讲座:http://www.doughellmann.com/PyMOTW/subprocess/
编辑:
另一个例子:
>>> process = subprocess.Popen(['/bin/bash'], shell=False, stdin=subprocess.PIPE, stdout=subprocess.PIPE)
>>> process.stdin.write('echo it works!\n')
>>> process.stdout.readline()
'it works!\n'
>>> process.stdin.write('date\n')
>>> process.stdout.readline()
'wto, 13 mar 2012, 17:25:35 CET\n'
>>>
import subprocess
import threading
p = subprocess.Popen(["bash"], stderr=subprocess.PIPE,shell=False, stdin=subprocess.PIPE, stdout=subprocess.PIPE)
exit = False
def read_stdout():
while not exit:
msg = p.stdout.readline()
print("stdout: ", msg.decode())
def read_stderro():
while not exit:
msg = p.stderr.readline()
print("stderr: ", msg.decode())
threading.Thread(target=read_stdout).start()
threading.Thread(target=read_stderro).start()
while not exit:
res = input(">")
p.stdin.write((res + '\n').encode())
p.stdin.flush()
测试结果:
>ls
>stdout: 1.py
stdout: 2.py
>ssss
stderr: bash: line 2: ssss: command not found
在我的其他答案中使用此示例:https://stackoverflow.com/a/43012138/3555925
您可以在该答案中获得更多详细信息。
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import os
import sys
import select
import termios
import tty
import pty
from subprocess import Popen
command = 'bash'
# command = 'docker run -it --rm centos /bin/bash'.split()
# save original tty setting then set it to raw mode
old_tty = termios.tcgetattr(sys.stdin)
tty.setraw(sys.stdin.fileno())
# open pseudo-terminal to interact with subprocess
master_fd, slave_fd = pty.openpty()
# use os.setsid() make it run in a new process group, or bash job control will not be enabled
p = Popen(command,
preexec_fn=os.setsid,
stdin=slave_fd,
stdout=slave_fd,
stderr=slave_fd,
universal_newlines=True)
while p.poll() is None:
r, w, e = select.select([sys.stdin, master_fd], [], [])
if sys.stdin in r:
d = os.read(sys.stdin.fileno(), 10240)
os.write(master_fd, d)
elif master_fd in r:
o = os.read(master_fd, 10240)
if o:
os.write(sys.stdout.fileno(), o)
# restore tty settings back
termios.tcsetattr(sys.stdin, termios.TCSADRAIN, old_tty)
交互式 bash 进程期望与 tty 交互。要创建伪终端,请使用 os.openpty()。这将返回一个slave_fd 文件描述符,您可以使用它来打开stdin、stdout 和stderr 文件。然后,您可以写入和读取 master_fd 以与您的进程进行交互。请注意,如果您正在进行稍微复杂的交互,您还需要使用选择模块来确保不会陷入死锁。
我写了一个模块来方便*nix shell和python之间的交互。
def execute(cmd):
if not _DEBUG_MODE:
## Use bash; the default is sh
print 'Output of command ' + cmd + ' :'
subprocess.call(cmd, shell=True, executable='/bin/bash')
print ''
else:
print 'The command is ' + cmd
print ''
在 github 上查看全部内容:https://github.com/jerryzhujian9/ez.py/blob/master/ez/easyshell.py