4

我正在尝试通过 stdout 和 PIPE 与实时 subprocess.Popen非常相似的东西

但是,我也想将输入发送到正在运行的进程。

如果我在单独的线程中使用

process = subprocess.Popen(cmd,stdout=subprocess.PIPE, stderr=subprocess.STDOUT)

我可以使用终端发送输入。

我将如何从其他来源发送输入,例如不在线程中的单独函数?

我无法使用Popen.communicate,因为运行过程永远不会完成,因为我正在尝试与程序进行实时交互。

提前致谢。

这是我的完整代码,我希望在单击发送按钮时将输入发送到子进程。

from Tkinter import *`
from ttk import *`
import subprocess
from threading import Thread

class Example(Frame):

    def __init__(self, parent):
       Frame.__init__(self, parent)   

        self.parent = parent
        self.initUI()


    def initUI(self):    

        self.parent.title("Test Client")
        self.style = Style()
        self.style.theme_use("default")
        self.pack(fill=BOTH, expand=1)

        #Label, doesnt change
        lbl = Label(self, text="Client:")
        lbl.grid(row=0, column=1, sticky=W )

        #when output from client is shown
        global display
        display = Text(self,width=50,height=20)
        display.grid(row=1, column=1, sticky=E+W+N+S)

        #where user input is taken
        global prompt
        prompt = Entry(self,width=50)
        prompt.grid(row=3, column=1, sticky=E+W+N+S)

        #Button that will send input to client
        send = Button(self,text="Send",command=self.send)
        send.grid(row=3, column=2, sticky=N)
        get = Button(self,text="Get",command=self.get)
        get.grid(row=2, column=2, sticky=S)

    def get(self):
        print foo

    def send(self):
        sent = prompt.get()


def MyThread():
     global sent
     sent = 2
     cmd = ['nc', '-l', '-p', '50000']

     process = subprocess.Popen(cmd,stdout=subprocess.PIPE, stderr=subprocess.STDOUT)

    while True:
        out = process.stdout.read(1)
        if out == '' and process.poll() != None:
            break
        if out != '':
            display.insert(INSERT, out)
            sys.stdout.write(out)
            sys.stdout.flush()

def main():
    root = Tk()
    root.geometry("500x410+300+300")
    app = Example(root)

    thread = Thread(target = MyThread, args=())
    thread.start()

    root.mainloop()

if __name__ == '__main__':
    main()  
4

3 回答 3

5

标准库中的select模块是为这些情况而设计的:

process = subprocess.Popen(cmd,stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE)

while True:
   reads,writes,excs = select.select([process.stdout, process.stderr], [process.stdin], [], 1)
   for r in reads:
       out = r.read(1)
       display.insert(INSERT, out)
       sys.stdout.write(out)
       sys.stdout.flush()
   for w in writes:
       w.write('a')

您可以传递文件对象或文件描述符列表,select()这些文件将返回那些已准备好读取/写入数据或直到可选超时的文件。

select 模块适用于 Windows 和类 Unix 系统(Linux、Mac 等)。

于 2013-04-18T19:25:25.040 回答
5

首先,你显然需要添加stdin=subprocess.PIPEPopen构造函数中,然后你可以process.stdin.write像你一样process.stdout.read

但是很明显,就像read如果还没有数据write可以阻塞一样,如果孩子没有阅读也可以阻塞。

甚至超出了显而易见的范围,实际上很难获得正确的细节,Popen以便在交互式程序中双向使用 PIPE 而不会阻塞任何地方。如果您真的想这样做,请查看源代码communicate以了解它是如何工作的。(在 3.2 之前有已知的错误,所以如果你在 2.x 上,你可能需要做一些反向移植。)你必须自己实现代码,如果你希望它是跨平台的,你将不得不在communicate内部完成整个混乱(为管道生成读取器和写入器线程等),当然还要添加另一个线程以在每次尝试通信时不阻塞主线程,以及某种消息机制孩子准备好时的主线程,依此类推。

或者,您可以查看 PyPI 上的各种“异步子流程”项目。我今天所知道的最简单的是async_subprocess,它基本上只是为您提供了一个communicate您可以在没有阻塞的情况下使用的。

或者,如果您可以使用twisted(或可能使用其他基于事件的网络框架),则有围绕子流程的包装器插入其事件循环。(如果您可以等待 Python 3.4,或者使用tulip3.3 上的进行中的工作,那么有人围绕它构建了类似的东西tulip,可能会使其进入 3.4。)twisted甚至知道如何插入Tkinter,因此您不必手动处理两个独立的事件循环并在它们之间进行通信。

如果您只关心现代 POSIX 系统(而不是 Windows),您可以通过将管道置于非阻塞模式并像处理套接字一样编写代码来使其更简单。

但最简单的解决方案可能是使用类似的东西pexpect而不是尝试手动编写脚本。(正如 JF Sebastian 指出的那样,pexpect它仅适用于 Unix,但您可以pexpect为 Unix 和winpexpectWindows 使用包装器。)

于 2013-04-18T19:07:42.243 回答
0

在您的情况下,对代码更改最少的简单可移植解决方案可能是创建一个编写器线程,该线程从队列中获取项目并将它们写入进程的标准输入,然后在按下按钮时将值放入队列:

from subprocess import Popen, PIPE, STDOUT
from Queue import Queue

class Example(Frame):
    def __init__(self, parent, queue):
       # ...
       self.queue = queue
    # ...
    def send(self): # should be call on the button press
        self.queue.put(prompt.get())

def writer(input_queue, output): 
    for item in iter(input_queue.get, None): # get items until None found
        output.write(item)
    output.close()

def MyThread(queue):
    # ...
    #NOTE: you must set `stdin=PIPE` if you wan't to write to process.stdin
    process = Popen(cmd, stdin=PIPE, stdout=PIPE, stderr=STDOUT)
    Thread(target=writer, args=[queue, process.stdin]).start()
    # ...

def main():
    # ...
    queue = Queue()
    app = Example(root, queue)
    Thread(target=MyThread, args=[queue]).start()
    # ...
    root.mainloop()
    queue.put(None) # no more input for the subprocess
于 2013-04-18T20:30:40.860 回答