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

时间:2016-04-25 21:12:52

标签: python flask-sqlalchemy flask-wtforms

我正在构建一个包含个人资料管理页面的简单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')])

我的个人资料页面视图:

@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 %}

感谢您对所有这些的任何帮助。

1 个答案:

答案 0 :(得分:5)

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

一个特殊的细节是我必须从当前用户的查询结果手动分配每个字段值(除了QuerySelectField,park)。这是因为,在我的情况下,简单地将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)