python操作带参的装饰器-Python教程-PHP中文网

python操作带参的装饰器

尊渡假赌尊渡假赌尊渡假赌

发布： 2025-11-16 19:57:05

原创

425人浏览过

带参装饰器是返回装饰器的函数，通过三层函数嵌套实现：最外层接收参数，中间层接收被装饰函数，内层执行扩展逻辑。示例中@repeat(3)使say_hello调用时打印三次"Hello!"，利用functools.wraps可保留原函数名称和文档字符串。实际应用如@log_if('debug')可根据级别控制日志输出，提升代码灵活性与复用性。

python操作带参的装饰器

带参数的装饰器其实是一个返回装饰器的函数，它在 Python 中非常实用，尤其当你想根据不同的参数控制装饰器的行为时。理解它的关键在于：它多了一层函数嵌套。

1. 带参装饰器的基本结构

一个带参数的装饰器通常包含三层函数：

最外层函数：接收装饰器的参数
中间层函数：接收被装饰的函数
最内层函数：实际执行逻辑，可调用原函数并扩展行为

示例代码：

def repeat(times):
    def decorator(func):
        def wrapper(*args, **kwargs):
            for _ in range(times):
                result = func(*args, **kwargs)
            return result
        return wrapper
    return decorator
<p>@repeat(3)
def say_hello():
print("Hello!")</p><p>say_hello()

登录后复制

输出会打印三次 "Hello!"。这里 repeat(3) 返回的是真正的装饰器 decorator，然后它再装饰 say_hello 函数。

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

帮衣帮-AI服装设计

AI服装设计神器，AI生成印花、虚拟试衣、面料替换

106

查看详情

2. 使用 functools.wraps 保留原函数信息

如果不做处理，被装饰的函数名字和文档会变成 wrapper 的。为了保持原函数的元信息，推荐使用 functools.wraps：

from functools import wraps
<p>def repeat(times):
def decorator(func):
@wraps(func)
def wrapper(*args, *<em>kwargs):
for _ in range(times):
result = func(</em>args, **kwargs)
return result
return wrapper
return decorator</p><p>@repeat(2)
def greet(name):
"""打招呼函数"""
print(f"Hi, {name}")</p><p>print(greet.<strong>name</strong>)  # 输出: greet（而不是 wrapper）
print(greet.<strong>doc</strong>)   # 输出: 打招呼函数

登录后复制

3. 实际应用场景举例：日志级别控制

比如你想写一个装饰器，根据传入的级别决定是否打印日志：

def log_if(level):
    def decorator(func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            if level == 'debug':
                print(f"[DEBUG] 调用函数: {func.__name__}")
            elif level == 'info':
                print(f"[INFO] 正在执行 {func.__name__}")
            return func(*args, **kwargs)
        return wrapper
    return decorator
<p>@log_if('debug')
def process_data():
print("处理数据中...")</p><p>process_data()

登录后复制

这样你可以灵活控制不同函数的日志输出行为。

总结一下：Python 中带参装饰器本质是“返回装饰器的函数”，通过三层嵌套实现。合理使用能极大提升代码复用性和灵活性。基本上就这些。

以上就是python操作带参的装饰器的详细内容，更多请关注php中文网其它相关文章！