如何自定义flask管理QuerySelectMultipleField选项?

时间:2016-11-02 13:23:49

标签: python flask flask-sqlalchemy flask-admin

我正在使用flask_admin来允许管理员用户访问数据库,其中显示了一对多关系。编辑条目时,我希望下拉菜单只显示符合条件的选项。我虽然query_factory可以做到这一点。以下是我现在所拥有的最小例子:

class OneSideObj(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    active = db.Column(db.Boolean)
    many_side_obj_id = db.Column(
        db.Integer,
        db.ForeignKey('many_side_obj.id')
    )

class ManySideObj(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    one_side_objs = db.relationship(
        'OneSideObj',
        backref = 'many_side_obj',
        lazy = 'dynamic',
    )

def get_manysideobj_id(*args):
    ##################
    # how to get id? #
    ##################
    return id

class ManySideObjView(ModelView):
    id = get_manysideobj_id(*some_args) # <--- get current many_side_obj id

    form_args = {
        'one_side_objs': {
            'query_factory': lambda: query_filter(id)
        }
    }

def query_filter(id):
    return OneSideObj.query.filter(
        (OneSideObj.many_side_obj_id == id) | (OneSideObj.active == False)
    )

我希望该字段显示的查询全部是非活动OneSideObj,或是活动的,但匹配many_side_obj_id。获取当前模型实例ID,我感到困惑。这似乎是一项简单的任务,但我找不到办法。或者这种做法可能不可行?

2 个答案:

答案 0 :(得分:1)

经过数小时的尝试/错误和谷歌搜索,似乎this几乎是我想要做的。问题归结为edit_form启动时不会调用ModelView这一事实。因此,我认为没有直接的方法来访问正在编辑的模型实例。解决方法是覆盖edit_form并且hijact传递模型实例的obj变量。这是为我工作的代码(基于最小化的示例)

class ManySideObjView(ModelView):

    # model instance is passed to edit_form as obj
    def edit_form(self, obj):
        return self._filtered(
            super(ManySideObjView, self).edit_form(obj), obj.id
        )

    # save id in self._instance_id and access it from _get_list
    def _filtered(self, form, id):
        self._instance_id = id
        form.one_side_objs.query_factory = self._get_list
        return form

    # actual query logic
    def _get_list(self):
        id = self._instance_id

        return self.session.query(OneSideObj).filter(
            (OneSideObj.active == False) | (OneSideObj.many_side_obj_id == id)
        )

答案 1 :(得分:0)

您是对的,模型实例在query_factory中不可用,但是实际上没有必要具有传递给该字段的函数。相反,您可以只创建一个查询并将其直接传递给表单。这样可以避免实例变量,并且可以通过单个方法完成。

class ManySideObjView(ModelView):
    def edit_form(self, obj):
        form = super(ManySideObjView, self).edit_form(obj)

        query = self.session.query(OneSideObj).filter(
                    (OneSideObj.active == False) | (OneSideObj.many_side_obj_id == obj.id))

        form.one_side_objs.query = query
        return form