如何为协程实现类似Callable的类型别名简写？

在Python 3.11及以上版本，要实现类似Callable[[参数], 返回值]的协程类型别名，得结合ParamSpec、TypeVar和Coroutine类型来定义，具体方案如下：

Python 3.11兼容写法 #

from typing import TypeVar, ParamSpec, Coroutine, TypeAlias

# 定义返回值类型变量和参数捕获器
RV = TypeVar("RV")
P = ParamSpec("P")

# 核心：将参数P和返回值RV绑定到协程类型
AsyncCallable: TypeAlias = Coroutine[None, None, RV][P]

实际使用示例 #

import asyncio

# 用类型别名标注协程函数
async def add(a: int, b: int) -> int:
    return a + b

def run_async(func: AsyncCallable[int, int, int]) -> None:
    asyncio.run(func(1, 2))

# 正常调用，无类型警告
run_async(add)

# 错误示例：返回值类型不匹配，会触发类型检查警告
async def wrong_add(a: int, b: int) -> str:
    return str(a + b)

run_async(wrong_add)  # 类型检查器会提示返回值应为int而非str

Python 3.12简化写法 #

Python 3.12对Callable做了优化，直接支持协程类型的定义，写法更简洁：

from typing import TypeVar, ParamSpec, Callable, TypeAlias

RV = TypeVar("RV")
P = ParamSpec("P")

# 直接用Callable关联参数和协程返回值
AsyncCallable: TypeAlias = Callable[P, Coroutine[None, None, RV]]

这个版本和3.11的实现功能完全一致，只是语法更贴近大家熟悉的Callable写法，可读性更好。

关键细节 #

• ParamSpec("P")用来捕获协程函数的完整参数列表
• TypeVar("RV")用来绑定协程最终的返回值类型
• Coroutine[None, None, RV]表示该协程不需要发送值、也不指定异常类型，最终返回RV类型
• 通过[P]或嵌套Callable[P, ...]的方式，把参数和返回值关联起来，确保类型检查器能准确校验参数个数、类型以及返回值类型是否匹配

内容的提问来源于stack exchange，提问作者Nikolay Prokopyev