Improve the trio typing

Author: pgjones

src/hypercorn/trio/__init__.py

@@ -17,7 +17,7 @@ async def serve(
     config: Config,
     *,
     shutdown_trigger: Callable[..., Awaitable[None]] | None = None,
-    task_status: trio.TaskStatus = trio.TASK_STATUS_IGNORED,
+    task_status: trio.TaskStatus[list[str]] = trio.TASK_STATUS_IGNORED,
     mode: Literal["asgi", "wsgi"] | None = None,
 ) -> None:
     """Serve an ASGI framework app given the config.

src/hypercorn/trio/run.py

@@ -33,7 +33,7 @@ async def worker_serve(
     *,
     sockets: Sockets | None = None,
     shutdown_trigger: Callable[..., Awaitable[None]] | None = None,
-    task_status: trio.TaskStatus = trio.TASK_STATUS_IGNORED,
+    task_status: trio.TaskStatus[list[str]] = trio.TASK_STATUS_IGNORED,
 ) -> None:
     config.set_statsd_logger_class(StatsdLogger)
 
@@ -86,7 +86,7 @@ async def worker_serve(
                 bind = repr_socket_addr(sock.family, sock.getsockname())
                 await config.log.info(f"Running on https://{bind} (QUIC) (CTRL + C to quit)")
 
-            task_status.started(binds)  # type: ignore[call-overload]
+            task_status.started(binds)
             try:
                 async with trio.open_nursery(strict_exception_groups=True) as nursery:
                     if shutdown_trigger is not None:

src/hypercorn/trio/worker_context.py

@@ -11,10 +11,10 @@
 def _cancel_wrapper(func: Callable[[], Awaitable[None]]) -> Callable[[], Awaitable[None]]:
     @wraps(func)
     async def wrapper(
-        task_status: trio.TaskStatus = trio.TASK_STATUS_IGNORED,
+        task_status: trio.TaskStatus[trio.CancelScope] = trio.TASK_STATUS_IGNORED,
     ) -> None:
         cancel_scope = trio.CancelScope()
-        task_status.started(cancel_scope)  # type: ignore[call-overload]
+        task_status.started(cancel_scope)
         with cancel_scope:
             await func()