mirror of
https://gitlab.nic.cz/labs/bird.git
synced 2024-12-22 17:51:53 +00:00
Python GUI: Reworked the asyncio helper completely
The original asyncio helper in the example is busy-waiting, playing ping-pong with Qt scheduler for the whole time. Using a separate worker thread instead.
This commit is contained in:
parent
bef77416af
commit
fff44269b9
@ -1,4 +1,4 @@
|
|||||||
from PySide6.QtCore import (Qt, QEvent, QObject, Signal, Slot)
|
from PySide6.QtCore import (Qt, QEvent, QObject, QRunnable, QThreadPool, Signal, Slot)
|
||||||
from PySide6.QtWidgets import (QApplication, QLabel, QMainWindow, QPushButton, QVBoxLayout, QWidget)
|
from PySide6.QtWidgets import (QApplication, QLabel, QMainWindow, QPushButton, QVBoxLayout, QWidget)
|
||||||
|
|
||||||
from BIRD import BIRD
|
from BIRD import BIRD
|
||||||
@ -7,6 +7,24 @@ import asyncio
|
|||||||
import signal
|
import signal
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
|
# Async worker thread
|
||||||
|
class AsyncWorker(QRunnable):
|
||||||
|
@Slot()
|
||||||
|
def run(self):
|
||||||
|
self.loop = asyncio.new_event_loop()
|
||||||
|
self.loop.run_forever()
|
||||||
|
|
||||||
|
def stop(self):
|
||||||
|
asyncio.run_coroutine_threadsafe(self._stop_internal(), self.loop)
|
||||||
|
|
||||||
|
async def _stop_internal(self):
|
||||||
|
self.loop.stop()
|
||||||
|
|
||||||
|
def dispatch(coro):
|
||||||
|
asyncio.run_coroutine_threadsafe(coro, AsyncWorker.worker.loop)
|
||||||
|
|
||||||
|
if not hasattr(AsyncWorker, "worker"):
|
||||||
|
AsyncWorker.worker = AsyncWorker()
|
||||||
|
|
||||||
|
|
||||||
class MainWindow(QMainWindow):
|
class MainWindow(QMainWindow):
|
||||||
@ -30,100 +48,28 @@ class MainWindow(QMainWindow):
|
|||||||
layout.addWidget(self.text, alignment=Qt.AlignmentFlag.AlignCenter)
|
layout.addWidget(self.text, alignment=Qt.AlignmentFlag.AlignCenter)
|
||||||
|
|
||||||
async_trigger = QPushButton(text="Connect")
|
async_trigger = QPushButton(text="Connect")
|
||||||
async_trigger.clicked.connect(self.async_start)
|
async_trigger.clicked.connect(self.connect)
|
||||||
layout.addWidget(async_trigger, alignment=Qt.AlignmentFlag.AlignCenter)
|
layout.addWidget(async_trigger, alignment=Qt.AlignmentFlag.AlignCenter)
|
||||||
|
|
||||||
@Slot()
|
@Slot()
|
||||||
def async_start(self):
|
def connect(self):
|
||||||
self.start_signal.emit()
|
async def f():
|
||||||
|
async with self.bird as b:
|
||||||
|
await b.version.update()
|
||||||
|
await b.status.update()
|
||||||
|
|
||||||
async def bird_connect(self):
|
self.text.setText(f"Connected to {b.version.name} {b.version.version}")
|
||||||
async with self.bird as b:
|
|
||||||
await b.version.update()
|
|
||||||
await b.status.update()
|
|
||||||
|
|
||||||
self.text.setText(f"Connected to {b.version.name} {b.version.version}")
|
AsyncWorker.dispatch(f())
|
||||||
|
|
||||||
self.done_signal.emit()
|
|
||||||
|
|
||||||
class AsyncHelper(QObject):
|
|
||||||
|
|
||||||
class ReenterQtObject(QObject):
|
|
||||||
""" This is a QObject to which an event will be posted, allowing
|
|
||||||
asyncio to resume when the event is handled. event.fn() is
|
|
||||||
the next entry point of the asyncio event loop. """
|
|
||||||
def event(self, event):
|
|
||||||
if event.type() == QEvent.Type.User + 1:
|
|
||||||
event.fn()
|
|
||||||
return True
|
|
||||||
return False
|
|
||||||
|
|
||||||
class ReenterQtEvent(QEvent):
|
|
||||||
""" This is the QEvent that will be handled by the ReenterQtObject.
|
|
||||||
self.fn is the next entry point of the asyncio event loop. """
|
|
||||||
def __init__(self, fn):
|
|
||||||
super().__init__(QEvent.Type(QEvent.Type.User + 1))
|
|
||||||
self.fn = fn
|
|
||||||
|
|
||||||
def __init__(self, worker, entry):
|
|
||||||
super().__init__()
|
|
||||||
self.reenter_qt = self.ReenterQtObject()
|
|
||||||
self.entry = entry
|
|
||||||
self.loop = asyncio.new_event_loop()
|
|
||||||
self.done = False
|
|
||||||
|
|
||||||
self.worker = worker
|
|
||||||
if hasattr(self.worker, "start_signal") and isinstance(self.worker.start_signal, Signal):
|
|
||||||
self.worker.start_signal.connect(self.on_worker_started)
|
|
||||||
if hasattr(self.worker, "done_signal") and isinstance(self.worker.done_signal, Signal):
|
|
||||||
self.worker.done_signal.connect(self.on_worker_done)
|
|
||||||
|
|
||||||
@Slot()
|
|
||||||
def on_worker_started(self):
|
|
||||||
""" To use asyncio and Qt together, one must run the asyncio
|
|
||||||
event loop as a "guest" inside the Qt "host" event loop. """
|
|
||||||
if not self.entry:
|
|
||||||
raise Exception("No entry point for the asyncio event loop was set.")
|
|
||||||
asyncio.set_event_loop(self.loop)
|
|
||||||
self.loop.create_task(self.entry())
|
|
||||||
self.loop.call_soon(self.next_guest_run_schedule)
|
|
||||||
self.done = False # Set this explicitly as we might want to restart the guest run.
|
|
||||||
self.loop.run_forever()
|
|
||||||
|
|
||||||
@Slot()
|
|
||||||
def on_worker_done(self):
|
|
||||||
""" When all our current asyncio tasks are finished, we must end
|
|
||||||
the "guest run" lest we enter a quasi idle loop of switching
|
|
||||||
back and forth between the asyncio and Qt loops. We can
|
|
||||||
launch a new guest run by calling launch_guest_run() again. """
|
|
||||||
self.done = True
|
|
||||||
|
|
||||||
def continue_loop(self):
|
|
||||||
""" This function is called by an event posted to the Qt event
|
|
||||||
loop to continue the asyncio event loop. """
|
|
||||||
if not self.done:
|
|
||||||
self.loop.call_soon(self.next_guest_run_schedule)
|
|
||||||
self.loop.run_forever()
|
|
||||||
|
|
||||||
def next_guest_run_schedule(self):
|
|
||||||
""" This function serves to pause and re-schedule the guest
|
|
||||||
(asyncio) event loop inside the host (Qt) event loop. It is
|
|
||||||
registered in asyncio as a callback to be called at the next
|
|
||||||
iteration of the event loop. When this function runs, it
|
|
||||||
first stops the asyncio event loop, then by posting an event
|
|
||||||
on the Qt event loop, it both relinquishes to Qt's event
|
|
||||||
loop and also schedules the asyncio event loop to run again.
|
|
||||||
Upon handling this event, a function will be called that
|
|
||||||
resumes the asyncio event loop. """
|
|
||||||
self.loop.stop()
|
|
||||||
QApplication.postEvent(self.reenter_qt, self.ReenterQtEvent(self.continue_loop))
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
app = QApplication(sys.argv)
|
app = QApplication(sys.argv)
|
||||||
main_window = MainWindow()
|
threadpool = QThreadPool()
|
||||||
async_helper = AsyncHelper(main_window, main_window.bird_connect)
|
threadpool.start(AsyncWorker.worker)
|
||||||
|
|
||||||
main_window.show()
|
mainwindow = MainWindow()
|
||||||
|
mainwindow.show()
|
||||||
|
|
||||||
signal.signal(signal.SIGINT, signal.SIG_DFL)
|
signal.signal(signal.SIGINT, signal.SIG_DFL)
|
||||||
app.exec()
|
app.exec()
|
||||||
|
AsyncWorker.worker.stop()
|
||||||
|
Loading…
Reference in New Issue
Block a user