Flask装饰器简单使用

Flask装饰器使用

1. 基于FBV的装饰器设置使用时需要注意装饰器嵌入的位置，装饰器需要在请求进入路由之前即在请求未走原逻辑代码的时候介入，对原业务逻辑进行业务拓展，下面直接上代码

from flask import Flask, request
from functool import wraps

app = Flask(__name__)

def login(f):
	@wraps(f)
	def wrapper(*args, **kwargs):
		print('处理登录逻辑部分', request.url)
		func = f(*args, **kwargs)
		return func
	return wrapper

@app.route('/data1', methods=['GET', 'POST'])
@login # 装饰器放在路由逻辑之前
def data1():
	pass

2. 基于CBV的装饰器设置使用就显得更加细分化，可以定制管理专属功能，在外部定义装饰器可以全局使用，内部定义可以针对特定路由函数特殊处理。(Django中类装饰器要借助method_decorator方法实现)

from flask import Flask, request, views
from functools import wraps

app = Flask(__name__)

def login(f):
	@wraps(f)
	def wrapper(*args, **kwargs):
		print('处理登录逻辑部分', request.method)
		func = f(*args, **kwargs)
		return func
	return wrapper

class Data(views.MethodView):
	@login	
	def get(self):
		return request.args
		
	@login
	def post(self):
		return request.json

app.add_url_rule(rule='/data', view_func=Data.as_view('data'))