之前去面试,有一道装饰器的面试题没有答上来。
所以专门写这篇装饰器的文章,主要是为了给自己看,不用再网上去搜别人写的东西。
可参考廖雪峰的这篇文章
装饰器
在不改变原函数情况下进行功能扩展。
这个不改变包括函数内部的逻辑,和函数的调用代码。
只需要在原来函数上方加一个魔术方法。
装饰器学习
"""调试用的代码"""
import time
def foo():
print(time.time())
return ""
def decorator(func, *arg, **kwargs):
print(time.time(), 'decorator')
def wrapper(*arg, **kwargs):
t1 = time.time()
ret = func(*arg, **kwargs)
t2 = time.time()
print('time count:', t2 - t1)
return ret
return wrapper
t = 0
def decorator_arg(*args_d, **kwargs_d):
print(time.time())
def decorator_3(func):
print(time.time(), 'decorator')
def wrapper(*args, **kwargs):
t1 = time.time()
ret = func(*args, **kwargs)
t2 = time.time()
print('time count:', t2 - t1, args_d, kwargs_d)
return ret
return wrapper
return decorator_3
t1 = 0
a = 0
# @decorator(f3)
@decorator_arg('99999')
def f1(name="hyman"):
text = "This func is f1, name is %s!" % name
for i in range(9999999):
pass
print(text)
return text
a1 = 0
@decorator
def f2(name="hyman"):
text = "This func is f2, name is %s!" % name
print(text)
return text
def f3(*arg, **kwargs):
print('f3', arg, kwargs)
# print(f1())
# print(f2('llll'))
我觉得可以这么理解,把@后面的代码视为一个整体foo, 这个foo实际上是一个函数,
这个foo可以用一个变量名表示(即函数名),也可以用执行函数的返回结果(这个结果必须是函数)表示。
而@符号相当于调用foo()
,foo的输入是固定的,即@下面函数func,返回也是固定的,是一个新的函数。
这个新的函数是有扩展功能的函数,他取代了原来函数func的引用。所以当你去调用func的时候,
他执行的@运算所生成的新函数。foo就是装饰器,@运算会调用foo来生成一个新的函数来取代func函数.
可以把代码全打上断点,看代码的运行流程。
面试题
** 设计一个装饰器,使api请求资源时会读取缓存,并且可以设置缓存的超时时间。**
代码
"""设计一个装饰器,使api请求资源时会读取缓存,并且可以设置缓存的超时时间"""
import time
from datetime import datetime
from flask import Flask
app = Flask(__name__)
created = 0
cache_response = None
def cache(timeout=5):
"""缓存请求的资源,默认缓存失效时间为5秒"""
def decorator(func):
def wrapper(*arg, **kwargs):
global created, cache_response
now = time.time()
if now - created > timeout:
created = now
cache_response = func(*arg, **kwargs)
return cache_response
return wrapper
return decorator
# 添加header的装饰器
def set_header(headers):
def decorator(func):
def wrapper(*args, **kwargs):
response = func(*args, **kwargs)
response.headers.update(headers)
return response
wrapper.__name__ = func.__name__
return wrapper
return decorator
@app.route('/')
@cache(15)
def index():
return datetime.now().strftime("%Y-%m-%d %H:%M:%S") + " > Hello! "
if __name__ == '__main__':
app.run(debug=True)