如何为Python可变参数函数c指定Callable类型提示并强定义类型？

我编写了如下Python函数：

def c(*cargs):
    def composed(*args,**kwargs):
        res = cargs[-1](*args,**kwargs)
        for a in cargs[:-1][::-1]:
            res = a(res)
        return res
    
    return composed 

示例调用如下：

In [33]: c(sorted,list,map)(lambda x:x*2 , [6,4,5])
Out[33]: [8, 10, 12]

这个函数的逻辑是：composed函数接收最后一个函数的参数，依次从后往前调用传入的函数（除最后一个外逆序调用），最终返回第一个函数的返回值。我想知道能否为cargs中的首尾函数指定Callable[T,U]类型来强定义函数c的类型？同时希望确保所有参数都是可调用对象，这应该可以实现吧？

当然可以实现，借助Python标准库的typing模块（Python 3.10+也可使用标准库types模块，或用typing_extensions兼容旧版本），我们可以通过TypeVar和ParamSpec来精确标注这个组合函数的类型。

基础实现方案（兼容Python 3.8+） #

首先定义类型变量，用于捕获参数规格和链式调用的类型关系：

from typing import Callable, TypeVar, ParamSpec

# 捕获最后一个函数的参数规格
P = ParamSpec("P")
# 链式调用的中间类型变量
T1 = TypeVar("T1")
T2 = TypeVar("T2")
# 最终返回值类型
R = TypeVar("R")

然后为函数c添加类型标注，明确首尾函数和中间函数的类型约束：

def c(
    first: Callable[[T2], R],
    *middle: Callable[[T1], T2],
    last: Callable[P, T1]
) -> Callable[P, R]:
    def composed(*args: P.args, **kwargs: P.kwargs) -> R:
        res = last(*args, **kwargs)
        for func in reversed(middle):
            res = func(res)
        res = first(res)
        return res
    return composed

这个标注的逻辑：

• last函数接收参数P，返回类型T1
• middle中的每个函数都接收前一个函数的返回值（从T1转换为T2）
• first函数接收T2，返回最终结果R
• 最终c返回的composed函数，参数与last完全一致，返回值与first一致

用你的示例测试（调整为关键字参数传入最后一个函数）：

c(sorted, list, last=map)(lambda x:x*2, [6,4,5])

类型检查器会自动推断：

• map的参数是Callable[[int], int]和Iterable[int]，返回map[int, int]
• list接收map[int, int]，返回list[int]
• sorted接收list[int]，返回list[int]
• 最终composed函数的参数和返回值类型完全符合预期

更灵活的变长链式标注（Python 3.11+） #

如果需要支持任意长度的中间函数链式转换，可以使用TypeVarTuple实现更通用的类型约束：

from typing import Callable, ParamSpec
from typing_extensions import TypeVarTuple, Unpack

P = ParamSpec("P")
Ts = TypeVarTuple("Ts")

def c(
    *cargs: Callable[[Unpack[Ts]], Unpack[Ts]][0],
    last: Callable[P, Unpack[Ts][-1]]
) -> Callable[P, Unpack[Ts][0]]:
    def composed(*args: P.args, **kwargs: P.kwargs):
        res = last(*args, **kwargs)
        for func in reversed(cargs):
            res = func(res)
        return res
    return composed

这种方式允许中间函数有任意多的类型转换步骤，类型推断会自动匹配链式调用的输入输出关系。

保持原位置参数形式的可选方案 #

如果不想用last关键字参数，希望保持原有的位置参数传参方式，可以通过Protocol定义链式可调用的结构，但这种方式类型推断的精度会稍低，实现也更复杂，上面的关键字参数方案是更推荐的选择。

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