Python 3.9下带固定首参的强类型回调函数类型提示实现

我定义了几十个以Tested对象为第一个参数的回调函数，示例如下：

def test_value(tested: Tested, value: int) -> bool:
    return tested.value >= value

def test_value_and_name(tested: Tested, value: int, name: str) -> bool:
    return tested.value >= value and tested.name == name

def test_owned_thing(tested: Tested, thing: Thing) -> bool:
    return tested.thing == thing

现有全局测试函数tester_all，需要遍历所有Tested对象并调用传入的回调函数，但无法为其添加正确的类型提示，以实现如下参数校验效果：

# 正确调用
tester_all(test_value, 5)
tester_all(test_value, value=5)

# 类型错误：缺少参数
# tester_all(test_value)

# 类型错误：参数应为int
# tester_all(test_value, 'string value')

# 类型错误：参数过多
# tester_all(test_value, 5, 2137682)

# 正确调用
tester_all(test_value_and_name, 5, 'some name')

# 类型错误：参数不足
# tester_all(test_value_and_name, 5)

# 类型错误：第二个参数应为字符串
# tester_all(test_value_and_name, 5, 5)

# 类型错误：参数顺序错误
# tester_all(test_value_and_name, 'some name', 5)

# 正确调用
tester_all(test_owned_thing, Thing())

# 类型错误：int不是Thing类型
# tester_all(test_owned_thing, 5)

我尝试过多种方案（包括使用Protocol），但均无法完全满足需求：要么丢失回调参数的类型和数量校验，要么新增回调函数时需同步更新外部定义的类型。当前Python版本固定为3.9，可引入简单类型工具，求可行解决方案。

解决方案 #

依赖准备 #

由于Python 3.9原生不支持ParamSpec，需要先安装typing_extensions库：

pip install typing_extensions

实现代码 #

from typing_extensions import ParamSpec, TypeVar
from typing import Callable, Iterable

# 假设已定义的业务类型
class Tested:
    def __init__(self, value: int, name: str, thing: "Thing"):
        self.value = value
        self.name = name
        self.thing = thing

class Thing:
    pass

# 定义泛型参数，用于捕获回调函数的剩余参数
P = ParamSpec("P")
R = TypeVar("R")

def tester_all(
    callback: Callable[[Tested, *P], R],
    *args: P.args,
    **kwargs: P.kwargs
) -> list[R]:
    # 核心逻辑：遍历Tested对象集合并调用回调
    tested_objects: Iterable[Tested] = []  # 替换为实际的Tested对象数据源
    results = []
    for obj in tested_objects:
        results.append(callback(obj, *args, **kwargs))
    return results

方案优势 #

1. 自动适配回调参数：通过ParamSpec自动捕获回调函数除第一个Tested参数外的所有参数的类型、数量和顺序，无需手动维护额外类型定义。
2. 完整参数校验：静态类型检查工具（如mypy）会对以下场景报错：
   • 参数数量不足/过多
   • 参数类型不匹配
   • 关键字参数名称错误
   • 位置参数顺序错误
3. 扩展性强：新增回调函数时，无需修改tester_all的类型提示，直接兼容。

注意事项 #

需要使用静态类型检查工具（如mypy）来触发类型错误提示，Python运行时本身不会进行这些校验，但静态检查完全可以满足需求中的校验效果。

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