登录
首页 >  文章 >  python教程

Python装饰器如何处理带参数函数？

时间：2026-03-03 11:51:10 277浏览 收藏

Python装饰器处理带参数函数的核心在于灵活传递参数、支持装饰器自身配置、保留原函数元信息及严谨继承类型提示：通过*args和**kwargs实现通用参数转发；采用三层嵌套结构应对装饰器参数；借助@functools.wraps确保__name__、__doc__等属性不丢失；结合ParamSpec与TypeVar在Python 3.10+中精准保留函数签名，让装饰器既强大又健壮，成为编写可维护、可调试、类型安全的高阶函数的关键利器。

Python 装饰器如何正确处理带参数函数？

装饰器处理带参数函数的关键，在于让装饰器内部的包装函数（wrapper）能接收并传递任意参数，而不是写死参数名或数量。

用 *args 和 **kwargs 通用接收

这是最常用也最稳妥的方式。包装函数不关心原函数具体要几个参数、是位置还是关键字，统一用 *args 接收所有位置参数，**kwargs 接收所有关键字参数，再原样传给被装饰函数。

例如：

def log_call(func):
    def wrapper(*args, **kwargs):
        print(f"调用 {func.__name__}，参数：{args}, {kwargs}")
        result = func(*args, **kwargs)
        print(f"{func.__name__} 返回：{result}")
        return result
    return wrapper
<p>@log_call
def add(a, b, c=10):
return a + b + c</p><p>add(1, 2)        # 输出：调用 add，参数：(1, 2), {'c': 10}
add(5, y=3, x=1) # 也能正常工作（只要原函数支持这些参数）</p>

装饰器自身带参数时：多套一层函数

如果想让装饰器能接收配置（比如日志级别、重试次数），就需要三层嵌套：最外层接收装饰器参数，中间返回真正的装饰器，最内层是 wrapper。

关键点是中间层必须返回一个可调用对象（即装饰器），它接收被装饰函数并返回 wrapper。

例如：

def retry(max_attempts=3):
    def decorator(func):
        def wrapper(*args, **kwargs):
            for i in range(max_attempts):
                try:
                    return func(*args, **kwargs)
                except Exception as e:
                    if i == max_attempts - 1:
                        raise e
                    print(f"第 {i+1} 次尝试失败，重试中...")
        return wrapper
    return decorator
<p>@retry(max_attempts=2)
def fetch_data(url):</p><h1>模拟可能失败的操作</h1><pre class="brush:python;toolbar:false;">raise ConnectionError("网络错误")

保持原函数元信息（__name__、__doc__ 等）

直接写 wrapper 会导致 help(add) 显示的是 wrapper 的信息，而非原函数。用 @functools.wraps(func) 可自动复制元数据。

推荐写法：

from functools import wraps
<p>def log_call(func):
@wraps(func)  # 这一行很重要
def wrapper(*args, *<em>kwargs):
print(f"调用 {func.<strong>name</strong>}")
return func(</em>args, **kwargs)
return wrapper</p><p>@log_call
def greet(name):
"""向某人打招呼"""
return f"Hello, {name}"</p><p>print(greet.<strong>doc</strong>)  # 正确输出："向某人打招呼"
print(greet.<strong>name</strong>) # 正确输出："greet"</p>

处理有类型提示的函数

若原函数带类型注解（如 def calc(x: int, y: float) -> str:），wrapper 本身也应保留这些提示，否则静态检查工具（如 mypy）可能报错。使用 @wraps 后，类型提示默认不会自动继承，但大多数现代类型检查器能通过 functools.wraps 推断。更严谨的做法是手动标注 wrapper 的签名（需配合 typing 模块和 ParamSpec，Python 3.10+）：

from typing import Callable, TypeVar, ParamSpec
from functools import wraps
<p>P = ParamSpec('P')
R = TypeVar('R')</p><p>def log_call(func: Callable[P, R]) -> Callable[P, R]:
@wraps(func)
def wrapper(*args: P.args, *<em>kwargs: P.kwargs) -> R:
print(f"调用 {func.<strong>name</strong>}")
return func(</em>args, **kwargs)
return wrapper</p>

今天关于《Python装饰器如何处理带参数函数？》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！

资料下载
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>