ITQC/pydase
24
0
Fork 0
mirror of https://github.com/tiqi-group/pydase.git synced 2025-04-22 17:10:02 +02:00
Code Issues Packages Projects Releases Wiki Activity

using explicit loop to create task even if loop is not running yet

Browse Source
This commit is contained in:
Mose Müller 2024-08-06 10:57:19 +02:00
parent 85d6229aa6
commit 15322b742d
1 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/pydase/task/task.py
View File

@ -55,7 +55,7 @@ class Task(pydase.data_service.data_service.DataService, Generic[R]):
self._bound_func = func self._bound_func = func
else: else:
self._func = func self._func = func
self._task: asyncio.Task[R] | None = None self._task: asyncio.Task[R | None] | None = None
self._status = TaskStatus.NOT_RUNNING self._status = TaskStatus.NOT_RUNNING
self._result: R | None = None self._result: R | None = None
if autostart: if autostart:
@ -69,7 +69,7 @@ class Task(pydase.data_service.data_service.DataService, Generic[R]):
if self._task: if self._task:
return return
def task_done_callback(task: asyncio.Task[R]) -> None: def task_done_callback(task: asyncio.Task[R | None]) -> None:
"""Handles tasks that have finished. """Handles tasks that have finished.
Removes a task from the tasks dictionary, calls the defined Removes a task from the tasks dictionary, calls the defined
@ -94,12 +94,17 @@ class Task(pydase.data_service.data_service.DataService, Generic[R]):
self._result = task.result() self._result = task.result()
logger.info("Starting task %s", self._func_name) async def run_task() -> R | None:
if inspect.iscoroutinefunction(self._bound_func): if inspect.iscoroutinefunction(self._bound_func):
res: Coroutine[None, None, R] = self._bound_func() logger.info("Starting task %r", self._func_name)
self._task = asyncio.create_task(res)
self._task.add_done_callback(task_done_callback)
self._status = TaskStatus.RUNNING self._status = TaskStatus.RUNNING
res: Coroutine[None, None, R] = self._bound_func()
return await res
return None
logger.info("Creating task %r", self._func_name)
self._task = self._loop.create_task(run_task())
self._task.add_done_callback(task_done_callback)
def stop(self) -> None: def stop(self) -> None:
if self._task: if self._task: