当前位置: 首页 > news >正文

什么是装饰器?

装饰器:在不修改原函数代码的情况下,动态地扩展函数的功能。装饰器本质上是一个接收函数作为参数并返回新函数的高阶函数通过@符号简洁地应用到目标函数上,常用于添加日志、权限验证、性能测试等横切关注点功能。

一、装饰器的分类

1. 按应用对象分类

(1)​函数装饰器​(最常见)

修饰普通函数或方法:

def log(func):                    # ① 装饰器函数(接收被装饰函数)def wrapper(*args, ​**kwargs):  # ② 包装函数(实际替换原函数)print(f"调用函数: {func.__name__}")  # ③ 新增功能return func(*args, ​**kwargs)         # ④ 调用原函数return wrapper                 # ⑤ 返回包装后的函数@log                              # ⑥ 应用装饰器
def greet(name):                  # ⑦ 被装饰函数print(f"Hello, {name}")greet("Alice")                    # ⑧ 调用被装饰函数调用函数: greet
"Hello, Alice"
(2)​类装饰器

修饰整个类或通过类实现装饰逻辑:

def add_method(cls):  # 类装饰器函数,接收被装饰的类cls.new_method = lambda self: print("动态添加的方法")  # 动态添加实例方法return cls  # 必须返回修改后的类(与函数装饰器返回函数不同)@add_method  # 等价于 MyClass = add_method(MyClass)
class MyClass: passobj = MyClass()
obj.new_method()  # 输出: 动态添加的方法

2. 按参数形式分类

(1)​无参装饰器

直接接收被装饰函数:

def cache(func):          # ① 接收被装饰函数(如fibonacci)_cache = {}           # ② 创建缓存字典(闭包特性保存状态)def wrapper(*args):   # ③ 包装函数if args not in _cache:                  # ④ 检查是否已缓存_cache[args] = func(*args)          # ⑤ 未缓存则计算结果return _cache[args]                     # ⑥ 返回缓存结果return wrapper         # ⑦ 返回包装后的函数@cache
def fibonacci(n):return n if n < 2 else fibonacci(n-1) + fibonacci(n-2)
(2)​带参装饰器

外层接收额外参数,返回真正的装饰器:

import random
import time# 定义重试装饰器
def retry(max_attempts):def decorator(func):def wrapper(*args, **kwargs):for attempt in range(max_attempts):try:return func(*args, **kwargs)except Exception as e:if attempt == max_attempts - 1:  # 最后一次失败print("❌ 达到最大重试次数,放弃")raisewait_time = (attempt + 1) * 2  # 重试等待时间(指数退避)print(f"⚠️ 第 {attempt + 1} 次失败: {e}, {wait_time}秒后重试...")time.sleep(wait_time)return Nonereturn wrapperreturn decorator# 模拟不稳定的API(50%失败率)
@retry(max_attempts=3)
def get_weather(city):if random.random() < 0.5:  # 50%概率失败raise ConnectionError(f"无法获取{city}的天气数据")return f"🌤 {city}: 晴天 25°C"# 测试
print("=== 第一次调用 ===")
print(get_weather("北京"))  # 可能成功或触发重试print("\n=== 第二次调用 ===")
print(get_weather("上海"))  # 另一次独立调用# === 第一次调用 ===
# ⚠️ 第 1 次失败: 无法获取北京的天气数据, 2秒后重试...
# 🌤 北京: 晴天 25°C
# 
# === 第二次调用 ===
# ⚠️ 第 1 次失败: 无法获取上海的天气数据, 2秒后重试...
# 🌤 上海: 晴天 25°C

3. 按实现方式分类

(1)​函数实现装饰器

通过嵌套函数实现(前文示例均为函数实现)。

(2)​类实现装饰器

通过类的 __call__ 方法实现:

