Python装饰器导致函数调用缺参无警告的原因与解决方法

• 编辑器的静态代码提示（比如缺参警告）依赖函数的签名元数据和静态分析能力。当装饰器放在独立文件中时，编辑器的静态分析无法跨文件完整追踪被装饰函数的原始签名——你的wrapper函数覆盖了原函数sample_func的签名，但没有保留原函数的参数信息，导致编辑器误以为被装饰后的函数就是wrapper的签名，从而丢失了原函数的必填参数检查。
• 当装饰器和函数同文件时，编辑器可以直接分析装饰器内部逻辑，关联到原函数的签名，所以能正常触发缺参警告。
• 至于传入错误类型仍有警告，是因为编辑器能通过函数调用链追踪到原函数的类型注解，即使装饰器在外部文件，也能识别到类型不匹配的问题。

1. 使用functools.wraps保留原函数元数据 #

这是最常用的方案，它会把原函数的签名、文档字符串等元数据复制到wrapper函数上，让编辑器能正确识别原函数的参数要求：

# 在装饰器文件中导入functools
import functools

def decorator1(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        print('decorator 1')
        return func(*args, **kwargs)

    return wrapper

这里把wrapper的参数改成*args, **kwargs，确保能适配任意参数的原函数，同时通过@functools.wraps(func)保留原函数的所有元数据，编辑器就能正常触发缺参警告了。

2. 类型安全的装饰器（适配PEP 612） #

如果需要严格的类型检查（比如让mypy等工具也能识别），可以使用typing.ParamSpec和Concatenate来定义类型安全的装饰器，适用于Python 3.10+：

import functools
from typing import Callable, ParamSpec, TypeVar

P = ParamSpec('P')
R = TypeVar('R')

def decorator1(func: Callable[P, R]) -> Callable[P, R]:
    @functools.wraps(func)
    def wrapper(*args: P.args, **kwargs: P.kwargs) -> R:
        print('decorator 1')
        return func(*args, **kwargs)

    return wrapper

这种写法能让类型检查工具和编辑器完全识别原函数的参数类型和必填要求，同时保留装饰器的功能。

3. 手动复制函数签名（进阶） #

如果需要更精细的控制，可以使用inspect模块手动复制原函数的签名到wrapper上：

import functools
import inspect

def decorator1(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        print('decorator 1')
        return func(*args, **kwargs)

    # 手动复制原函数的签名
    wrapper.__signature__ = inspect.signature(func)
    return wrapper

这个方案和functools.wraps配合，能确保编辑器和类型检查工具完全识别原函数的参数签名。

内容的提问来源于stack exchange，提问作者Reyhaneh Sharifzadeh