Python自定义装饰器导致Pylance类型检测错误如何解决？-Python教程-PHP中文网

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

霞舞

发布： 2024-11-18 08:21:16

原创

491人浏览过

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

如何解决使用自定义装饰器时 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


@execute
def 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]。

挖错网

一款支持文本、图片、视频纠错和AIGC检测的内容审核校对平台。

查看详情

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

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

from typing import Callable

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

登录后复制

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

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

大家都在看：

python常量是什么 Python调用API接口如何使用多线程_Python多线程调用API接口提升效率的方法 python语句和缩进的实现 python列表读取的方法如何创建Python虚拟环境_创建Python虚拟环境的多种方法与对比