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

task: removes check if function is bound (not used)

Browse Source
This commit is contained in:
Mose Müller 2025-01-17 19:59:51 +01:00
parent 303de82318
commit e25511768d
1 changed files with 40 additions and 46 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,5 +1,4 @@
import asyncio import asyncio
import inspect
import logging import logging
import os import os
import signal import signal
@ -163,58 +162,53 @@ class Task(pydase.data_service.data_service.DataService, Generic[R]):
self._result = task.result() self._result = task.result()
async def run_task() -> R | None: async def run_task() -> R | None:
if inspect.iscoroutinefunction(self._func): logger.info("Starting task %r", self._func_name)
logger.info("Starting task %r", self._func_name) self._status = TaskStatus.RUNNING
self._status = TaskStatus.RUNNING attempts = 0
attempts = 0 start_time = None
start_time = None
if self._timeout_start_sec: if self._timeout_start_sec:
# Wait for timeout_start_sec seconds # Wait for timeout_start_sec seconds
await asyncio.sleep(self._timeout_start_sec) await asyncio.sleep(self._timeout_start_sec)
while True: while True:
res: Coroutine[None, None, R | None] = self._func() res: Coroutine[None, None, R | None] = self._func()
try: try:
await res await res
except asyncio.CancelledError: except asyncio.CancelledError:
logger.info("Task '%s' was cancelled", self._func_name) logger.info("Task '%s' was cancelled", self._func_name)
raise raise
except Exception as e: except Exception as e:
if start_time is None: if start_time is None:
start_time = time()
attempts += 1
logger.exception(
"Task %r encountered an exception: %r [attempt %s since %s].",
self._func.__name__,
e,
attempts,
datetime.fromtimestamp(start_time),
)
if not self._restart_on_failure:
if self._exit_on_failure:
raise e
break
if self._start_limit_interval_sec is not None:
if (time() - start_time) > self._start_limit_interval_sec:
# reset attempts if start_limit_interval_sec is exceeded
start_time = time() start_time = time()
attempts = 1
attempts += 1 elif attempts > self._start_limit_burst:
logger.exception( logger.error(
"Task %r encountered an exception: %r [attempt %s since %s].", "Task %r exceeded restart burst limit. Stopping.",
self._func.__name__, self._func.__name__,
e, )
attempts,
datetime.fromtimestamp(start_time),
)
if not self._restart_on_failure:
if self._exit_on_failure: if self._exit_on_failure:
raise e raise e
break break
if self._start_limit_interval_sec is not None: await asyncio.sleep(self._restart_sec)
if (time() - start_time) > self._start_limit_interval_sec:
# reset attempts if start_limit_interval_sec is exceeded
start_time = time()
attempts = 1
elif attempts > self._start_limit_burst:
logger.error(
"Task %r exceeded restart burst limit. Stopping.",
self._func.__name__,
)
if self._exit_on_failure:
raise e
break
await asyncio.sleep(self._restart_sec)
return None
logger.warning(
"Cannot start task %r. Function has not been bound yet", self._func_name
)
return None return None
logger.info("Creating task %r", self._func_name) logger.info("Creating task %r", self._func_name)