如何为可接受任意数量无参函数的Python异步执行函数实现泛型类型注解
解决任意数量不同返回类型无参函数的泛型类型注解问题
要实现支持任意数量、不同返回类型的无参函数的泛型类型标注,你需要用到Python 3.10+引入的**可变参数泛型(Variadic Generics)**特性,通过TypeVarTuple和Unpack来精准对应每个函数的返回类型,完美避免所有函数必须返回同类型的限制。
具体实现步骤
- 导入必要的类型工具:
from typing import Callable, TypeVarTuple, Unpack # 如果是Python 3.9及以下版本,需要从typing_extensions导入对应工具 # from typing_extensions import TypeVarTuple, Unpack
- 定义代表任意数量类型的元组类型变量:
Ts = TypeVarTuple('Ts')
这个Ts可以看作是一组类型的集合,用来对应每个输入函数的返回类型。
- 编写泛型函数的类型注解:
def execute_many(fs: tuple[Callable[[], Unpack[Ts]]]) -> tuple[Unpack[Ts]]: # 这里实现你的异步执行逻辑,比如用asyncio批量执行 import asyncio return asyncio.run(asyncio.gather(*(f() for f in fs)))
- 参数
fs的类型tuple[Callable[[], Unpack[Ts]]]表示:这是一个元组,每个元素都是无参函数,且每个函数的返回类型对应Ts中的一个独立类型。 - 返回值
tuple[Unpack[Ts]]表示:返回的元组类型和输入函数的返回类型一一对应,顺序完全一致。
测试示例
用你给出的函数来验证:
def get_int() -> int: return 42 def get_str() -> str: return "hello world" # 类型推断完全正确,int_val被识别为int,str_val被识别为str int_val, str_val = execute_many((get_int, get_str))
此时类型检查器(比如mypy)不会再抛出类型错误,因为它能准确推断出返回元组的每个元素类型。
为什么之前的写法会报错?
你之前尝试的tuple[Func[T], ...]是同类型可变长度元组的写法,它要求元组里的所有元素都必须是返回T类型的函数,所以当你传入返回int和str的函数时,类型检查器会认为类型不匹配。而可变参数泛型则允许我们为元组中的每个元素指定独立的类型,完美适配你的需求。
内容的提问来源于stack exchange,提问作者Sam Broster
相关产品推荐
相关产品推荐

