从python与bash交互

时间:2012-03-12 19:43:25

标签: python bash subprocess

我一直在玩Python的subprocess模块,我想用python的bash做一个“交互式会话”。我希望能够像在终端仿真器上一样从Python读取bash输出/写入命令。我猜一个代码示例更好地解释了它:

>>> proc = subprocess.Popen(['/bin/bash'])
>>> proc.communicate()
('user@machine:~/','')
>>> proc.communicate('ls\n')
('file1 file2 file3','')

(很明显,它不会那样工作。)这样的事情是可能的,怎么样?

非常感谢

4 个答案:

答案 0 :(得分:12)

试试这个例子:

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'
>>> 

答案 1 :(得分:3)

交互式bash进程希望与tty进行交互。要创建伪终端,请使用os.openpty()。这将返回一个slave_fd文件描述符,您可以使用它来打开stdin,stdout和stderr的文件。然后,您可以写入master_fd并从中读取以与您的流程进行交互。请注意,如果您正在进行稍微复杂的交互,那么您还需要使用select模块来确保不会出现死锁。

答案 2 :(得分:3)

我写了一个模块来促进* 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

答案 3 :(得分:1)

在我的另一个答案中使用此示例: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)