Python 装饰器怎么带参数?5 分钟能学会吗?

Python装饰器参数详解:5分钟轻松掌握高阶技巧

一、装饰器是什么?为什么需要参数化?

装饰器被称为"代码的化妆师",它能动态增强函数功能而不修改原代码。基础装饰器通过函数嵌套实现功能扩展,但当我们需要根据不同场景调整装饰行为时,就需要引入参数化装饰器。

1. 参数化的必要性

设想这些场景:
日志装饰器需要记录不同级别的信息
缓存装饰器需要设置过期时间
重试装饰器需要指定尝试次数
通过参数传递,我们可以让同一个装饰器适配多种使用场景

二、三层嵌套结构解析

带参数的装饰器采用三层嵌套结构,这是掌握参数化装饰器的核心要点。

1. 基础模板代码

def param_decorator(参数):
    def decorator(func):
        def wrapper(args, kwargs):
             使用参数的装饰逻辑
            return func(args, kwargs)
        return wrapper
    return decorator

2. 参数传递流程

  1. 最外层接收装饰器参数
  2. 中间层接收被装饰函数
  3. 最内层执行增强逻辑

三、实战案例:带参数的装饰器应用

1. 重复执行函数

def repeat_execution(n_times):
    def decorator(func):
        def wrapper(args, kwargs):
            for _ in range(n_times):
                result = func(args, kwargs)
            return result
        return wrapper
    return decorator

@repeat_execution(3)
def greet(name):
    print(f"Hello {name}")

调用greet("Alice")将会打印3次问候语。

2. 智能重试机制

def retry_operation(max_attempts=3):
    def decorator(func):
        def wrapper(args, kwargs):
            for attempt in range(1, max_attempts+1):
                try:
                    return func(args, kwargs)
                except Exception as e:
                    print(f"Attempt {attempt} failed: {str(e)}")
            raise Exception("All attempts failed")
        return wrapper
    return decorator

四、5分钟掌握技巧总结

  • 牢记三层结构:参数接收层 → 函数接收层 → 执行层
  • 通过闭包特性保存参数:每个嵌套层都能访问上层参数
  • 使用@装饰器(参数)语法糖调用
  • 保持装饰器功能单一:每个参数对应一个明确的功能调整

通过本文的案例解析,相信你已经掌握了参数化装饰器的实现精髓。现在可以尝试为你的项目添加支持超时设置、权限分级等功能的参数化装饰器了。5分钟的学习将给你的编程效率带来质的提升!