Flask 简单获取用户信息以及权限认证

目录

1.下载并导入flask工具包flask_login

2.创建工具包，导入flask_login

3.装饰器login_manager.user_loader

4.启动页导入使用login_manager

5.模型类导入flask_login中的UserMixin

1.下载flask_login

pip install flask_login

2.创建工具包，导入flask_login (我起的目录名字是tools,文件名字是login.py)

3.导入flask_login并使用装饰器  login_manager.user_loader

 4.启动页使用login_manager

#从自己创建的工具包中导入
from tools.login import login_manager
#使用
login_manager.init_app(app)

 5.模型类中导入flask_login 中的  UserMixin

#以用户模型为例
from flask_login import UserMixin

class User(db.Model,UserMixin):
    id = db.Column(db.Integer,primary_key=True)
    login_time = db.Column(db.DateTime,default=datetime.now)
    img = db.Column(db.String(2048))
    password = db.Column(db.String(1024))
    make_password = db.Column(db.String(16))
    mobile = db.Column(db.String(11))
    address = db.Column(db.String(42))
    real_name = db.Column(db.String(42))
    id_card = db.Column(db.Integer)
    landlord = db.Column(db.Boolean,server_default='0')

以上步骤均已完成，下面就可以直接在试图使用，并认证用户的登录状态

6.视图中使用

# 登录
@user.route('/login',methods=['post'])
def user_login():
    #从前端获取用户手机号校验和密码
    mobile = request.form.get('mobile')
    pwd = request.form.get('password')
    #获取用户的信息
    user = User.query.filter_by(mobile=mobile).first()
    #这里判断前端用户的密码和数据库里面的数据是否保持一致（使用了md5加密算法，主要是安全）
    if user and user.password == md5(pwd.encode()+MD5_SALT).hexdigest():
        #保存用户信息生成cooike
        login_user(user)
        #返回响应(my_router 和 item2dict 是自己封装的一个方法)
        return my_router(200,item2dict(user))
    return my_router(206)

# 退出
@user.route('/login_out',methods=['get'])
@login_required
def user_out():
    print(current_user)
    #logout_user()也是flask_login中的一个方法，可以直接导入logout_user
    logout_user()
    return my_router(200)


# 获取用户信息
@user.route('/info',methods=['get'])
@login_required
def get_user_info():
    return my_router(200, item2dict(current_user,filter=['password']))