普通装饰器使用:
def decorator(func):
def wrapper(*args, **kwargs):
return func()
return wrapper
@decorator
def test_func():
return 'Done'
多重装饰器:
from flask import Flask, redirect
import pymysql
app = Flask(__name__)
app.debug = True
def login_decorator(func): # 模拟登陆验证
def wrapper(*args, **kwargs):
print('into login')
a = 1
if a != 1:
return redirect('http://www.baidu.com')
else:
return func(*args, **kwargs)
return wrapper
def db_connect(func): # 用于统一连接、关闭数据库
def wrapper(*args, **kwargs):
print('into db')
db = pymysql.connect(host='127.0.0.1', port=3306, user='root', passwd='root', db='test')
cursor = db.cursor()
result = func(db, cursor)
cursor.close()
db.close()
return result
return wrapper
@app.route('/')
@login_decorator
@db_connect
def hello_world(db, cursor):
cursor.execute('INSERT INTO test(msg) VALUES ("test")')
db.commit()
return 'Done'
if __name__ == '__main__':
app.run(host='0.0.0.0', port=8889)
装饰器的本质是返回函数的高阶函数,离函数最近的装饰器会最先发挥其装饰的功能,但是第一个装饰器返回的wrapper中的内容不会被执行,因为wrapper还会被第二个装饰器装饰,以此类推,直到所有装饰器执行完。等装饰器执行完后,开始执行各装饰器wrapper中的内容,这时,反而是最后装饰的装饰器中的wrapper先被执行。总结来说就是离函数近的先装饰后执行。