闭包和装饰器是Python进阶必须掌握的核心语法,也是理解Flask、Django等框架源码、实现日志、鉴权、缓存等功能的基石。很多开发者学了许久仍无法手写装饰器,根本原因在于没吃透闭包原理。本文从函数作为一等公民讲起,逐步拆解闭包三要素、装饰器本质、带参装饰器、多层装饰器执行顺序,最后给出三个可直接复用的企业级装饰器。所有代码均经过测试,可放心使用。
一、前置基础:函数是一等公民
Python中函数是一等对象,这意味着:
1. 函数可以赋值给变量
2. 函数可以作为参数或返回值
3. 函数可以嵌套定义
这种特性是闭包和装饰器成立的底层基础。例如:- def outer():
- def inner():
- print("内部函数执行")
- return inner
- res = outer()
- res()
复制代码 正是由于函数可以被嵌套并且作为返回值,才诞生了闭包这种结构。
二、闭包详解:定义、条件与原理
2.1 闭包的三大必要条件
同时满足以下三点即为闭包:
- 函数嵌套(外层函数 + 内层函数)
- 内层函数引用外层函数的局部变量
- 外层函数返回内层函数
2.2 闭包实战代码- def outer(num):
- def inner():
- print(f"传入数值:{num}")
- return inner
- func = outer(100)
- func()
复制代码
2.3 闭包核心特性:变量常驻
普通函数执行结束后,局部变量会被销毁。但闭包会保留外层函数的局部变量不被GC回收,从而实现状态保存和数据常驻。
2.4 经典场景:计数器- def count_factory():
- cnt = 0
- def counter():
- nonlocal cnt
- cnt += 1
- print(f"当前计数:{cnt}")
- return counter
- c1 = count_factory()
- c1() # 输出:当前计数:1
- c1() # 输出:当前计数:2
- c1() # 输出:当前计数:3
复制代码 需注意nonlocal关键字用于在内层函数中修改外层函数的局部变量。
三、装饰器本质:基于闭包的语法糖
装饰器 = 高阶闭包 + @语法糖。核心作用:在不修改原函数代码、不改变调用方式的前提下,给函数新增功能,符合“开闭原则”(对扩展开放,对修改关闭)。
装饰器解决的问题包括:
- 避免重复代码(日志、计时、鉴权等通用逻辑复用)
- 业务代码与通用工具代码解耦
- 统一项目规范,便于维护
四、手写无参装饰器:日志统计实战
4.1 原生闭包写法(理解底层)- import time
- def timer_decorator(func):
- def wrapper(*args, **kwargs):
- start = time.time()
- res = func(*args, **kwargs)
- end = time.time()
- print(f"函数【{func.__name__}】执行耗时:{end - start:.4f}s")
- return res
- return wrapper
- @timer_decorator
- def calc_sum(n):
- s = 0
- for i in range(n):
- s += i
- return s
- calc_sum(100000)
复制代码
4.2 装饰器的执行流程
@timer_decorator 等价于 calc_sum = timer_decorator(calc_sum)。调用 calc_sum() 实际执行 wrapper():先运行前置逻辑(记录开始时间),然后执行原函数,再运行后置逻辑(计算耗时),最后返回原函数结果。
4.3 解决函数信息被覆盖的问题
直接装饰会使原函数名、文档字符串被wrapper覆盖。使用 functools.wraps 修复:- from functools import wraps
- def timer_decorator(func):
- @wraps(func)
- def wrapper(*args, **kwargs):
- start = time.time()
- res = func(*args, **kwargs)
- end = time.time()
- print(f"函数【{func.__name__}】执行耗时:{end - start:.4f}s")
- return res
- return wrapper
复制代码
五、进阶:带参数装饰器 + 多层装饰器
5.1 带参数装饰器
普通装饰器只能接收函数作为参数。带参装饰器可以自定义配置参数,实现功能开关等。例如实现一个可以控制是否输出日志的装饰器:- from functools import wraps
- def log_decorator(switch=True):
- def decorator(func):
- @wraps(func)
- def wrapper(*args, **kwargs):
- if switch:
- print(f"【日志】函数{func.__name__}开始执行")
- return func(*args, **kwargs)
- return wrapper
- return decorator
- @log_decorator(switch=True)
- def test_func():
- print("业务逻辑执行")
- test_func()
复制代码
5.2 多层装饰器叠加执行顺序
加载顺序:自下而上(离函数最近的装饰器先加载)。执行顺序:自上而下(先执行最外层装饰器的前置逻辑)。口诀:先装饰的后执行,后装饰的先执行。- from functools import wraps
- def deco1(func):
- @wraps(func)
- def wrapper():
- print("deco1 前置")
- func()
- print("deco1 后置")
- return wrapper
- def deco2(func):
- @wraps(func)
- def wrapper():
- print("deco2 前置")
- func()
- print("deco2 后置")
- return wrapper
- @deco1
- @deco2
- def hello():
- print("核心业务执行")
- hello()
- # 输出顺序:
- # deco1 前置
- # deco2 前置
- # 核心业务执行
- # deco2 后置
- # deco1 后置
复制代码
六、企业级实战三大装饰器(可直接复用)
6.1 接口鉴权装饰器- from functools import wraps
- def login_auth(func):
- @wraps(func)
- def wrapper(user_token, *args, **kwargs):
- if not user_token:
- return {"code": 401, "msg": "未登录,权限不足"}
- return func(user_token, *args, **kwargs)
- return wrapper
- @login_auth
- def get_user_info(token):
- return {"code": 200, "data": "用户信息数据"}
- print(get_user_info("")) # 返回401
- print(get_user_info("valid_token")) # 返回200
复制代码
6.2 简易缓存装饰器(避免重复计算)- from functools import wraps
- def cache_decorator(func):
- cache = {}
- @wraps(func)
- def wrapper(*args):
- if args not in cache:
- cache[args] = func(*args)
- return cache[args]
- return wrapper
- @cache_decorator
- def fib(n):
- if n <= 2:
- return 1
- return fib(n-1) + fib(n-2)
- print(fib(30)) # 首次计算递归后缓存结果,后续调用直接读取缓存
复制代码 此装饰器以函数参数为键存储结果,适合纯函数(相同输入永远得到相同输出)。
6.3 异常捕获装饰器(统一异常处理)- from functools import wraps
- def exception_catch(func):
- @wraps(func)
- def wrapper(*args, **kwargs):
- try:
- return func(*args, **kwargs)
- except Exception as e:
- print(f"【函数异常】{func.__name__}执行失败:{str(e)}")
- return None
- return wrapper
- @exception_catch
- def divide(a, b):
- return a / b
- print(divide(10, 0)) # 输出异常信息并返回None
复制代码
七、面试高频真题整理
Q1:什么是闭包?闭包的作用和缺点?
闭包是一种嵌套函数结构:外层函数返回内层函数,且内层函数引用了外层函数的局部变量。作用:保存函数执行状态、实现数据常驻、简化代码、实现装饰器。缺点:外层变量常驻内存不释放,长时间使用易导致内存泄漏。
Q2:装饰器的原理是什么?为什么不修改原代码?
装饰器基于闭包和高阶函数实现,通过包装原函数在调用前后扩展逻辑。符合开闭原则,使得业务逻辑与通用逻辑解耦,无需修改原函数内部实现。
Q3:wraps装饰器的作用?
functools.wraps 用于修复因装饰器导致的原函数信息丢失问题,保留原函数的 __name__、__doc__ 等属性。
Q4:多层装饰器的执行顺序?
加载顺序(装饰器声明顺序)自下而上,执行顺序自上而下。即最靠近函数的装饰器先被装饰,但执行时其前置逻辑最后执行。
通过以上系统讲解,闭包与装饰器的原理、手写方法、进阶用法及企业级实战已清晰呈现。建议读者动手运行每一个代码示例,加深理解。 |