class CountCalls:#双下划线方法在类实例化时自动调用,用于初始化对象。def __init__(self, func): #构造方法self.func = funcself.calls = 0def __call__(self, *args, ​**kwargs): #让类的实例可以像函数一样被调用(obj())self.calls += 1print(f"第 {self.calls} 次调用")return self.func(*args, ​**kwargs) # 调用原始函数#当调用 say_hello() 时,实际上是调用了 CountCalls 实例的 __call__ 方法,从而实现了装饰器的功能​(记录调用次数并执行原函数)。@CountCalls
def say_hello():print("Hello!")#​**CountCalls(say_hello)**​ 会调用 __init__,创建一个 CountCalls 实例,并把 say_hello 函数存储到 self.func,同时初始化 self.calls = 0。#​**say_hello 不再是原来的函数**,而是 CountCalls 的一个实例,但仍然可以像函数一样调用(因为实现了 __call__)。say_hello()  # 输出: 第 1 次调用 → Hello!
say_hello()  # 输出: 第 2 次调用 → Hello!

4. 按功能场景分类

类型内置/第三方示例作用
属性管理@property将方法转为属性访问
静态方法@staticmethod定义不依赖实例的方法
类方法@classmethod定义操作类而非实例的方法
缓存@functools.lru_cache自动缓存函数结果
异步@asyncio.coroutine标记协程函数

5. 特殊类型

(1)​装饰器工厂

返回装饰器的函数(即带参装饰器的外层):

def decorator_factory(prefix):          # (1) 装饰器工厂def decorator(func):                # (2) 真正的装饰器def wrapper(*args, ​**kwargs):   # (3) 包装函数print(f"[{prefix}] 函数被调用")return func(*args, ​**kwargs)return wrapperreturn decorator                     # 返回装饰器@decorator_factory("DEBUG")             # 调用工厂,先接受参数,再返回一个装饰器
def test():passtest()  # 输出: [DEBUG] 函数被调用@decorator_factory("INFO")
def hello():print("Hello!")@decorator_factory("ERROR")
def fail():print("Failed!")hello()  # 输出: [INFO] 函数被调用 → Hello!
fail()   # 输出: [ERROR] 函数被调用 → Failed!
(2)​多层装饰器

多个装饰器叠加使用:

@log
@timer
@cache
def complex_operation():print("执行原始函数")return 42complex_operation()
# 等价于: complex_operation = log(timer(cache(complex_operation)))[cache] 检查缓存...
执行原始函数
[cache] 结果已缓存
[timer] 耗时: 0.001s
[log] 函数执行完成

总结图示

装饰器分类
├─ 按应用对象
│  ├─ 函数装饰器
│  └─ 类装饰器
├─ 按参数形式
│  ├─ 无参装饰器
│  └─ 带参装饰器
├─ 按实现方式
│  ├─ 函数实现
│  └─ 类实现
└─ 按功能场景├─ 属性管理├─ 缓存└─ 异步等

相关文章:

  • 紧急救援!Ubuntu崩溃修复大赛
  • 三次贝塞尔曲线,二次贝塞尔曲线有什么区别
  • 通达信 玄学首板 抓首版指标
  • 11.OpenCV—联合QT环境配置
  • PoE供电异常如何排查?
  • leetcode 搜索插入位置 java
  • ArcGIS中英文切换
  • 62-STM32的ISP一键下载电路
  • 华为OD机试_2025 B卷_磁盘容量排序(Python,100分)(附详细解题思路)
  • 语言模型是怎么工作的?通俗版原理解读!
  • 第六十四节:基于EasyOCR的中英文文本识别与图像标注技术研究
  • 微信小程序 按钮点击事件
  • 概率期望DP
  • Luckfox Pico Pi RV1106学习<4>:RV1106的帧率问题
  • 易语言模拟真人鼠标轨迹算法 - 非贝塞尔曲线
  • 4,QT文件操作
  • 电路笔记(元器件):PCM5102、PCM510xA I2S音频协议 DAC芯片 PCM5102模块与接线说明
  • Ubuntu中ESP32使用记录
  • 电阻篇---常见作用
  • LeetCode第 454 场周赛题解
  • 网页设计网站搭建/工业和信息化部
  • 一家专门做直销的网站/如何免费自己创建网站
  • 猪八戒做网站 纠纷/北京seo网络优化师
  • 汕头手机模板建站/semseo
  • 西安网站网页设计/互联网搜索引擎
  • 昆明专业的网站制作建设/网站优化提升排名