如何在Flask+MySQLAlchemy注册表单中验证邮箱与用户名并防重复注册？

Hey there! 要避免用户重复注册，关键是在插入新用户数据前，先检查数据库中是否已存在相同的用户名或邮箱。我来帮你修改现有的路由代码，实现这个验证逻辑：

修改后的完整路由代码 #

@app.route('/register', methods=['GET', 'POST'])
def register():
    if request.method == 'POST':
        name = request.form.get('name')
        username = request.form.get('username')
        email = request.form.get('email')
        password = request.form.get('password')
        confirm = request.form.get('confirm')

        # 第一步：检查用户名和邮箱是否已存在
        # 查询是否有匹配的用户名
        existing_user = db.execute('SELECT * FROM users WHERE username = :username', {'username': username}).fetchone()
        # 查询是否有匹配的邮箱
        existing_email = db.execute('SELECT * FROM users WHERE email = :email', {'email': email}).fetchone()

        if existing_user:
            flash("该用户名已被注册，请更换", "danger")
            return render_template('register.html')
        if existing_email:
            flash("该邮箱已被注册，请更换", "danger")
            return render_template('register.html')
        
        # 第二步：验证密码是否匹配
        if password == confirm:
            secure_password = sha256_crypt.encrypt(str(password))
            # 第三步：插入新用户数据
            db.execute('INSERT INTO users(name, username, email, password) VALUES(:name,:username,:email,:password)', 
                       {'name':name,'username':username,'email':email,'password':secure_password})
            db.commit()
            flash("注册成功！现在可以登录啦", "success")
            return redirect(url_for('login'))
        else:
            flash("两次输入的密码不匹配", "danger")
            return render_template('register.html')
    return render_template('register.html')

关键逻辑解释 #

• 检查重复数据：用SELECT语句查询数据库，通过fetchone()判断是否有匹配记录——如果返回结果不为空，说明该用户名/邮箱已被使用。
• 精准错误提示：分开检查用户名和邮箱，给用户明确的错误反馈，比模糊的“账号已存在”体验更好。
• 执行顺序：先做重复验证，再检查密码匹配，最后才插入数据，避免不必要的加密和数据库操作。

另外，为了从数据库层面彻底防止重复，你还可以给users表的username和email字段添加唯一约束：

ALTER TABLE users ADD UNIQUE (username);
ALTER TABLE users ADD UNIQUE (email);

这样即使代码层面出现疏漏，数据库也会拒绝插入重复数据，双重保障更安全。

内容的提问来源于stack exchange，提问作者Leeuw