Flask/WTForms/QuerySelectField - 使用基于现有数据库值选择的选项呈现 QuerySelectField

Question

我正在构建一个包含个人资料管理页面的简单 CMS。在此个人资料页面以及其他字段上，存在一个下拉列表，我使用 wtforms 扩展的 QuerySelectField 从表“parks”生成：

from wtforms.ext.sqlalchemy.fields import QuerySelectField

个人资料页面上的其余字段由“用户”表填充。

假设用户已经登录，并且他们在“用户”表中的记录已经被填充。 'user' 表中的字段是来自 'parks' 表中的值的外键：'park_id'。

当登录的用户导航到他们的个人资料页面时，我想在下拉列表中显示与该用户的 park_id 外键值相对应的选择。

请注意，Flask/Python 等对我来说是非常新的。另外，我正在使用 Flask-Security，下面的用户模型基于他们的示例。也许我需要与用户/公园建立另一种关系？

显示用户保存的个人资料信息，包括可能出现在可供编辑的下拉列表中的信息，对我来说似乎非常基本。我担心我错过了一些基本的东西。到目前为止，关于 SO 和其他地方的其他类似问题/答案似乎让我非常困惑。

我的用户模型：

# Define models
roles_users = db.Table('roles_users',
        db.Column('user_id', db.Integer(), db.ForeignKey('user.id')),
        db.Column('role_id', db.Integer(), db.ForeignKey('role.id')))

class Role(db.Model, RoleMixin):
    id = db.Column(db.Integer(), primary_key=True)
    name = db.Column(db.String(80), unique=True)
    description = db.Column(db.String(255))

class User(db.Model, UserMixin):
    id = db.Column(db.Integer, primary_key=True)
    email = db.Column(db.String(255), unique=True)
    password = db.Column(db.String(255))
    active = db.Column(db.Boolean())
    confirmed_at = db.Column(db.DateTime())
    roles = db.relationship('Role', secondary=roles_users,
                            backref=db.backref('users', lazy='dynamic'))
    park_id = db.Column(db.Integer)
    weather_zip = db.Column(db.String(255))
    flickr_api_key = db.Column(db.String(255))
    flickr_api_secret = db.Column(db.String(255))
    flickr_user_id = db.Column(db.String(255))
    flickr_photo_set_id = db.Column(db.String(255))

我的公园模型：

class Parks(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    park = db.Column(db.String(255))

我的个人资料表格（带有 query_factory 的东西）

def get_parks():
    return Parks.query

class ProfileForm(Form):
    id = HiddenField()
    email = HiddenField()
    park_id = QuerySelectField('Park',[validators.InputRequired(u'Please select a park')], default=19, query_factory=get_parks, get_label='park', allow_blank=True)
    weather_zip = StringField('Zip Code',[validators.Length(min=5, max=5, message=u'Zip must contain 5 numbers'), validators.InputRequired(u'please enter a 5-digit zip code')])
    flickr_api_key = StringField('Flickr API Key', [validators.InputRequired(u'Please supply a Flickr API key')])
    flickr_api_secret = StringField('Flickr API Secret', [validators.InputRequired(u'Please supply a Flickr API secret')])
    flickr_user_id = StringField('Flickr User ID', [validators.InputRequired(u'Please supply a Flickr user id')])
    flickr_set_id = StringField('Flickr Set ID', [validators.InputRequired(u'Please supply a Flickr set id')])

我的 profile_page 视图：

@app.route('/profile/<id>', methods=["GET","POST"])
@login_required
def profile_page(id):

    id = id
    current_user = User.query.filter_by(id=id).first()

    form = ProfileForm(obj=current_user)

    return render_template("profile.html", form=form)

我的模板表格：

    {% from "_formhelpers.html" import render_field %}
    <form method=post action="/profile">
      <dl>
        {{ form.id(value=current_user.id) }}
        {{ form.email(value=current_user.email) }}
        {{ render_field(form.park_id) }}
        {{ render_field(form.weather_zip) }}
        {{ render_field(form.flickr_api_key) }}
        {{ render_field(form.flickr_api_secret) }}
        {{ render_field(form.flickr_user_id) }}
        {{ render_field(form.flickr_set_id) }}
      </dl>
      <p><input type=submit value=Register>
    </form>

渲染字段的宏（可能相关？）：

{% macro render_field(field) %}
    <dt>{{ field.label }}
    <dd>{{ field(**kwargs) |safe }}
    {% if field.errors %}
        <ul class=errors>
        {% for error in field.errors %}
            <li>{{ error }}</li>
        {% endfor %}
        </ul>
    {% endif %}
    </dd>
{% endmacro %}

感谢您对这一切的任何帮助。

Answer 1

喝了几杯啤酒并主演了这篇文章：SQLAlchemy/WTForms: set default selected value for QuerySelectField 直到我睁大眼睛，我发现我需要将查询中的结果对象传递给我的 parks 表，该表按当前用户的 id 过滤。

一个特殊的细节是，我必须从当前用户的查询结果中手动分配每个字段值（除了 QuerySelectField、parks）。这是因为，在我的情况下，简单地将 current_user 结果传递给“obj=”arg 将覆盖我的 QuerySelectField 关键字（在我的情况下为“park_id”），因为“obj=current_user”分配已经包含它自己的 park_id 参数实例，尽管它不适用于在我的下拉菜单中显示默认选择。

所以...修复方法是按如下方式更新我的 profile_page 视图功能：

@app.route('/profile/<id>', methods=["GET","POST"])
@login_required
def profile_page(id):

    id = id
    current_user = User.query.filter_by(id=id).first()
    park = Parks.query.filter_by(id=current_user.park_id).first()
    form = ProfileForm(park_id=park, # this is the main thing
                       weather_zip=current_user.weather_zip,
                       flickr_api_key=current_user.flickr_api_key,
                       flickr_api_secret=current_user.flickr_api_secret,
                       flickr_user_id=current_user.flickr_user_id,
                       flickr_photo_set_id=current_user.flickr_photo_set_id)

    return render_template("profile.html", form=form)