python - PyQt:将 partial() 用于插槽时，moveToThread 不起作用

Question

我正在构建一个运行生产者( worker )的小型 GUI 应用程序，该 GUI 按需使用输出并绘制它(使用 pyqtgraph)。

由于生产者是一个阻塞函数(需要一段时间才能运行)，我(据说)将它移到了自己的线程中。

当从生产者调用 QThread.currentThreadId() 时，它输出与主 GUI 线程相同的数字。因此，首先执行 worker，然后执行所有绘图函数调用(因为它们在同一个线程的事件队列中排队)。我该如何解决这个问题？

部分运行示例:

gui thread id 140665453623104
worker thread id: 140665453623104

这是我的完整代码:

from PyQt4 import QtCore, QtGui
from PyQt4.QtCore import pyqtSignal
import pyqtgraph as pg
import numpy as np

from functools import partial
from Queue import Queue
import math
import sys
import time


class Worker(QtCore.QObject):

    termino = pyqtSignal()

    def __init__(self, q=None, parent=None):
        super(Worker, self).__init__(parent) 
        self.q = q

    def run(self, m=30000):
        print('worker thread id: {}'.format(QtCore.QThread.currentThreadId()))
        for x in xrange(m):
            #y = math.sin(x)
            y = x**2
            time.sleep(0.001) # Weird, plotting stops if this is not present...
            self.q.put((x,y,y))
        print('Worker finished')

        self.termino.emit()

class MainWindow(QtGui.QWidget):

    def __init__(self, parent=None):
        super(MainWindow, self).__init__(parent)

        self.q = Queue()
        self.termino = False       

        self.worker = Worker(self.q)
        self.workerThread = None
        self.btn = QtGui.QPushButton('Start worker')
        self.pw = pg.PlotWidget(self)
        pi = self.pw.getPlotItem()
        pi.enableAutoRange('x', True)
        pi.enableAutoRange('y', True)
        self.ge1 = pi.plot(pen='y')
        self.xs = []
        self.ys = []

        layout = QtGui.QVBoxLayout(self)
        layout.addWidget(self.pw)
        layout.addWidget(self.btn)

        self.resize(400, 400)

    def run(self):

        self.workerThread = QtCore.QThread()
        self.worker.moveToThread(self.workerThread)
        self.worker.termino.connect(self.setTermino)
        # moveToThread doesn't work here
        self.btn.clicked.connect(partial(self.worker.run, 30000))
        # moveToThread will work here
        # assume def worker.run(self): instead of def worker.run(self, m=30000)
        # self.btn.clicked.connect(self.worker.run)
        self.btn.clicked.connect(self.graficar)

        self.workerThread.start()
        self.show()

    def setTermino(self):
        self.termino = True

    def graficar(self):

        if not self.q.empty():
            e1,e2,ciclos = self.q.get()       
            self.xs.append(ciclos)
            self.ys.append(e1)
            self.ge1.setData(y=self.ys, x=self.xs)

        if not self.termino:
            QtCore.QTimer.singleShot(1, self.graficar)

if __name__ == '__main__':

    app = QtGui.QApplication([])
    window = MainWindow()

    QtCore.QTimer.singleShot(0, window.run);
    sys.exit(app.exec_())

Answer 1

问题是 Qt 尝试根据 slot 存在于哪个线程来选择连接类型(当您调用 signal.connect(slot) 时)。因为您已使用 partial 将插槽包装在 QThread 中，您要连接的插槽驻留在 MainThread(GUI 线程)中。您可以覆盖连接类型(作为 connect() 的第二个参数，但这无济于事，因为 partial 创建的方法将始终存在于 MainThread 中，因此将连接类型设置为 Qt.QueuedConnection 没有帮助。

我能看到的解决这个问题的唯一方法是设置一个中继信号，其唯一目的是有效地将没有参数的发射信号(例如来自按钮的点击信号)更改为具有一个参数的信号(您的 m 参数)。这样你就不需要用 partial() 包装 QThread 中的插槽。

代码如下。我在主窗口类中创建了一个带有一个名为“relay”的参数(int)的信号。按钮 clicked 信号连接到主窗口类中的一个方法，这个方法有一行代码发出我创建的自定义信号。您可以扩展此方法 (relay_signal()) 以获取整数作为 m(在本例中为 500)传递给 QThread，从任何您喜欢的地方!

代码如下:

from functools import partial
from Queue import Queue
import math
import sys
import time

class Worker(QtCore.QObject):

    termino = pyqtSignal()

    def __init__(self, q=None, parent=None):
        super(Worker, self).__init__(parent) 
        self.q = q

    def run(self, m=30000):
        print('worker thread id: {}'.format(QtCore.QThread.currentThreadId()))
        for x in xrange(m):
            #y = math.sin(x)
            y = x**2
            #time.sleep(0.001) # Weird, plotting stops if this is not present...
            self.q.put((x,y,y))
        print('Worker finished')

        self.termino.emit()

class MainWindow(QtGui.QWidget):

    relay = pyqtSignal(int)

    def __init__(self, parent=None):
        super(MainWindow, self).__init__(parent)

        self.q = Queue()
        self.termino = False       

        self.worker = Worker(self.q)

        self.workerThread = None
        self.btn = QtGui.QPushButton('Start worker')
        self.pw = pg.PlotWidget(self)
        pi = self.pw.getPlotItem()
        pi.enableAutoRange('x', True)
        pi.enableAutoRange('y', True)
        self.ge1 = pi.plot(pen='y')
        self.xs = []
        self.ys = []

        layout = QtGui.QVBoxLayout(self)
        layout.addWidget(self.pw)
        layout.addWidget(self.btn)

        self.resize(400, 400)

    def run(self):
        self.workerThread = QtCore.QThread()
        self.worker.termino.connect(self.setTermino)
        self.worker.moveToThread(self.workerThread)
        # moveToThread doesn't work here
        # self.btn.clicked.connect(partial(self.worker.run, 30000))
        # moveToThread will work here
        # assume def worker.run(self): instead of def worker.run(self, m=30000)
        #self.btn.clicked.connect(self.worker.run)        
        self.relay.connect(self.worker.run)
        self.btn.clicked.connect(self.relay_signal)
        self.btn.clicked.connect(self.graficar)

        self.workerThread.start()
        self.show()

    def relay_signal(self):
        self.relay.emit(500)

    def setTermino(self):
        self.termino = True

    def graficar(self):
        if not self.q.empty():
            e1,e2,ciclos = self.q.get()       
            self.xs.append(ciclos)
            self.ys.append(e1)
            self.ge1.setData(y=self.ys, x=self.xs)

        if not self.termino or not self.q.empty():
            QtCore.QTimer.singleShot(1, self.graficar)

if __name__ == '__main__':

    app = QtGui.QApplication([])
    window = MainWindow()

    QtCore.QTimer.singleShot(0, window.run);
    sys.exit(app.exec_())

我还修改了 graficar 方法以在队列中仍有数据时继续绘图(即使在线程终止后)。我认为这可能就是您需要 QThread 中的 time.sleep 的原因，它现在也被删除了。

另外关于您在代码中关于放置 moveToThread 的位置的评论，现在它的位置是正确的。它应该在将 QThread 槽连接到信号的调用之前，其原因在这个堆栈溢出帖子中讨论:PyQt: Connecting a signal to a slot to start a background operation