Python库flask实现激活码用户创建、存储和校验

为了完善激活码功能的示例，我们将实现用户创建时自动生成激活码并将其保存至数据库（这里使用SQLite作为示例数据库），然后提供激活码的校验逻辑。请注意，在真实项目中，你可能需要考虑更多因素，比如数据加密、错误处理、日志记录等。

1. 安装Flask-SQLAlchemy

首先，确保安装了Flask-SQLAlchemy，这是一个Flask扩展，用于操作SQLAlchemy ORM与数据库交互。

pip install flask-sqlalchemy

2. 初始化Flask应用及数据库模型

接下来，我们定义一个User模型来表示用户，包含激活码字段，并初始化Flask应用及数据库。

from flask import Flask, request, render_template, redirect, url_for
from flask_sqlalchemy import SQLAlchemy
import secrets

app = Flask(__name__)
app.config['SECRET_KEY'] = 'your_secret_key'
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///site.db'  # 使用SQLite数据库
db = SQLAlchemy(app)

class User(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    email = db.Column(db.String(120), unique=True, nullable=False)
    activation_code = db.Column(db.String(32), nullable=False)
    activated = db.Column(db.Boolean, default=False)

db.create_all()  # 创建所有表

# 用户注册逻辑（简化版本）
@app.route('/register', methods=['GET', 'POST'])
def register():
    if request.method == 'POST':
        email = request.form.get('email')
        activation_code = secrets.token_hex(16)
        new_user = User(email=email, activation_code=activation_code)
        db.session.add(new_user)
        db.session.commit()
        send_activation_email(email, activation_code)
        return redirect(url_for('sent'))
    return render_template('register.html')  # 假设存在一个register.html模板

@app.route('/sent')
def sent():
    return "激活邮件已发送，请查收邮件并完成激活!"

# 激活逻辑
@app.route('/activate/<string:code>')
def activate(code):
    user = User.query.filter_by(activation_code=code).first()
    if user:
        user.activated = True
        db.session.commit()
        return "账户已成功激活!"
    else:
        return "无效的激活码。"

def send_activation_email(to, activation_code):
    # 邮件发送逻辑，与之前示例相同
    pass

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

说明

• 在这个示例中，我们定义了一个User模型，其中包含了email（用户邮箱）、activation_code（激活码）和activated（是否已激活）字段。
• 用户注册时，通过POST请求提交邮箱地址，后端会生成一个激活码并将其与用户信息一起保存到数据库。
• 注册后，系统会自动发送一封包含激活链接的邮件给用户。激活链接中包含了激活码作为参数。
• 当用户点击邮件中的链接访问/activate/<code>时，后端会检查该激活码是否有效，如果有效则将用户状态标记为已激活。

请确保你有相应的HTML模板文件（例如register.html）来处理用户界面展示。此外，实际部署时应确保数据库连接信息的安全，以及考虑使用更安全的邮件发送方法。