wtforms - 如何在 model_form 中使用 current_user

Question

我需要过滤一个仅显示用户公司中的组的选择菜单。使用 sqlalchemy 有 QuerySelectField，但是，这在 flask-mongoengine 中似乎不可用。理想情况下，我想将 flask_login current_user 传递给 model_form，如下所示：

wtfUser = model_form(User, wtf.Form, exclude=[ 'password'], field_args = {
    'group' : { 'queryset': Group.objects(company = current_user.group.company) },
    'roles' : { 'queryset': Role.objects(name__ne = 'admin').order_by('name') }
})

但是，由于这会将 current_user 置于请求上下文之外，因此会导致错误。有没有人知道如何做到这一点？

Answer 1

然而，这不是最优雅的方法，但它确实有效。在 views.py 中，我执行以下操作。如果您有更好的方法，请发布。

def user(id):
    user = User.objects.get_or_404(id = id)
    wtf_user = wtfUser(obj = user)
    wtf_user.group.queryset = Group.objects(id = current_user.group.id)
    return render_template('user.html', wtf_user = wtf_user)