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

珍惜时间，勤奋学习！今天给大家带来，正文内容主要涉及到等等，如果你正在学习文章，或者是对文章有疑问，欢迎大家关注我！后面我会持续更新相关内容的，希望都能帮到正在学习的大家！

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

使用自定义装饰器时，pylance 可能会报告类型检测错误，即使实际代码运行正常。这通常是由于 pylance 无法正确识别自定义装饰器的返回类型。

考虑以下示例代码：

from typing import callable

def execute(func) -> callable[..., result]:
    def inner_wrapper(*args, **kwargs):
        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

此代码的一个问题是 pylance 假设 query_data_source 函数返回 select 对象，而不是 result 对象。为了解决此错误，修改 execute 装饰器以明确声明其返回类型的返回类型：

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

此修改通知 pylance execute 装饰的函数返回 result 对象，从而消除类型检测错误。

今天关于《Pylance类型检测报错：如何解决自定义装饰器导致的类型错误？》的内容介绍就到此结束，如果有什么疑问或者建议，可以在公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！