例子2

Reads: 1051 Edit

models.py

# -*- coding: utf-8 -*-
from flask_login import UserMixin

class User(UserMixin):
    pass

users = [
    {'id':'tom', 'username': 'tom', 'password': '111'},
    {'id':'mic', 'username': 'mic', 'password': '111'}
]

def query_user(user_id):
    for user in users:
        if user_id == user['id']:
            return user

app.py

from flask import Flask, request, redirect, url_for, render_template, flash
from flask_login import LoginManager, login_user, logout_user, current_user, login_required
from models import User, query_user

app = Flask(__name__)
app.secret_key = '1234567'

login_manager = LoginManager()
login_manager.login_view = 'login'
login_manager.login_message_category = 'info'
login_manager.login_message = '请登录'
login_manager.session_protection = "strong"
login_manager.init_app(app)


@login_manager.user_loader
def load_user(user_id):
    print('--- load_user user_id={}'.format(user_id) )
    if query_user(user_id) is not None:
        curr_user = User()
        curr_user.id = user_id

        return curr_user


@app.route('/')
@login_required
def index():
    return 'Logged in as: %s' % current_user.get_id()


@app.route('/login', methods=['GET', 'POST'])
def login():
    if request.method == 'POST':
        user_id = request.form.get('userid')
        user = query_user(user_id)
        if user is not None and request.form['password'] == user['password']:

            curr_user = User()
            curr_user.id = user_id

            # 通过Flask-Login的login_user方法登录用户
            login_user(curr_user)

            return redirect(url_for('index'))

        flash('错误的用户名和密码,Wrong username or password!')

    # GET 请求
    return render_template('login.html')

@app.route('/home')
@login_required
def home():
    user_id = current_user.get_id()
    print('user_id={}'.format(user_id))
    return render_template('hello.html')

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


if __name__ == '__main__':
    app.run(debug=True)

templates/login.html

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Login Sample</title>
</head>
<body>
    <h1>Login</h1>
    {% with messages = get_flashed_messages() %}
        <div>{{ messages[0] }}</div>
    {% endwith %}
    <form action="{{ url_for('login') }}" method="POST">
        <input type="text" name="userid" id="userid" placeholder="用户ID"></input>
        <input type="password" name="password" id="password" placeholder="密码"></input>
        <input type="submit" name="submit"></input>
    </form>
</body>
</html>

templates/hello.html

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Login Sample</title>
</head>
<body>

{% if current_user.is_authenticated %}
  <h1>Hello {{ current_user.get_id() }}!</h1>
{% endif %}

</body>
</html>

Comments

Make a comment

www.ultrapower.com ,王硕的博客,专注于研究互联网产品和技术,提供中文精品教程。 本网站与其它任何公司及/或商标无任何形式关联或合作。