我想知道是否可以对类中的某些方法使用装饰器,例如,在下面的代码中,我希望token_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"