首页 > 解决方案 > Flask-WTF SelectMultipleField 填充数据库 ID 而不是名称

问题描述

我想先说英语不是我的母语,如果我的任何解释含糊不清或没有意义,请告诉我,我会尽量让它们更清楚。

SelectMultipleField我在使用 Flask-WTF 中的正确数据填充 a 时遇到问题。从下面的代码中可以看出,我将用户对象传递给表单。这可以完美地工作,因为它会在呈现页面时填写所有字段。我遇到的问题SelectMultipleField是填充了角色 ID,而不是角色的名称。

在表单部分中,我生成了一个列表,choices其中包含应显示的所有可用选项。我可以将其反转为choices.append((role.id, role.name)),它将使用名称而不是 ID 填充字段。但是,如果我这样做,分配给用户对象(来自数据库)的角色将不会显示为预选。

如何使该字段预填充名称而不是 ID,并保持预选分配的角色?

任何朝着正确方向的指导或轻推将不胜感激。如果缺少任何我不想添加的信息,请告诉我,我会添加它。

# Models
class Role(db.Document, RoleMixin):
    meta = {'collection': 'role'} 
    name = db.StringField(max_length=80, unique=True)
    description = db.StringField(max_length=255)

    def __repr__(self):
        return '<Role {}>'.format(self.name)

    def __str__(self):
        return self.name

class User(db.Document, UserMixin):
    meta = {'collection': 'User'}   
    first_name = db.StringField(max_length=30)
    last_name = db.StringField(max_length=30)
    email = db.StringField(max_length=120)
    password_hash = db.StringField(max_length=255)
    active = db.BooleanField(default=True)
    roles = db.ListField(db.ReferenceField(Role), default=[])

    def __repr__(self):
        return '<User {}>'.format(self.email)

    def __str__(self):
        return self.email

# Forms
class EditUserForm(FlaskForm):
    choices = []
    for role in Role.objects:
        choices.append((role.name, role.id))

    first_name = StringField('First name', validators=[DataRequired()])
    last_name = StringField('Last name', validators=[DataRequired()])
    email = StringField('Email', validators=[DataRequired(), Email()])
    password = PasswordField('Password')
    password2 = PasswordField('Repeat password', validators=[EqualTo('password')])
    roles = SelectMultipleField('User roles', choices=choices)
    submit = SubmitField('Save')

# Routes
@app.route('/users/edit/<id>', methods=['GET', 'POST'])
@roles_required('admin')
def edit_user(id):
    user = User.objects(id=id).first()
    form = EditUserForm(obj=user)

    if form.validate_on_submit() and request.method == 'POST':
        user.first_name=form.first_name.data
        user.last_name=form.last_name.data
        user.email=form.email.data
        user.roles = []

        for role in form.roles.data:
            r = Role.objects(name=role).first()
            user.set_role(r.id)

        if form.password.data:
            user.set_password(form.password.data)

        user.save()
        return redirect(url_for('users'))

    return render_template('edit_user.html', title='Edit user', user=user, form=form)

模板:

{% extends "layout.html" %}
{% set active_page = "edit_user" %}
{% block jumbotron %}

    <h1>{{ title }}</h1>
    <form action="" method="post">
        {{ form.hidden_tag() }}
        <p>
            {{ form.first_name.label }}<br>
            {{ form.first_name(size=32) }}<br>
            {% for error in form.first_name.errors %}
            <span style="color: red;">[{{ error }}]</span>
            {% endfor %}
        </p>
        <p>
            {{ form.last_name.label }}<br>
            {{ form.last_name(size=32) }}<br>
            {% for error in form.last_name.errors %}
            <span style="color: red;">[{{ error }}]</span>
            {% endfor %}
        </p>
        <p>
            {{ form.email.label }}<br>
            {{ form.email(size=32) }}<br>
            {% for error in form.email.errors %}
            <span style="color: red;">[{{ error }}]</span>
            {% endfor %}
        </p>
        <p>
            {{ form.password.label }}<br>
            {{ form.password(size=32) }}<br>
            {% for error in form.password.errors %}
            <span style="color: red;">[{{ error }}]</span>
            {% endfor %}
        </p>
        <p>
            {{ form.password2.label }}<br>
            {{ form.password2(size=32) }}<br>
            {% for error in form.password2.errors %}
            <span style="color: red;">[{{ error }}]</span>
            {% endfor %}
        </p>
        <p>
            {{ form.roles.label }}<br>
            {{ form.roles(size=32) }}<br>
            {% for error in form.roles.errors %}
            <span style="color: red;">[{{ error }}]</span>
            {% endfor %}
        </p>
        <p>{{ form.submit() }}</p>
    </form>

{% endblock %}

我尝试使用下面的代码创建自己的对象以传递给表单,但是我没有成功。将该对象传递给表单时,没有填充任何字段。

user = User.objects(id=id).first()
temps = []

u = user.to_mongo()
u['_id'] = {'$oid': str(u['_id'])}
temp_roles = []
for role in user.roles:
    r = role.to_mongo()
    r['_id'] = {'$oid': str(r['_id'])}
    temp_roles.append(r)

u['roles'] = temp_roles

标签: pythonflaskflask-wtformsflask-mongoengine

解决方案


我认为 PRMoureu 是对的。

你可以choices.append((role.name, role.name))

供参考。

SelectField 的选择是一个元组(值,标签)对的列表,请访问此链接了解更多详细信息。

选择字段保留一个选择属性,该属性是(值,标签)对的序列。理论上,值部分可以是任何类型,但由于表单数据是由浏览器作为字符串发送的,因此您需要提供一个函数,该函数可以将字符串表示强制转换为可比较的对象。

默认强制unicodeint如果您使用 (id, name)。


推荐阅读