flask_login简单使用

from flask import g, request, Flask, current_app, jsonify, render_template, flash, redirect, url_for
from flask_cors import CORS

app = Flask(__name__)

from flask_login import UserMixin, logout_user, login_user, login_required, LoginManager


class User(UserMixin):
    pass


CORS(app, resources=r'/*')
app.secret_key = 'abc'
login_manager = LoginManager()

login_manager.login_view = 'api/login'
login_manager.login_message_category = 'info'
login_manager.login_message = 'Access denied.'

login_manager.init_app(app)


@login_manager.user_loader
def load_user(user_id):
    curr_user = User()
    curr_user.id = user_id

    return curr_user


@app.route('/api/login', methods=['GET', 'POST'])
def login():
    if request.method == 'POST':
        account = request.json.get('account')
        if account == "123456" and request.json['password'] == "123456":
            curr_user = User()
            curr_user.id = account
            # 通过Flask-Login的login_user方法登录用户
            login_user(curr_user)
            return {"code": 200, "message": "success"}
        else:
            return {"code": 501, "message": "登陆失败"}


    # GET 请求
    return {"code": 202, "message": "get is nothing"}


@app.route('/logout')
@login_required
def logout():
    logout_user()
    return 'Logged out successfully!'