### Flask 实现用户登录和注册功能 在 Flask 中实现用户登录和注册功能可以通过多种方式完成，通常涉及使用 Flask 的扩展库来简化开发过程。以下是具体方法： #### 使用 Flask-Login 和 Flask-WTF 处理登录功能 Flask-Login 是一个常用的扩展库，用于管理用户的会话状态。它提供了 `UserMixin` 类和其他工具函数，可以轻松处理用户的身份验证。 ```python from flask import Flask, redirect, url_for, render_template, request, flash from flask_sqlalchemy import SQLAlchemy from flask_login import LoginManager, UserMixin, login_user, logout_user, current_user, login_required app = Flask(__name__) app.config['SECRET_KEY'] = 'your_secret_key' app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///users.db' db = SQLAlchemy(app) login_manager = LoginManager() login_manager.init_app(app) login_manager.login_view = 'login' class User(UserMixin, db.Model): id = db.Column(db.Integer, primary_key=True) username = db.Column(db.String(15), unique=True) password = db.Column(db.String(80)) @login_manager.user_loader def load_user(user_id): return User.query.get(int(user_id)) @app.route('/login', 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 not user or not (user.password == password): flash('Please check your login details and try again.') return redirect(url_for('login')) login_user(user) return redirect(url_for('dashboard')) return render_template('login.html') @app.route('/logout') @login_required def logout(): logout_user() return redirect(url_for('index')) @app.route('/dashboard') @login_required def dashboard(): return f'Welcome to the Dashboard {current_user.username}!' if __name__ == '__main__': app.run(debug=True) ``` 上述代码实现了基本的用户登录流程[^1]。 #### 添加用户注册功能 为了支持新用户的注册，可以创建一个新的视图函数并设计相应的 HTML 表单。下面展示如何集成注册功能到现有应用程序中。 ```python @app.route('/register', methods=['GET', 'POST']) def register(): if request.method == 'POST': username = request.form['username'] password = request.form['password'] existing_user = User.query.filter_by(username=username).first() if existing_user is None: new_user = User(username=username, password=password) db.create_all() db.session.add(new_user) db.session.commit() return '<h1>New user has been registered</h1>' else: return '<h1>This username already exists!</h1>' return render_template('register.html') ``` 此部分代码允许用户提交他们的用户名和密码以创建账户[^3]。 #### 数据库存储配置 如果需要更复杂的数据库操作或者希望连接 MySQL 数据库，则可以根据实际情况调整设置。例如，在 PyCharm 配合 MySQL 进行开发时，可参考如下示例[^4]: ```python import pymysql.cursors connection = pymysql.connect( host='localhost', user='root', password='your_password_here', database='flask_db', charset='utf8mb4', cursorclass=pymysql.cursors.DictCursor ) try: with connection.cursor() as cursor: sql = "INSERT INTO users (`username`, `password`) VALUES (%s, %s)" cursor.execute(sql, ('testuser', 'securepass')) connection.commit() finally: connection.close() ``` 以上脚本演示了如何向 MySQL 数据库插入数据记录。 ---