Flask-WTF中的SelectMultipleField显示数据库ID而非名称

3

首先声明一下,英语不是我的母语,如果我的解释含糊或者不清楚,请告诉我,我会尽力让它更加明确。

我在使用 Flask-WTF 的时候遇到了一个问题,无法正确地填充 SelectMultipleField 中的数据。如下面的代码所示,我将用户对象传递给表单,这可以完美地填充所有字段。但是,我遇到的问题是,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

2
如果你在两种情况下都需要name字段,为什么不使用choices.append((role.name, role.name))呢? - PRMoureu
1个回答

4

我认为PRMoureu是正确的。

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

顺便说一下,SelectField 的选项是一个元组(value, label)对的列表,请参阅此链接获取更多详细信息。

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

如果使用(id, name),默认的coerceunicodeint


1
现在我感觉自己像个傻瓜。今天早些时候,我花了好几个小时来看这个问题,但就是无论如何也弄不明白。这段代码完全符合我所需的功能。非常感谢你详尽的回答。 - Kristoffer

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,