Python 仅对某些方法应用Flask MethodView装饰器
我想知道是否有任何方法可以只对我的类中的一些方法使用decorator,例如,在下面的代码中,我希望Python 仅对某些方法应用Flask MethodView装饰器,python,flask,view,python-decorators,Python,Flask,View,Python Decorators,我想知道是否有任何方法可以只对我的类中的一些方法使用decorator,例如,在下面的代码中,我希望token\u required来修饰除POST之外的所有方法,我怎么可能做到这一点 class UserAPI(MethodView): def token_required(view_method): @wraps(view_method) def decorated(*args, **kwargs): token = None
token\u required
来修饰除POST之外的所有方法,我怎么可能做到这一点
class UserAPI(MethodView):
def token_required(view_method):
@wraps(view_method)
def decorated(*args, **kwargs):
token = None
if 'token' in request.headers:
token = request.headers['token']
if not token:
return "no token"
return view_method(*args, **kwargs)
return decorated
decorators = [token_required]
def get(self, user_id):
if user_id is None:
# return a list of users
return "no user id"
else:
# expose a single user
return "user id"
def post(self):
# create a new user
return "create new user"
def delete(self, user_id):
# delete a single user
return "delete user id"
def put(self, user_id):
# update a single user
return "update user id"
这里有一些方法
1)将请求方法的条件添加到装饰器中
:
@wraps(view_method)
def decorated(*args, **kwargs):
if request.method == 'POST':
# skip token processing - just result of view method
return view_method(*args, **kwargs)
token = None # etc... (your code)
2)С为任何点创建一个通用的装饰者
,并在没有装饰者
属性的情况下使用他
from functools import wraps
def token_required():
def _token_required(f):
@wraps(f)
def __token_required(*args, **kwargs):
token = None
if 'token' in request.headers:
token = request.headers['token']
if not token:
return "no token"
return f(*args, **kwargs)
return __token_required
return _token_required
当需要令牌时,只需添加到特定端点:
@token_required()
def get(self, user_id):
...
希望这有帮助