Pyright 类型检查器中的上下文管理器装饰器回归问题分析

在Python静态类型检查工具Pyright的最新版本1.1.394中，开发者发现了一个影响上下文管理器作为装饰器使用的严重回归问题。这一问题在之前的1.1.393版本中并不存在，导致许多依赖此特性的代码在升级后出现类型检查错误。

问题现象

当开发者尝试使用@contextmanager装饰器作为方法装饰器时，Pyright会错误地报告参数缺失的错误。例如以下两种典型场景：

基础使用场景：

from contextlib import contextmanager

@contextmanager
def foo_context():
    yield

class Ben:
    @foo_context()
    def bar(self, b) -> None:
        pass

带类型注解的场景：

from collections.abc import Iterator
from contextlib import contextmanager

@contextmanager
def foo_context() -> Iterator[None]:
    yield

class Ben:
    @foo_context()
    def bar(self, b: int) -> None:
        print(b)

在这两种情况下，Pyright 1.1.394都会错误地报告"Argument missing for parameter 'b'"的错误，而实际上这些代码是完全合法的Python代码，且在运行时能够正常工作。

问题根源

经过深入分析，这个问题并非由Pyright核心逻辑的变更引起，而是源于typeshed类型存根库中contextlib.pyi文件的最近修改。typeshed是Python标准库的类型提示存根文件集合，被多个类型检查器共享使用。

具体来说，typeshed中对@contextmanager装饰器的类型定义做出了不合理的假设，特别是关于ParamSpec行为的假设，这些假设并未在Python类型系统规范中明确定义，因此不应该被依赖。这种变更破坏了上下文管理器作为装饰器使用的类型推断能力。

解决方案与修复过程

typeshed维护团队已经意识到这个问题，并迅速回滚了导致问题的变更。Pyright项目维护者也采取了以下措施：

1. 及时更新了Pyright内置的typeshed存根文件
2. 验证了回归问题已被修复
3. 加速发布流程，确保修复能够尽快到达用户手中

这一修复已经包含在Pyright 1.1.395版本中，用户升级后即可解决此问题。

对开发者的建议

对于遇到此问题的开发者，可以采取以下临时解决方案：

1. 暂时回退到Pyright 1.1.393版本
2. 等待1.1.395版本的正式发布
3. 在CI/CD流程中暂时禁用相关检查

同时，这也提醒我们在类型系统的高级特性使用中需要注意：

• 上下文管理器作为装饰器是一种强大但需要谨慎使用的模式
• 类型系统的复杂特性如ParamSpec需要各工具的一致支持
• 及时关注类型检查工具的更新公告和已知问题

Pyright团队表示将继续密切监控typeshed的变更，确保类似问题不会再次发生，为用户提供稳定可靠的类型检查体验。