Flask-Login插件中带了6种信号,可以基于其中的信号做一些额外工作,比如user_logged_in来记录用户的登录次数和登录IP等。
安装
pip install flask-login
初始化
import flask import Flask
from flask_login import LoginManager
app = Flask(__name__)
login_manager = LoginManager()
login_manager.init_app(app)
UserMixin
flask-login提供了UserMixin,有一些用户相关的属性。
- is_authenticated:是否被验证
- is_active:是否被激活
- is_anonymous:是否匿名用户
- get_id():获得用户的Id,并转换Unicode类型
可以在创建模型的时候继承UserMixin:
from flask_login import UserMixin,login_user
from sqlalchemy import create_engine,Column,String,Integer
from sqlalchemy.ext.declaractive import declaractive_base
engine = create_engine("mysql+pymysql://root:123456@localhost/pythondb?charset=utf8")
Base = declaractive_base()
class User(Base,UserMixin):
__tablename__ = "user"
id = Column(Integer,primary_key=True)
name = Column(String(16))
password = Column(String(16))
def login(self):
login_user(self)
当调用login方法时,登录用户就会触发user_logged_in信号,开发者可以在此做一些自定义操作,如增加登录次数:
from flask_login import user_logged_in
@user_logged_in.connect_via(app)
def _logged_in(sender,user,**extra):
...
使用user_loader装饰器的回调函数非常重要,它将决定user对象是否在登录状态:
# login_manager为上面初始化的login_manager
@login_manager.user_loader
def user_loader(id):
return session.query(User).filter_by(id=id).first()
login_required
在开发过程中,常常会遇到这样的需求,某个接口必须要登录情况下才可以获取信息,login_required正好可以满足这一需求
from flask_login import login_required
@app.route("/user_info")
@login_required
def user_info():
...
获取用户的信息,必须先登录才可以获取信息。