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

adds autostart flag to task

Browse Source
This commit is contained in:
Mose Müller 2024-08-06 09:34:43 +02:00
parent e4a3cf341f
commit 1e02f12794
2 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/pydase/task/decorator.py
View File

@ -26,6 +26,6 @@ def task(
logger.info("Task '%s' was cancelled", func.__name__) logger.info("Task '%s' was cancelled", func.__name__)
return None return None
return Task(wrapper) return Task(wrapper, autostart=autostart)
return decorator return decorator

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

@ -41,6 +41,8 @@ class Task(pydase.DataService, Generic[P, R]):
self, self,
func: Callable[Concatenate[Any, P], Coroutine[None, None, R | None]] func: Callable[Concatenate[Any, P], Coroutine[None, None, R | None]]
| Callable[P, Coroutine[None, None, R | None]], | Callable[P, Coroutine[None, None, R | None]],
*,
autostart: bool = False,
) -> None: ) -> None:
super().__init__() super().__init__()
self._func_name = func.__name__ self._func_name = func.__name__
@ -53,6 +55,8 @@ class Task(pydase.DataService, Generic[P, R]):
self._task: asyncio.Task[R] | None = None self._task: asyncio.Task[R] | None = None
self._status = TaskStatus.NOT_RUNNING self._status = TaskStatus.NOT_RUNNING
self._result: R | None = None self._result: R | None = None
if autostart:
self.start()
@property @property
def status(self) -> TaskStatus: def status(self) -> TaskStatus: