如何为字典标注类型以匹配泛型Callable的ParamSpec.kwargs？

你可以通过Python的ParamSpec特性实现需求，让静态类型检查器自动验证传入的字典与目标函数的关键字参数匹配，同时保持apply(fn, inputs)的调用方式。

实现代码 #

from typing import Callable, ParamSpec, TypeVar

# 定义参数规格和返回值类型变量
P = ParamSpec("P")
R = TypeVar("R")

def apply(fn: Callable[P, R], inputs: P.kwargs) -> R:
    return fn(**inputs)

类型检查效果验证 #

用你提供的测试函数验证：

def f(*, a: int, b: str) -> tuple[int, str]:
    return (a, b)

good = apply(f, {"a": 1, "b": "x"})           # ✅ 通过类型检查
bad  = apply(f, {"a": "oops"})                # ❌ 类型错误：str无法赋值给int类型的参数a
extra = apply(f, {"a": 1, "b": "x", "c": 0})  # ❌ 类型错误：存在未定义的关键字参数c

原理说明 #

1. ParamSpec的作用：P = ParamSpec("P")用于捕获目标函数fn的完整参数签名，包括位置参数和关键字参数。
2. P.kwargs的特性：P.kwargs会提取出fn的关键字参数对应的类型（本质是一个匿名的TypedDict），类型检查器会自动验证传入的字典是否满足以下要求：
   • 字典的键必须完全匹配fn的关键字参数名
   • 每个键对应的值类型必须与fn的参数类型一致
   • 不能包含fn未定义的额外键
3. 适配仅关键字参数的函数：如果目标函数像f一样用*标记仅接受关键字参数，P.args会被推导为空，类型检查器不会要求处理位置参数，完全符合你只想支持kwargs的需求。

注意事项 #

• 需要Python 3.10+版本，若使用更早版本，可通过typing_extensions库导入ParamSpec
• 确保类型检查器支持该特性：Pyright原生支持，mypy需升级到0.990+版本

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