◎欢迎参与讨论，请在这里发表您的看法、交流您的观点。
Python自定义装饰器导致Pylance类型检测错误如何解决？
ID:21516 / 打印
如何解决使用自定义装饰器时 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]。
立即学习“Python免费学习笔记（深入）”；
为了解决此问题，可以修改装饰器以显式指定函数返回类型：
from typing import Callable def execute(func) -> Callable[..., Result]:
通过指定返回类型注释，pylance 将能够正确推断装饰函数的返回类型并消除错误警告。
