各位自习室的同学,大家好~
在 Python 后端开发中,装饰器是一个“看似抽象、实则高频”的知识点。很多入门同学能看懂带 @ 符号的代码,却搞不清背后的逻辑,遇到复杂场景更是无从下手。
今天咱们就扎根「基础自习室」,从核心原理出发,拆解装饰器的底层逻辑,再结合 3 个后端常用场景实战,帮大家真正吃透这个实用技巧,夯实函数进阶的地基。
一、前置知识:搞懂这2点,再学装饰器不迷路
装饰器的本质是“函数的函数”,依赖 Python 两个核心特性,先把这两个知识点吃透,后续原理就会迎刃而解。
1. 函数是“一等公民”
在 Python 中,函数和整数、字符串一样,具备以下能力:
可以作为参数传递给另一个函数;
可以作为另一个函数的返回值;
可以赋值给变量,也可以存储在容器中。
示例代码:
def add(a, b): return a + b# 函数赋值给变量func = addprint(func(2, 3)) # 输出:5# 函数作为参数传递def wrapper(func, x, y): return func(x, y)print(wrapper(add, 2, 3)) # 输出:5
2. 闭包:函数嵌套的“数据隔离仓”
闭包是指“嵌套函数中,内层函数引用了外层函数的变量,且外层函数返回内层函数”。它能让内层函数在脱离外层函数作用域后,仍保留对外部变量的引用,这是装饰器实现“增强功能不修改原函数”的核心。
示例代码:
def outer(msg): # 外层函数定义变量 def inner(): # 内层函数引用外层变量 print(f"收到消息:{msg}") # 返回内层函数(不执行) return inner# 接收闭包,此时msg被“封存”在inner中func = outer("Hello 自习室")func() # 输出:收到消息:Hello 自习室
二、装饰器核心原理:给函数“穿衣服”不换“内核”
装饰器的核心作用是:在不修改原函数代码、不改变函数调用方式的前提下,为函数添加额外功能(如日志、计时、权限校验等)。
我们从“手动实现”到“语法糖简化”,一步步拆解原理。
1. 基础版装饰器:无参数场景
假设我们要给函数添加“执行前打印提示”的功能,用装饰器实现如下:
# 1. 定义装饰器函数(外层函数接收原函数)def log_decorator(func): # 2. 定义嵌套函数(内层函数实现增强逻辑+调用原函数) def wrapper(): print(f"【日志】函数 {func.__name__} 开始执行") result = func() # 调用原函数 print(f"【日志】函数 {func.__name__} 执行结束") return result # 返回原函数结果 # 3. 返回嵌套函数 return wrapper# 4. 定义原函数def hello(): print("Hello Python!")# 5. 给原函数“套上”装饰器(不修改原函数代码)hello = log_decorator(hello)# 6. 调用方式不变hello()# 输出:# 【日志】函数 hello 开始执行# Hello Python!# 【日志】函数 hello 结束执行
2. 语法糖简化:@符号的本质
上面第 5 步“给函数套装饰器”的操作,Python 提供了语法糖 @ 简化,直接放在原函数上方即可,效果完全一致:
def log_decorator(func): def wrapper(): print(f"【日志】函数 {func.__name__} 开始执行") result = func() print(f"【日志】函数 {func.__name__} 执行结束") return result return wrapper# 语法糖简化,等价于 hello = log_decorator(hello)@log_decoratordef hello(): print("Hello Python!")hello() # 调用方式不变
⚠️ 注意:@decorator 本质是“函数赋值”的语法糖,不要误以为它是特殊语法,底层逻辑和手动赋值完全相同。
3. 进阶:支持带参数的原函数
上面的装饰器仅支持无参函数,实际开发中函数往往带参数。此时可通过 *args(接收位置参数)和 **kwargs(接收关键字参数)让装饰器兼容任意参数:
def log_decorator(func): # 用*args和**kwargs接收任意参数 def wrapper(*args, **kwargs): print(f"【日志】函数 {func.__name__} 开始执行,参数:{args}, {kwargs}") result = func(*args, **kwargs) # 传递参数给原函数 print(f"【日志】函数 {func.__name__} 执行结束") return result return wrapper@log_decoratordef add(a, b): return a + b@log_decoratordef user_info(name, age=18): print(f"姓名:{name},年龄:{age}")print(add(2, 3)) # 支持位置参数user_info("张三", age=20) # 支持关键字参数
4. 避坑点:修复原函数元信息
直接使用上面的装饰器会有一个问题:原函数的元信息(如函数名、文档字符串)会被嵌套函数 wrapper 覆盖:
print(add.__name__) # 输出:wrapper(预期是add)
解决方案:使用 functools.wraps 装饰 wrapper,保留原函数元信息:
from functools import wrapsdef log_decorator(func): @wraps(func) # 保留原函数元信息 def wrapper(*args, **kwargs): print(f"【日志】函数 {func.__name__} 开始执行") result = func(*args, **kwargs) print(f"【日志】函数 {func.__name__} 执行结束") return result return wrapper@log_decoratordef add(a, b): return a + bprint(add.__name__) # 输出:add(正常)
这是后端开发中写装饰器的“标准操作”,避免后续调试时因元信息错误踩坑。
三、3个后端实用场景:装饰器的实际价值
掌握原理后,我们结合后端开发高频场景,看看装饰器如何简化代码、提升可维护性。
场景1:接口日志记录(最常用)
后端接口需要记录请求参数、响应结果、执行时间,用装饰器可统一实现,无需在每个接口中重复写日志代码:
from functools import wrapsimport timeimport logging# 配置日志logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")def api_logger(func): @wraps(func) def wrapper(*args, **kwargs): start_time = time.time() # 记录请求信息 logging.info(f"接口 {func.__name__} 收到请求,参数:{args}, {kwargs}") try: result = func(*args, **kwargs) # 记录响应结果 logging.info(f"接口 {func.__name__} 响应成功,结果:{result}") return result except Exception as e: # 记录异常信息 logging.error(f"接口 {func.__name__} 执行失败,异常:{str(e)}") raise e # 抛出异常,不影响原有错误处理 finally: # 记录执行时间 end_time = time.time() logging.info(f"接口 {func.__name__} 执行耗时:{end_time - start_time:.4f}s") return wrapper# 接口示例@api_loggerdef get_user(user_id): # 模拟数据库查询 time.sleep(0.1) return {"user_id": user_id, "name": "张三", "status": "正常"}@api_loggerdef create_order(user_id, goods_id, amount): if amount <= 0: raise ValueError("金额不能小于等于0") return {"order_id": "OD123456", "status": "创建成功"}# 调用接口get_user(1001)create_order(1001, "GOODS001", 99.9)
场景2:函数执行计时(性能优化)
开发中需要定位“慢函数”,用装饰器可快速统计函数执行时间,按需扩展到多函数:
from functools import wrapsimport timedef timer(func): @wraps(func) def wrapper(*args, **kwargs): start = time.perf_counter() # 高精度计时 result = func(*args, **kwargs) end = time.perf_counter() # 打印计时结果,保留4位小数 print(f"【计时】函数 {func.__name__} 执行耗时:{end - start:.4f}s") return result return wrapper# 模拟耗时函数@timerdef heavy_task(): # 模拟复杂计算 sum = 0 for i in range(10**6): sum += i return sumheavy_task() # 输出:【计时】函数 heavy_task 执行耗时:0.0452s
场景3:接口权限校验(身份验证)
后端接口需要验证用户是否登录、是否有权限访问,用装饰器统一拦截,避免重复校验逻辑:
from functools import wraps# 模拟用户会话(实际开发中从Redis/Session获取)current_user = {"id": 1001, "role": "admin", "is_login": True}def permission_required(required_role="user"): # 带参数的装饰器:外层接收权限参数,中层接收原函数 def decorator(func): @wraps(func) def wrapper(*args, **kwargs): # 1. 验证是否登录 if not current_user["is_login"]: raise PermissionError("请先登录") # 2. 验证权限 if current_user["role"] != required_role: raise PermissionError("无权限访问,需要{}权限".format(required_role)) # 校验通过,执行原函数 return func(*args, **kwargs) return wrapper return decorator# 普通用户接口(默认user权限)@permission_required()def get_my_order(user_id): return {"orders": [], "total": 0}# 管理员接口(需要admin权限)@permission_required(required_role="admin")def delete_user(user_id): return {"status": "success", "msg": "用户已删除"}# 调用接口print(get_my_order(1001)) # 正常执行print(delete_user(1002)) # 正常执行(当前用户是admin)
⚠️ 说明:带参数的装饰器本质是“三层嵌套函数”,外层接收装饰器参数,中层接收原函数,内层实现增强逻辑。
四、总结与练习
装饰器的核心逻辑的是“闭包+函数作为对象”,核心价值是“解耦增强逻辑与原函数”,让代码更简洁、可维护。
核心要点回顾:
📝 自习室小练习:尝试写一个“缓存装饰器”,将函数执行结果缓存到字典中,相同参数再次调用时直接返回缓存结果(避免重复计算),欢迎在评论区贴出你的代码~
下期我们继续深耕 Python 函数进阶,解锁更多后端实用技巧。关注「程序员的自习室」,一起夯实基础、进阶成长~