【Flask项目2】登录拦截器的使用和退出登录的接口开发（14）

comment—utils—》decorators.py

自定义一个装饰器，如果登录了继续访问。如果没有登录直接返回（不能继续访问）

from flask import g

def login_required(func):
	def wrapper(*args,**kwargs):
		if g.user_id is not None:
			return func(*args,**kwargs)
		else:
			return {'message':'用户没有登录，不能访问'}
	return wrapper

financial—》resource—》user—》user_resource.py
退出登录接口开发

from comment.utils.decorators import login_required

class LoginOut(Resource):
	'''
	退出登录
	'''
	#使用登录拦截的装饰器，post方法加上了登录拦截，因为前提必须登录后，才能退出登录
	
	method_decorators=[login_required]
	def post(self):
		if g.user_id:
			g.user_id=None
		return {'msg':'成功推迟登录'}
	

蓝图中加载资源

financial—》resource—》user—》init.py

#登录退出的资源
user_api.add_resource(LoginOut,'/loginOut',endpoint='loginOut')

使用了登录拦截的装饰器,资源类中的3个视图函数都加上了登录拦截

class Test(Resource):
	
	method_decorators = [login_required]

	def post(self):
	 	pass
	
	def get(self):
	 	pass
	
	def put(self):
	 	pass

使用了登录拦截的装饰器,在指定的视图方法下加登录拦截，post、get方法加上了登录拦截

class Test(Resource):
	
	method_decorators = {
		'post':[login_required],
		'get':[login_required]
	}

	def post(self):
	 	pass
	
	def get(self):
	 	pass
	
	def put(self):
	 	pass