Make run_qtrio invoke tractor at top level

bar_select
Tyler Goodlet 2020-07-08 15:40:35 -04:00
parent 2dec32e41f
commit 5f89a2bf08
1 changed files with 24 additions and 2 deletions

View File

@ -5,6 +5,7 @@ Run ``trio`` in guest mode on top of the Qt event loop.
All global Qt runtime settings are mostly defined here. All global Qt runtime settings are mostly defined here.
""" """
import traceback import traceback
from functools import partial
import PyQt5 # noqa import PyQt5 # noqa
from pyqtgraph import QtGui from pyqtgraph import QtGui
@ -12,6 +13,7 @@ from PyQt5 import QtCore
from PyQt5.QtCore import pyqtRemoveInputHook from PyQt5.QtCore import pyqtRemoveInputHook
import qdarkstyle import qdarkstyle
import trio import trio
import tractor
from outcome import Error from outcome import Error
@ -31,6 +33,7 @@ def run_qtrio(
func, func,
args, args,
main_widget, main_widget,
loglevel = None,
) -> None: ) -> None:
# avoids annoying message when entering debugger from qt loop # avoids annoying message when entering debugger from qt loop
pyqtRemoveInputHook() pyqtRemoveInputHook()
@ -79,10 +82,29 @@ def run_qtrio(
'main': instance, 'main': instance,
} }
# force mp since we may spawn an ib asyncio based backend
tractor._spawn.try_set_start_method('forkserver')
# setup tractor entry point args
args = (
# async_fn
func,
# args
(widgets,),
# kwargs
{'loglevel': 'info'},
# arbiter_addr
(
tractor._default_arbiter_host,
tractor._default_arbiter_port,
),
# name
'qtrio',
)
# guest mode # guest mode
trio.lowlevel.start_guest_run( trio.lowlevel.start_guest_run(
func, tractor._main,
widgets, *args,
run_sync_soon_threadsafe=run_sync_soon_threadsafe, run_sync_soon_threadsafe=run_sync_soon_threadsafe,
done_callback=done_callback, done_callback=done_callback,
) )