如何为通用函数meta编写类型提示以校验参数合法性？

想要为示例中的meta函数编写符合PEP484规范的类型提示，让类型检查工具能识别传入meta的目标函数参数是否存在错误。现有两个核心函数：

• meta：接收一个函数及该函数的参数，执行该函数并返回结果
• example_function：作为演示函数，用于展示应触发警告/错误的参数配置

原始代码示例 #

from typing import TypeVar, Callable

def example_function(f: str, g: int) -> None:
    assert isinstance(f, str)
    assert isinstance(g, int)


In = TypeVar('In')
Out = TypeVar('Out')
def meta(func: Callable[In, Out], *args, **kwargs) -> Out:
    a = func(*args, **kwargs)
    print(f'{func.__name__} executed')
    return a

合法调用示例 #

# 类型检查器不会报错
meta(example_function, f='', g=0)

应触发警告的错误调用示例 #

# 重复传递参数f
try:
    meta(example_function, '', f='', g=0)
except:
    ...

# 位置参数数量过多
try:
    meta(example_function, '', 0, 1)
except:
    ...

# 关键字参数数量过多
try:
    meta(example_function, f='', g=0, k='diff')
except:
    ...

# 参数类型不匹配
try:
    meta(example_function, 0, g='')
except:
    ...

问题 #

是否能在混合使用位置参数与关键字参数的场景下实现上述参数校验？若不能，仅使用位置参数或仅使用关键字参数是否可行？

回答 #

可以实现混合位置参数与关键字参数场景下的完整参数校验，需要用到Python 3.10+引入的ParamSpec和Concatenate（低版本Python可通过typing_extensions库兼容），这两个工具专门用于高阶函数的参数类型传递与校验。

修改后的类型提示代码如下：

from typing import TypeVar, Callable, ParamSpec

def example_function(f: str, g: int) -> None:
    assert isinstance(f, str)
    assert isinstance(g, int)

P = ParamSpec('P')
R = TypeVar('R')

def meta(func: Callable[P, R], *args: P.args, **kwargs: P.kwargs) -> R:
    result = func(*args, **kwargs)
    print(f'{func.__name__} executed')
    return result

代码说明

• ParamSpec('P')：定义一个参数规格变量，用来捕获目标函数func的所有参数类型（包括位置参数、关键字参数的类型和数量）
• TypeVar('R')：捕获目标函数的返回值类型
• *args: P.args 和 **kwargs: P.kwargs：明确指定meta接收的可变参数必须完全匹配目标函数func的参数规格
• 类型检查工具（如mypy、pyright）会自动对比传入的参数与目标函数的参数要求，触发对应的错误提示

校验效果

• 合法调用meta(example_function, f='', g=0)不会触发任何警告
• 所有列出的错误调用场景（重复传参、参数过多、类型不匹配等）都会被类型检查工具识别并提示错误

如果你的项目运行在Python 3.9及以下版本，只需将from typing import ParamSpec替换为from typing_extensions import ParamSpec即可实现兼容。

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