Python类型提示：能否基于字符串参数推导关键字参数的类型提示？

要实现让函数的关键字参数严格匹配第一个字符串参数指定的名称，你可以结合Python的泛型、字面量类型和TypedDict来完成，以下是具体方案：

核心实现代码 #

from typing import Generic, Literal, TypedDict, Unpack, TypeVar

# 定义绑定到字符串的类型变量，用于捕获关键字参数名称的字面量
K = TypeVar("K", bound=str)
# 可选：定义参数值的类型变量，支持任意类型的参数值
V = TypeVar("V")

# 泛型TypedDict，强制只能包含一个名为K的必填关键字参数
class RequiredKwarg(TypedDict, Generic[K, V]):
    __required_keys__: Literal[K]
    __optional_keys__: Literal[()]
    def __getitem__(self, key: K) -> V:
        ...

def magic_params(kwarg_name: Literal[K], **kwargs: Unpack[RequiredKwarg[K, V]]) -> None:
    print(kwarg_name)
    print(kwargs)

效果验证 #

# 合法调用：关键字参数与第一个参数指定的名称一致
magic_params("my_parameter", my_parameter=1)
magic_params("my_str_param", my_str_param="test")

# 非法调用：关键字参数名称不匹配，类型检查器会报错
magic_params("my_parameter", something_else=1)

关键原理说明 #

• Literal[K] 会捕获第一个参数传入的具体字符串字面量（比如"my_parameter"），让类型检查器明确关键字参数的预期名称。
• RequiredKwarg 作为泛型TypedDict，通过__required_keys__和__optional_keys__强制限定只能有一个名为K的必填参数，无可选参数。
• Unpack[RequiredKwarg[K, V]] 告诉类型检查器，**kwargs必须严格匹配这个TypedDict的结构，确保只有指定名称的关键字参数被允许。

注意：这个方案需要使用支持Python 3.11+特性的类型检查器（如Pyright、mypy 0.990及以上版本），因为Unpack用于TypedDict是3.11引入的语法。

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