如何将Python多进程的输出发送到Tkinter GUI

Question

我正在尝试将Python多进程的输出显示在Tkinter图形界面上。

我可以通过图形界面将进程的输出发送到命令行，比如在命令行中运行以下这个小脚本：

from multiprocessing import Process  
import sys  

def myfunc(text):    
    print text  
    sys.stdout.flush() 

def f1():  
    p1 = Process(target = myfunc, args = ("Surprise",))  
    p1.start()  

def f2():  
    p2 = Process(target = myfunc, args = ("Fear",))  
    p2.start()  

def fp():  
    myfunc("... and an almost fanatical devotion to the Pope")  

a = Tk()  

b1 = Button(a, text="Process 1", command=f1)  
b1.grid(row=0, column=0, pady=10, padx=10, sticky=SE)  
b2 = Button(a, text="Process 2", command=f2)  
b2.grid(row=0, column=1, pady=10, padx=10, sticky=SE)  
b3 = Button(a, text="Parent", command=fp)  
b3.grid(row=0, column=2, pady=10, padx=10, sticky=SE)  

if __name__ == "__main__":  
    a.mainloop()

我也可以将输出从父进程发送到文本框，比如通过修改上面的代码，注释掉myfunc中的标准输出刷新部分。

#    sys.stdout.flush()

然后在“b3.grid...”这一行后面添加以下内容：

class STDText(Text):
    def __init__(self, parent, cnf={}, **kw):
        Text.__init__(self, parent, cnf, **kw)
    def write(self, stuff):
        self.config(state=NORMAL)
        self.insert(END, stuff)
        self.yview_pickplace("end")
        self.config(state=DISABLED)

messages = STDText(a, height=2.5, width=30, bg="light cyan", state=DISABLED)   
messages.grid(row=1, column=0, columnspan=3)
sys.stdout = messages

但是我不知道怎么把进程的输出发送到文本框。是不是我漏掉了什么简单的东西？

Answer 1

假设你用这个过程的输出去调用 myfunc，那么你可以这样写 myfunc：

def myfunc(text):    
    textwidget.insert("end", text)

这里的 textwidget 是指向文本小部件的一个句柄。

Answer 2

你可以通过使用 multiprocessing.Pipe 在不同的进程之间传递（可序列化的）数据。

import Tkinter
import multiprocessing as mp

class STDText(Tkinter.Text):
    def __init__(self, parent, cnf={}, **kw):
        Tkinter.Text.__init__(self, parent, cnf, **kw)
    def write(self, stuff):
        self.config(state=Tkinter.NORMAL)
        self.insert(Tkinter.END, stuff)
        self.yview_pickplace("end")
        self.config(state=Tkinter.DISABLED)

def myfunc(conn,text):    
    conn.send(text)
    conn.close()

class Gui(object):
    def __init__(self):
        self.a=Tkinter.Tk()  
        b1=Tkinter.Button(self.a, text="Process 1", command=self.foo)  
        b1.grid(row=0, column=0, pady=10, padx=10, sticky=Tkinter.SE)  
        b2=Tkinter.Button(self.a, text="Process 2", command=self.bar)  
        b2.grid(row=0, column=1, pady=10, padx=10, sticky=Tkinter.SE)  
        b3=Tkinter.Button(self.a, text="Parent", command=self.baz)  
        b3.grid(row=0, column=2, pady=10, padx=10, sticky=Tkinter.SE)  
        self.messages=STDText(
            self.a, height=2.5, width=30, bg="light cyan", state=Tkinter.DISABLED)   
        self.messages.grid(row=1, column=0, columnspan=3)
        self.a.mainloop()        
    def call_myfunc(self,text):
        parent_conn, child_conn=mp.Pipe()
        proc=mp.Process(target=myfunc, args=(child_conn,text,))  
        proc.start()  
        self.messages.write(parent_conn.recv())
        proc.join()       
    def foo(self):
        self.call_myfunc('Foo\n')
    def bar(self):
        self.call_myfunc('Bar\n')        
    def baz(self):
        parent_conn, child_conn=mp.Pipe()
        myfunc(child_conn,'Baz\n')
        self.messages.write(parent_conn.recv())

if __name__ == "__main__":  
    Gui()

想了解更多信息，可以查看 Doug Hellman 的教程，里面有关于 multiprocessing 的详细介绍。

Answer 3

你可以在myfunc()这个函数里，把标准输出和错误输出重定向到一个StringIO对象里，然后把写入这个StringIO里的内容发送回父级（就像unutbu建议的那样）。你可以查看我对这个问题的回答，里面有一种实现这种重定向的方法。

因为那个例子做的事情比你需要的多，所以这里有一个更符合你需求的版本：

#!/usr/bin/env python
import sys
from cStringIO import StringIO
from code import InteractiveConsole
from contextlib import contextmanager
from multiprocessing import Process, Pipe

@contextmanager
def std_redirector(stdin=sys.stdin, stdout=sys.stdin, stderr=sys.stderr):
    tmp_fds = stdin, stdout, stderr
    orig_fds = sys.stdin, sys.stdout, sys.stderr
    sys.stdin, sys.stdout, sys.stderr = tmp_fds
    yield
    sys.stdin, sys.stdout, sys.stderr = orig_fds

class Interpreter(InteractiveConsole):
    def __init__(self, locals=None):
        InteractiveConsole.__init__(self, locals=locals)
        self.output = StringIO()
        self.output = StringIO()

    def push(self, command):
        self.output.reset()
        self.output.truncate()
        with std_redirector(stdout=self.output, stderr=self.output):
            try:
                more = InteractiveConsole.push(self, command)
                result = self.output.getvalue()
            except (SyntaxError, OverflowError):
                pass
            return more, result

def myfunc(conn, commands):
    output = StringIO()
    py = Interpreter()
    results = ""

    for line in commands.split('\n'):
        if line and len(line) > 0:
            more, result = py.push(line + '\n')
            if result and len(result) > 0:
                results += result

    conn.send(results)
    conn.close()

if __name__ == '__main__':
    parent_conn, child_conn = Pipe()

    commands = """
print "[42, None, 'hello']"

def greet(name, count):
    for i in range(count):
        print "Hello, " + name + "!"

greet("Beth Cooper", 5)
fugazi
print "Still going..."
"""
    p = Process(target=myfunc, args=(child_conn, commands))
    p.start()
    print parent_conn.recv()
    p.join()

这里有一些通常的安全注意事项（也就是说，除非你能信任这些代码片段的发送者，否则不要这样做，以免他们做出愚蠢或恶意的事情）。

另外，如果你不需要解析任意混合的Python表达式和语句，这样的操作可以简化很多。如果你只需要调用一个顶层函数来生成一些输出，像这样的代码可能更合适：

def dosomething():
    print "Doing something..."

def myfunc(conn, command):
    output = StringIO()
    result = ""
    with std_redirector(stdout=output, stderr=output):
        try:
            eval(command)
            result = output.getvalue()
        except Exception, err:
            result = repr(err)

    conn.send(result)
    conn.close()

if __name__ == '__main__':
    parent_conn, child_conn = Pipe()
    command = "dosomething()"
    p = Process(target=myfunc, args=(child_conn, command))
    p.start()
    print parent_conn.recv()
    p.join()