PySide ждет сигнала из основного потока в рабочем потоке
-
14-12-2019 - |
Вопрос
Я решил добавить графический интерфейс в один из своих скриптов.Скрипт представляет собой простой веб-скребок.Я решил использовать рабочий поток, поскольку загрузка и анализ данных может занять некоторое время.Я решил использовать PySide, но мои знания Qt в целом весьма ограничены.
Поскольку скрипт должен ждать ввода пользователя при обнаружении капчи, я решил, что ему следует подождать, пока QLineEdit
пожары returnPressed
а затем отправьте его содержимое в рабочий поток, чтобы он мог отправить его на проверку.Это должно быть лучше, чем ожидание нажатия клавиши возврата.
Кажется, что ожидание сигнала не так просто, как я думал, и после некоторого поиска я наткнулся на несколько решений, похожих на этот.Однако передача сигналов между потоками и локальный цикл событий в рабочем потоке немного усложняют мое решение.
Повозившись с ним несколько часов, он все равно не работает.
Что должно произойти:
- Загрузите данные до тех пор, пока не появится ссылка на капчу, и введите цикл.
- Скачайте капчу и отобразите ее пользователю, запустите
QEventLoop
позвонивself.loop.exec_()
- Выход
QEventLoop
позвонивloop.quit()
в слоте рабочих потоков, который подключен черезself.line_edit.returnPressed.connect(self.worker.stop_waiting)
вmain_window
сорт - Подтвердите капчу и выполните цикл, если проверка не удалась, в противном случае повторите последний URL-адрес, который должен быть доступен для загрузки сейчас, а затем перейдите к следующему URL-адресу.
Что происходит:
...см. выше...
Выход
QEventLoop
не работает.self.loop.isRunning()
возвращаетFalse
после звонкаexit()
.self.isRunning
возвращаетTrue
, как таковой поток, похоже, не умер при странных обстоятельствах.Тем не менее поток останавливается наself.loop.exec_()
линия.Таким образом, поток зависает при выполнении цикла событий, хотя цикл событий сообщает мне, что он больше не работает.Графический интерфейс реагирует так же, как и слоты класса рабочего потока.Я вижу текст, отправляемый в рабочий поток, состояние цикла событий и сам поток, но ничего после выполнения вышеупомянутой строки.
Код немного запутан, поэтому я добавляю немного смеси псевдокода и Python, оставляя неважное:
class MainWindow(...):
# couldn't find a way to send the text with the returnPressed signal, so I
# added a helper signal, seems to work though. Doesn't work in the
# constructor, might be a PySide bug?
helper_signal = PySide.QtCore.Signal(str)
def __init__(self):
# ...setup...
self.worker = WorkerThread()
self.line_edit.returnPressed.connect(self.helper_slot)
self.helper_signal.connect(self.worker.stop_waiting)
@PySide.QtCore.Slot()
def helper_slot(self):
self.helper_signal.emit(self.line_edit.text())
class WorkerThread(PySide.QtCore.QThread):
wait_for_input = PySide.QtCore.QEventLoop()
def run(self):
# ...download stuff...
for url in list_of_stuff:
self.results.append(get(url))
@PySide.QtCore.Slot(str)
def stop_waiting(self, text):
self.solution = text
# this definitely gets executed upon pressing return
self.wait_for_input.exit()
# a wrapper for requests.get to handle captcha
def get(self, *args, **kwargs):
result = requests.get(*args, **kwargs)
while result.history: # redirect means captcha
# ...parse and extract captcha...
# ...display captcha to user via not shown signals to main thread...
# wait until stop_waiting stops this event loop and as such the user
# has entered something as a solution
self.wait_for_input.exec_()
# ...this part never get's executed, unless I remove the event
# loop...
post = { # ...whatever data necessary plus solution... }
# send the solution
result = requests.post('http://foo.foo/captcha_url'), data=post)
# no captcha was there, return result
return result
frame = MainWindow()
frame.show()
frame.worker.start()
app.exec_()
Решение
Слот выполняется внутри потока, создавшего QThread
, а не в той ветке, что QThread
элементы управления.
Вам нужно переместить QObject
в поток и подключите его слот к сигналу, и этот слот будет выполнен внутри потока:
class SignalReceiver(QtCore.QObject):
def __init__(self):
self.eventLoop = QEventLoop(self)
@PySide.QtCore.Slot(str)
def stop_waiting(self, text):
self.text = text
eventLoop.exit()
def wait_for_input(self):
eventLoop.exec()
return self.text
class MainWindow(...):
...
def __init__(self):
...
self.helper_signal.connect(self.worker.signalReceiver.stop_waiting)
class WorkerThread(PySide.QtCore.QThread):
def __init__(self):
self.signalReceiver = SignalReceiver()
# After the following call the slots will be executed in the thread
self.signalReceiver.moveToThread(self)
def get(self, *args, **kwargs):
result = requests.get(*args, **kwargs)
while result.history:
...
self.result = self.signalReceiver.wait_for_input()
Другие советы
То, что вы описываете, идеально подходит для QWaitCondition
.
Простой пример:
import sys
from PySide import QtCore, QtGui
waitCondition = QtCore.QWaitCondition()
mutex = QtCore.QMutex()
class Main(QtGui.QMainWindow):
def __init__(self, parent=None):
super(Main, self).__init__()
self.text = QtGui.QLineEdit()
self.text.returnPressed.connect(self.wakeup)
self.worker = Worker(self)
self.worker.start()
self.setCentralWidget(self.text)
def wakeup(self):
waitCondition.wakeAll()
class Worker(QtCore.QThread):
def __init__(self, parent=None):
super(Worker, self).__init__(parent)
def run(self):
print "initial stuff"
mutex.lock()
waitCondition.wait(mutex)
mutex.unlock()
print "after returnPressed"
if __name__=="__main__":
app = QtGui.QApplication(sys.argv)
m = Main()
m.show()
sys.exit(app.exec_())