Flask-Admin：使用Peewee的“选择字段”的自定义查询？

Question

我有一个带有ForeignKeyField的模型，该模型在Flask-Admin中的创建/编辑表单中作为选择字段呈现。 我想通过自定义查询来限制选择字段中的选择，以便用户只能访问自己的源地址。

我找到的所有答案都指向WTForms的QuerySelectField的方向，但这仅用于SQLAlchemy，并且我正在使用Peewee。

这似乎是一件很普通的事情，所以还有其他办法吗？

class BulkMessage(BaseModel): title = CharField(null=True) source_address = ForeignKeyField( SourceAddress, related_name='bulk_messages', null=True )

Answer 1

实际上很简单：

只是覆盖edit_form在ModelView ，并在通过查询有创建字段choices ，如看到的文档 ：

def edit_form(self):
    form = model_form(BulkMessage)

    form.source_address = SelectField(
        'Source Address',
        choices=[(sa.id, sa.title) for sa in SourceAddress.select().where(SourceAddress.owner == current_user.id)]
    )

    return form(obj=obj)