Flask框架之Flask-login用户登录

最新推荐文章于 2024-05-05 22:23:21 发布

样子2018

最新推荐文章于 2024-05-05 22:23:21 发布

阅读量22

点赞数

文章标签： flask python 后端

本文链接：https://blog.csdn.net/qq_27781261/article/details/136014150

版权

文档：https://flask-login.readthedocs.io/en/latest/

安装：

pip install flask-login

一、效果图

二、示例代码

from flask import Flask, render_template, url_for, redirect, flash
from flask_login import LoginManager, login_required, UserMixin, login_user, logout_user, current_user
from flask_sqlalchemy import SQLAlchemy
from sqlalchemy import desc
from flask_wtf import FlaskForm
from wtforms import StringField, SubmitField, PasswordField
from wtforms.validators import DataRequired

app = Flask(__name__)

db = SQLAlchemy(app)

# 实例化登录管理对象
login_manager = LoginManager()
# 初始化应用
login_manager.init_app(app)
# 设置用户登录视图函数，即验证失败时要跳转的页面，这里设置为登录页
login_manager.login_view = "login"
login_manager.session_protection = "strong"

''' 数据库模型'''
class User(db.Model, UserMixin):
    __tablename__ = "user"

    id = db.Column(db.Integer, primary_key = True)
    username = db.Column(db.String(32), nullable = False)
    password = db.Column(db.String(32))

'''表单模型'''
class LoginForm(FlaskForm):
    username = StringField(label="用户名", validators=[DataRequired("请输入用户名")])
    password = PasswordField(label="密码", validators=[DataRequired("请输入密码")])
    submit = SubmitField(label="登录")

''' 每访问一次，执行一次该函数 '''
@login_manager.user_loader
def load_user(user_id):
    # print("user_id=" + str(user_id))
    return User.query.filter_by(id=user_id).first()

@app.route('/')
@login_required
def index():
    # print(current_user)
    return render_template("user/index.html")

@app.route("/users")
@login_required
def users():
    users = User.query.order_by(desc("id")).all()
    return render_template("user/users.html", users = users)

@app.route("/login", methods=["POST", "GET"])
def login():
    form = LoginForm()
    if form.validate_on_submit():
        username = form.username.data
        password = form.password.data
        # 1.从数据库查找用户
        user = User.query.filter_by(username=username).first()
        print(user)
        if user is not None:
            # 2.验证密码
            if user.password == password:
                # 3.登录
                login_user(user)
                return redirect(url_for('index'))
            else:
                flash("密码不正确")
        else:
            flash("用户名不正确")
    return render_template("user/login2.html", form=form)

@app.route("/logout")
@login_required
def logout():
    logout_user()
    return redirect(url_for('login'))

'''初始化数据'''
def init_data():
    u1 = User(username="yy", password="123456")
    u2 = User(username="admin", password="654321")

    db.session.add_all([u1, u2])
    db.session.commit()

if __name__ == '__main__':
    app.debug = True
    app.secret_key = "Yang"

    app.config['SQLALCHEMY_DATABASE_URI'] = "mysql://test01:123456@127.0.0.1:3306/test01"
    app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = True

    db.drop_all()
    db.create_all()
    # 初始化数据
    init_data()

    app.run()

<!DOCTYPE html>
<html lang="en">

<head>
    <meta charset="UTF-8">
    <meta http-equiv="X-UA-Compatible" content="IE=edge">
    <meta name="viewport" content="width=device-width, initial-scale=1.0">
    <title>登录</title>
</head>

<body>
    <form action="{{url_for('login')}}" method="post" novalidate>
        {{form.csrf_token}}

        {{form.username.label}}
        <p>{{form.username}}</p>
        {% for msg in form.username.errors%}
        <p>{{msg}}</p>
        {% endfor%}

        {{form.password.label}}
        <p>{{form.password}}</p>
        {% for msg in form.password.errors%}
        <p>{{msg}}</p>
        {% endfor%}

        {{form.submit}}

        {% for message in get_flashed_messages() %}
        <div class=flash>{{ message }}</div>
        {% endfor %}
    </form>
</body>

</html>

index.html

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <meta http-equiv="X-UA-Compatible" content="IE=edge">
    <meta name="viewport" content="width=device-width, initial-scale=1.0">
    <title>index</title>
</head>
<body>
    您好，{{ current_user.username }} <a href="{{url_for('logout')}}">退出登录</a> <a href="{{url_for('users')}}">用户列表</a>
</body>
</html>

样子2018

关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
打赏
0
评论
Flask框架之Flask-login用户登录

文档：https://flask-login.readthedocs.io/en/latest/安装：pip install flask-login一、效果图二、示例代码from flask import Flask, render_template, url_for, redirect, flashfrom flask_login import LoginManager...
复制链接

扫一扫