fix: Correct the type annotation of create_timer() (#61)

Backported-from: main
Backported-to: 1.x

changes/61.fix.md

@@ -0,0 +1 @@
+Correct the type annotation of the callback argument in `create_timer()`

src/aiotools/timer.py

@@ -11,6 +11,7 @@
 
 from .compat import get_running_loop
 from .taskgroup import TaskGroup
+from .types import CoroutineLike
 
 __all__ = (
     'create_timer',
@@ -28,9 +29,12 @@ class TimerDelayPolicy(enum.Enum):
     CANCEL = 1
 
 
-def create_timer(cb: Callable[[float], None], interval: float,
-                 delay_policy: TimerDelayPolicy = TimerDelayPolicy.DEFAULT,
-                 loop: Optional[asyncio.AbstractEventLoop] = None) -> asyncio.Task:
+def create_timer(
+    cb: Callable[[float], CoroutineLike[None]],
+    interval: float,
+    delay_policy: TimerDelayPolicy = TimerDelayPolicy.DEFAULT,
+    loop: Optional[asyncio.AbstractEventLoop] = None,
+) -> asyncio.Task:
     """
     Schedule a timer with the given callable and the interval in seconds.
     The interval value is also passed to the callable.
@@ -59,7 +63,7 @@ async def _timer():
                         fired_tasks.clear()
                     else:
                         fired_tasks[:] = [t for t in fired_tasks if not t.done()]
-                    t = task_group.create_task(cb(interval=interval))
+                    t = task_group.create_task(cb(interval))
                     fired_tasks.append(t)
                     await asyncio.sleep(interval)
         except asyncio.CancelledError:

src/aiotools/types.py

@@ -0,0 +1,21 @@
+from __future__ import annotations
+
+import sys
+from typing import (
+    Any,
+    Awaitable,
+    Coroutine,
+    Generator,
+    TypeVar,
+)
+from typing_extensions import TypeAlias
+
+_T = TypeVar("_T")
+
+# taken from the typeshed
+if sys.version_info >= (3, 12):
+    AwaitableLike: TypeAlias = Awaitable[_T]  # noqa: Y047
+    CoroutineLike: TypeAlias = Coroutine[Any, Any, _T]  # noqa: Y047
+else:
+    AwaitableLike: TypeAlias = Generator[Any, None, _T] | Awaitable[_T]
+    CoroutineLike: TypeAlias = Generator[Any, None, _T] | Coroutine[Any, Any, _T]