form实现登录代码 博客

这里提供一个使用 Flask-WTF 和 Flask-Login 实现登录功能的示例代码：

forms.py：

from flask_wtf import FlaskForm
from wtforms import StringField, PasswordField, SubmitField
from wtforms.validators import DataRequired

class LoginForm(FlaskForm):
    username = StringField('Username', validators=[DataRequired()])
    password = PasswordField('Password', validators=[DataRequired()])
    submit = SubmitField('Login')

views.py：

from flask import render_template, redirect, url_for
from flask_login import login_user, logout_user, login_required
from app import app, db, login_manager
from app.models import User
from app.forms import LoginForm

@login_manager.user_loader
def load_user(user_id):
    return User.query.get(int(user_id))

@app.route('/login', methods=['GET', 'POST'])
def login():
    form = LoginForm()
    if form.validate_on_submit():
        user = User.query.filter_by(username=form.username.data).first()
        if user and user.check_password(form.password.data):
            login_user(user)
            return redirect(url_for('index'))
    return render_template('login.html', form=form)

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

其中，`LoginForm` 继承自 FlaskForm，包含了用户名、密码和登录按钮三个字段。`login` 函数处理用户提交的登录请求，首先判断表单数据是否合法，然后查询数据库中是否存在该用户，如果存在并且密码验证通过，就调用 `login_user` 函数登录用户，并重定向到首页。`logout` 函数则是登出当前用户。

需要注意的是，上述代码中使用了 Flask-Login 扩展来管理用户登录状态，需要在 app 初始化时进行相关配置：

from flask_login import LoginManager

app = Flask(__name__)
app.config['SECRET_KEY'] = 'your_secret_key'
app.config['SQLALCHEMY_DATABASE_URI'] = 'your_database_uri'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False

db = SQLAlchemy(app)
login_manager = LoginManager(app)

在 `load_user` 函数中，通过用户 ID 获取用户对象，用于判断当前用户是否已登录。在需要登录才能访问的视图函数上使用 `@login_required` 装饰器，确保只有登录用户才能访问。