Flask 实现 Web 的登录和注册

在 Python 中，Flask 是一个流行的 web 框架，它提供了一个简单而强大的工具箱来构建现代 web 应用程序，本篇文章将详细介绍如何使用 Flask 构建一个包含登录和注册功能的基本网站。

安装 Flask 和 SQLAlchemy

你需要安装 Flask 和 SQLAlchemy 这两个库，你可以通过 pip 来安装它们：

pip install flask flask-sqlalchemy

创建 Flask 应用

创建一个新的 Flask 应用，并配置数据库连接，这里我们假设你已经有一个 SQLite 数据库,可以通过以下代码来初始化：

from flask import Flask
from flask_sqlalchemy import SQLAlchemy
app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///users.db'
db = SQLAlchemy(app)

定义用户模型

定义一个简单的用户模型，包括用户名、密码等字段，通常情况下,密码需要进行哈希处理以增强安全性：

class User(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(80), unique=True, nullable=False)
    password = db.Column(db.String(120), nullable=False)
    def __repr__(self):
        return '<User %r>' % self.username

注册新用户

为用户提供注册功能，包括用户名和密码输入框,保存到数据库中：

@app.route('/register', methods=['GET', 'POST'])
def register():
    if request.method == 'POST':
        username = request.form['username']
        password = request.form['password']
        new_user = User(username=username, password=password)
        db.session.add(new_user)
        db.session.commit()
        flash('注册成功，请登录！')
        return redirect(url_for('login'))
    return render_template('register.html')

登录功能

当用户尝试登录时，验证用户名和密码是否正确，如果正确则重定向到主页；否则显示错误信息：

@app.route('/', methods=['GET', 'POST'])
def login():
    if request.method == 'POST':
        username = request.form['username']
        password = request.form['password']
        user = User.query.filter_by(username=username).first()
        if user and user.password == password:
            session['logged_in'] = True
            return redirect('/')
        else:
            flash('Invalid credentials')
    return render_template('login.html')

设置会话

为了保持登录状态，我们需要设置一个会话变量 logged_in：

if not current_app.testing:
    from flask_login import LoginManager, UserMixin, login_required, login_user, logout_user, current_user
    login_manager = LoginManager()
    login_manager.init_app(current_app)
    class User(UserMixin):
        pass
    @login_manager.user_loader
    def load_user(user_id):
        return User.get(id=user_id)
    app.register_blueprint(auth_bp)  # 假设你有 auth_bp 蓝图用于其他路由

配置安全措施

确保所有敏感数据都被加密存储，还可以考虑启用 HTTPS 以提高安全性。

测试应用

运行你的 Flask 应用并测试它的各个功能，你可以使用 Postman 或任何 HTTP 客户端来模拟 API 请求。

就是在 Flask 中实现基本登录和注册功能的一系列步骤，通过这种方式，你可以轻松地扩展这个基础架构,添加更多的用户管理功能和高级特性。