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

百变鹏仔 5天前 #Python

文章标签自定义

如何解决使用自定义装饰器时 pylance 类型检测错误？

python 代码使用自定义装饰器来包装函数，使其在运行时执行特定的操作。然而，pylance 类型检查器有时无法识别装饰函数的返回类型，从而导致错误。

以下示例说明了这种错误：

def execute(func):    def inner_wrapper(*args, **kwargs) -> result[any]:        with session.begin() as session:            result = session.execute(func(*args, **kwargs))            return result    return inner_wrapper@executedef query_data_source(    start_id: int = 1, max_results_amount: int = 10) -> select:  # 忽略    stmt = (        select(            datasource.id,            datasource.name,            datasource.source_url,            datasource.author,            datasource.description,            datasource.cover_image_url,            datasource.start_date,            datasource.end_date,        )        .where(datasource.id >= start_id)        .limit(max_results_amount)        .order_by(datasource.id)    )    return stmt

在上面的示例中，query_data_source 函数被装饰了 execute 装饰器。pylance 导致类型错误，假设 query_data_source 返回 select 而不是装饰函数声明的 result[any]。

立即学习“Python免费学习笔记（深入）”；

为了解决此问题，可以修改装饰器以显式指定函数返回类型：

from typing import Callabledef execute(func) -> Callable[..., Result]:

通过指定返回类型注释，pylance 将能够正确推断装饰函数的返回类型并消除错误警告。

文章推荐

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

探秘Python在机器学习中的杰出应用案例

Python在智能交通系统中的应用案例分析

使用Python根据列表创建多个目录

将一个多项式转换为Python中的Hermite_e级数

如何在Tkinter GUI Python中添加PDF？