如何修改Flask Security的RegisterForm将字段值存入其他表

有两种实用方案可以实现你的需求，无需直接查找框架内部的POST请求处理代码：

方案一：利用user_registered信号处理 #

Flask-Security内置了user_registered信号，会在用户注册成功后自动触发，你可以通过监听这个信号来处理自定义字段的存储逻辑：

1. 导入所需模块：

from flask_security import user_registered
from flask import current_app

2. 定义信号回调函数，提取表单中的role值并写入目标表：

@user_registered.connect_via(current_app._get_current_object())
def handle_user_registered(sender, user, form, **extra):
    # 假设你的目标表模型是RoleAssignment，包含user_id和role_name字段
    from your_app.models import RoleAssignment, db
    # 从表单获取选中的角色值
    selected_role = form.role.data
    # 创建并保存新记录
    role_assignment = RoleAssignment(user_id=user.id, role_name=selected_role)
    db.session.add(role_assignment)
    db.session.commit()

方案二：自定义注册视图覆盖默认逻辑 #

如果需要更灵活的流程控制，可以直接重写Flask-Security的注册视图，完全掌控POST请求的处理过程：

1. 导入必要模块：

from flask_security import register_user, render_template, redirect, url_for, flash
from flask_security.utils import get_message
from your_app.forms import Registerform_roles

2. 定义自定义注册路由：

@app.route("/register", methods=["GET", "POST"])
def custom_register():
    form = Registerform_roles()
    
    if form.validate_on_submit():
        # 调用Flask-Security内置方法创建用户
        user = register_user(**form.to_dict(only=["email", "password"]))
        if user:
            # 处理自定义role字段，写入目标表
            from your_app.models import RoleAssignment, db
            selected_role = form.role.data
            role_assignment = RoleAssignment(user_id=user.id, role_name=selected_role)
            db.session.add(role_assignment)
            db.session.commit()
            
            flash(*get_message("REGISTRATION_SUCCESSFUL"))
            return redirect(url_for("security.login"))
        else:
            flash(*get_message("EMAIL_ALREADY_ASSOCIATED"))
    
    return render_template("security/register_user.html", form=form)

注意：使用自定义视图时，需要在初始化Flask-Security时设置register_route=False，禁用框架默认的注册路由，避免路由冲突。

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