From 861e89f37a7a181c911490e525cd939d7adab082 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mose=20M=C3=BCller?= Date: Mon, 5 Aug 2024 16:49:16 +0200 Subject: [PATCH] task: using functools to get correct func name --- src/pydase/task/decorator.py | 2 ++ src/pydase/task/task.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/src/pydase/task/decorator.py b/src/pydase/task/decorator.py index ee704a9..224e119 100644 --- a/src/pydase/task/decorator.py +++ b/src/pydase/task/decorator.py @@ -1,4 +1,5 @@ import asyncio +import functools import logging from collections.abc import Callable, Coroutine from typing import Any, Concatenate, ParamSpec, TypeVar @@ -17,6 +18,7 @@ def task( def decorator( func: Callable[Concatenate[Any, P], Coroutine[None, None, R]], ) -> Task[P, R]: + @functools.wraps(func) async def wrapper(self: Any, *args: P.args, **kwargs: P.kwargs) -> R | None: try: return await func(self, *args, **kwargs) diff --git a/src/pydase/task/task.py b/src/pydase/task/task.py index 7b9a628..781988a 100644 --- a/src/pydase/task/task.py +++ b/src/pydase/task/task.py @@ -61,7 +61,7 @@ class Task(pydase.DataService, Generic[P, R]): self._result = task.result() - logger.info("Starting task") + logger.info("Starting task %s", self._func.__name__) if inspect.iscoroutinefunction(self._bound_func): res: Coroutine[None, None, R] = self._bound_func(*args, **kwargs) self._task = asyncio.create_task(res)