自定义装饰器导致Pylance类型检测错误如何解决？

解决自定义装饰器导致 pylance 类型检测错误的方案

在使用自定义装饰器时，可能会遇到 pylance 返回类型错误的警告。这通常是因为 pylance 无法识别装饰的函数返回类型与注解中指定的类型不同。

对于给定的代码片段，pylance 预期装饰器函数返回 result[any] 类型的函数，而 query_data_source 函数注解为 select 类型。为了解决此问题，可以在 execute 装饰器中明确定义返回类型为 callable[..., result]，如下所示：

from typing import Callable

def execute(func) -> Callable[..., Result]:
    ...

此修改后，pylance 将准确推断 query_data_source 函数的返回类型，并且警告将消失。它明确声明装饰器函数将返回一个接受任意数量的参数并返回 result[any] 类型函数的函数。

值得注意的是，在实际代码中，result 和 select 类型应使用正确的模块前缀，例如 from sqlalchemy.orm import result, select。

以上就是自定义装饰器导致Pylance类型检测错误如何解决？的详细内容，更多请关注硕下网其它相关文章！