Pyright类型检查器中的泛型装饰器参数处理问题分析

在Python类型检查工具Pyright中，最近发现了一个关于泛型装饰器和参数连接(Concatenate)处理的特殊情况。这个问题揭示了类型系统在处理复杂泛型场景时的一些微妙之处。

问题背景

当开发者尝试定义一个泛型装饰器，该装饰器将一个高阶函数转换为自定义泛型类型时，Pyright在某些情况下会错误地推断参数数量。具体表现为：当使用Concatenate类型操作符向函数添加参数时，类型检查器会错误地复制该参数，导致参数数量判断失误。

问题复现

考虑以下代码示例：

from typing import Any, Concatenate, Callable

class MyGeneric[**Args]: ...

def my_decorator[
    **Args,
](dec_func: Callable[[Callable[Args, Any]], Any]) -> MyGeneric[Args]: ...

@my_decorator
def dumb_thingy[**P](func: Callable[Concatenate[int, P], Any]): ...

_v: MyGeneric[[int]] = dumb_thingy  # 错误地报类型错误
_v2: MyGeneric[[int, int]] = dumb_thingy  # 错误地通过类型检查

在这个例子中，dumb_thingy函数应该只接受一个额外的int参数，但Pyright错误地推断出需要两个int参数。

技术分析

这个问题实际上包含两个独立但相关的特殊情况：

1. 参数重复问题：类型评估过程中错误地复制了int参数，导致参数数量判断错误。正常情况下，应该只有一个int参数。

2. 类型兼容性判断问题：_v2的赋值本应触发类型错误，但Pyright错误地允许了这种赋值。这表明类型系统在处理参数连接和泛型组合时存在逻辑不一致。

值得注意的是，这个问题仅出现在用户自定义的泛型类型中。如果使用Python内置的Callable类型替代MyGeneric，类型检查行为是正确的。同样，如果不使用Concatenate而直接指定参数列表，问题也不会出现。

解决方案

Pyright团队在1.1.393版本中解决了这个问题。修复内容包括：

1. 修正了参数评估逻辑，确保Concatenate操作不会导致参数重复。
2. 改进了类型兼容性检查，确保不兼容的类型组合会正确触发错误。

对开发者的启示

这个问题提醒我们，在使用高级类型特性时需要注意：

1. 复杂的泛型组合可能会暴露类型检查器的特殊情况。
2. 不同的类型检查工具(如Pyright和mypy)可能在处理相同代码时表现不同。
3. 当类型推断结果与预期不符时，可以尝试简化类型表达式来定位问题。

对于依赖精确类型检查的项目，建议在升级类型检查工具版本后全面重新检查类型提示，以确保修复不会引入新的兼容性问题。