前言:
在web开发过程中,经常会遇到数据查询慢的问题,对于查询一些时效性不是特别强的数据可以将数据缓存到redis中, 查询的时候直接到缓存中读取数据,这样就可以大大加快查询速度。
但是在开发前,事先没有增加缓存这个功能,并且需要增加缓存功能的地方又有很多,为了在不改变原有代码的基础上添加新的功能,这就要用到python的装饰器。这里简要记录一下装饰器的使用!
说明:
弄清了装饰器函数的执行流程,以及参数如何传递等,装饰器的使用基本上就差不多了。这里通过一下几步来学习装饰器的使用:
1.装饰器的引入;
2.被装饰函数无参的装饰器;
3.被装饰函数带参的装饰器;
4.装饰器带参的装饰器。
装饰器的引入
# coding:utf-8
import time
def decorator(fun):
"""统计函数的运行时间"""
start = time.time()
fun()
end = time.time()
print end - start
def do_something():
for i in range(5):
time.sleep(1)
decorator(do_something)
代码解读:
将do_something函数作为decorator函数的参数传入,在decorator函数中实现新的业务逻辑,并调用do_something函数, 这样就实现了在添加新功能的同时,不改变原有代码的逻辑,该写法不够简洁。
被装饰函数无参的装饰器
# coding:utf-8
import time
def decorator(fun):
def wrapper():
start = time.time()
fun()
end = time.time()
print end - start
return wrapper
@decorator
def do_something():
for i in range(5):
time.sleep(1)
do_something()
# 等价于:decorator(do_something)()
代码解读:
do_something()添加装饰器后,等价于do_something() = decorator(do_something)(),执行过程如下:
- 执行decorator方法,并将do_something方法作为参数传递;
- 返回内层函数wrapper的引用;
- 执行wrapper方法,等价于decorator(do_something)()
被装饰函数带参的装饰器
# coding:utf-8
import time
def decorator(fun):
def wrapper(*args, **kwargs):
start = time.time()
fun(*args, **kwargs)
end = time.time()
print end - start
return wrapper
@decorator
def do_something(times):
for i in range(times):
time.sleep(1)
do_something(5)
# 等价于:decorator(do_something)(5)
代码解读:
do_something(params)添加装饰器后,等价于do_something(params) = decorator(do_something)(params),执行过程如下:
- 执行decorator方法,并将do_something方法作为参数传递;
- 返回内层函数wrapper的引用;
- 执行wrapper方法,等价于decorator(do_something)(params),可知,参数是通过内层函数wrapper传递进来的。
装饰器带参的装饰器
# coding:utf-8
import time
def decorator(name="带参装饰器"):
def wrapper(fun):
def inner(*args, **kwargs):
start = time.time()
fun(*args, **kwargs)
end = time.time()
print end - start
print name
return inner
return wrapper
@decorator(name="带参装饰器")
def do_something(times):
for i in range(times):
time.sleep(1)
do_something(5)
# 等价于:decorator("带参装饰器")(do_something)(params)
代码解读:
do_something(params)添加装饰器后,等价于do_something(params) = decorator("带参装饰器")(do_something)(params),执行过程如下:
- 执行decorator方法,并传递参数name,
- 返回内层函数wrapper的引用,wrapper = decorator("带参装饰器")
- 执行wrapper方法, 并传递do_something方法的引用,返回inner函数的引用,inner = decorator("带参装饰器")(do_something)
- 执行inner方法,并传递参数params,实现inner方法的内部逻辑, 即:decorator("带参装饰器")(do_something)(params)
喜欢点个赞!