如何使用QPushButton(Python)停止线程

Question

我目前正在开发一个应用程序，该应用程序需要不断地将我的屏幕流发送到其中的一个窗口。但是，我希望有一种方法可以停止并使用停止和启动按钮重新启动它，我也使用layout.addWidget(QtWidgets.QPushButton('start'(，0,0(，layout.addeWidget。然而，我不知道如何将这些按钮正确地连接到非活动/活动线程，分别使用这两个按钮启动和停止线程。

这是我的代码：

from PyQt5 import QtCore, QtGui, QtWidgets 
import cv2
import numpy as np
from mss import mss

class Thread(QtCore.QThread):
changePixmap = QtCore.pyqtSignal(QtGui.QImage)
scaled_size = QtCore.QSize(1500, 1000)


def run(self):
mon = {'top': 0, 'left': 0, 'width': 1920, 'height': 1080}
with mss() as sct:
while True:
ret = True
if ret:
img = sct.grab(mon)
#    cv2.imshow('test', np.array(img))
rgbImage = cv2.cvtColor(np.array(img), cv2.COLOR_BGR2RGB)
convertToQtFormat = QtGui.QImage(rgbImage.data, rgbImage.shape[1], rgbImage.shape[0], QtGui.QImage.Format_RGB888)
p = convertToQtFormat.scaled(self.scaled_size, QtCore.Qt.KeepAspectRatio)
self.changePixmap.emit(p)
def scaled(self, scaled_size):
self.scaled_size = scaled_size

class PlayStreaming(QtWidgets.QWidget):
def __init__(self):
super(PlayStreaming,self).__init__()
self.initUI()
@QtCore.pyqtSlot(QtGui.QImage)
def setImage(self, image):
self.label.setPixmap(QtGui.QPixmap.fromImage(image))
def initUI(self):
self.setWindowTitle("Image")
# create a label
self.label = QtWidgets.QLabel(self)
th = Thread(self)
th.changePixmap.connect(self.setImage)
th.start()
lay = QtWidgets.QVBoxLayout(self)
lay.addWidget(self.label, alignment=QtCore.Qt.AlignCenter)

class UIWidget(QtWidgets.QWidget):
def __init__(self, parent=None):
super(UIWidget, self).__init__(parent)
# Initialize tab screen
self.tabs = QtWidgets.QTabWidget()
self.tab1 = QtWidgets.QWidget()   
self.tab2 = QtWidgets.QWidget()
self.tab3 = QtWidgets.QWidget()

# Add tabs
self.tabs.addTab(self.tab1,"Face")
# Create first tab
self.createGridLayout()
self.tab1.layout = QtWidgets.QVBoxLayout()
self.display = PlayStreaming()
self.tab1.layout.addWidget(self.display, stretch=1)
self.tab1.layout.addWidget(self.horizontalGroupBox)
self.tab1.setLayout(self.tab1.layout)
# Add tabs to widget        
layout = QtWidgets.QVBoxLayout(self)
layout.addWidget(self.tabs)
def createGridLayout(self):
self.horizontalGroupBox = QtWidgets.QGroupBox("Control")
self.horizontalGroupBox.setStyleSheet("QGroupBox { background-color: red}");
layout = QtWidgets.QGridLayout()
layout.addWidget(QtWidgets.QPushButton('Start'),0,0) 
layout.addWidget(QtWidgets.QPushButton('Stop'),0,1) 
self.horizontalGroupBox.setLayout(layout)

if __name__ == '__main__':
import sys
app = QtWidgets.QApplication(sys.argv)
w = UIWidget()
w.resize(1000, 800)
w.show()
sys.exit(app.exec_())

提前感谢

需要注意的是，我在以下位置使用了代码：如何在PyQT5中更改随窗口大小而变化的显示大小？以及使用OpenCV和Python-2.7作为我的代码模板的Screen Capture。

Answer 1

您需要一种方法来中断线程的run方法中的while True循环，以便停止线程。您可以保留一个布尔变量，该变量在调用Thread.stop()时将设置为False。

def run(self):
self._go = True

mon = {'top': 0, 'left': 0, 'width': 1920, 'height': 1080}
with mss() as sct:
while self._go:
ret = True
if ret:
img = sct.grab(mon)
#    cv2.imshow('test', np.array(img))
rgbImage = cv2.cvtColor(np.array(img), cv2.COLOR_BGR2RGB)
convertToQtFormat = QtGui.QImage(rgbImage.data, rgbImage.shape[1], rgbImage.shape[0], QtGui.QImage.Format_RGB888)
p = convertToQtFormat.scaled(self.scaled_size, QtCore.Qt.KeepAspectRatio)
self.changePixmap.emit(p)

def stop(self):
self._go = False

我建议在UIWidget类中构造线程，并从那里连接它的信号。

class UIWidget(QtWidgets.QWidget):
def __init__(self, parent=None):
super(UIWidget, self).__init__(parent)
# Initialize tab screen
self.tabs = QtWidgets.QTabWidget()
self.tab1 = QtWidgets.QWidget()   
self.tab2 = QtWidgets.QWidget()
self.tab3 = QtWidgets.QWidget()
# Add tabs
self.tabs.addTab(self.tab1,"Face")
self.display = PlayStreaming()
self.th = Thread(self)
self.th.changePixmap.connect(self.display.setImage)
# Create first tab
self.createGridLayout()
self.tab1.layout = QtWidgets.QVBoxLayout()
self.tab1.layout.addWidget(self.display, stretch=1)
self.tab1.layout.addWidget(self.horizontalGroupBox)
self.tab1.setLayout(self.tab1.layout)
# Add tabs to widget        
layout = QtWidgets.QVBoxLayout(self)
layout.addWidget(self.tabs)
def createGridLayout(self):
self.horizontalGroupBox = QtWidgets.QGroupBox("Control")
self.horizontalGroupBox.setStyleSheet("QGroupBox { background-color: red}");
layout = QtWidgets.QGridLayout()
layout.addWidget(QtWidgets.QPushButton('Start', clicked=self.th.start), 0, 0) 
layout.addWidget(QtWidgets.QPushButton('Stop', clicked=self.th.stop), 0, 1)
self.horizontalGroupBox.setLayout(layout)

现在，当您单击停止按钮时，当前迭代将完成，然后它将脱离循环并停止。如果您希望线程在循环中的任何位置都立即停止，那么请包含self.th.setTerminationEnabled(True)并将停止按钮信号连接到self.th.terminate。但请注意，文档警告不要使用QThread.terminate。