fix: loop_wrapper decorators were returning coro task instead of func

timoniq · Jul 12, 2024 · 5f742d4 · 5f742d4
1 parent d57de84
commit 5f742d4
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/telegrinder/tools/loop_wrapper/loop_wrapper.py b/telegrinder/tools/loop_wrapper/loop_wrapper.py
@@ -63,13 +63,11 @@ class Lifespan:
     shutdown_tasks: list[CoroutineTask[typing.Any]] = dataclasses.field(default_factory=lambda: [])
 
     def on_startup(self, task_or_func: Task) -> Task:
-        task_or_func = to_coroutine_task(task_or_func)
-        self.startup_tasks.append(task_or_func)
+        self.startup_tasks.append(to_coroutine_task(task_or_func))
         return task_or_func
 
     def on_shutdown(self, task_or_func: Task) -> Task:
-        task_or_func = to_coroutine_task(task_or_func)
-        self.shutdown_tasks.append(task_or_func)
+        self.shutdown_tasks.append(to_coroutine_task(task_or_func))
         return task_or_func
 
     def start(self, loop: asyncio.AbstractEventLoop) -> None: