如果为None或Null,则从查询中删除对象

时间:2016-01-30 04:18:25

标签: python postgresql flask-sqlalchemy dynamicquery dynamic-queries

我正在尝试构建一个获取表单数据并删除任何None" "提交的查询,但我不确定如何处理逻辑。这是代码;

@app.route('/filterassets', methods=['GET', 'POST'])
def searchassets():
form = FilterAssetsForm()
results = None
if request.method == "POST":
    if form.validate_on_submit():
        try:
            horsepower = form.horsepower_search.data
            voltage = form.voltage_search.data
            rpm = form.rpm_search.data
            results = Motor.query.filter_by(horsepower=horsepower, voltage=voltage, rpm=rpm).all()
        except Exception as e:  
            flash(e)
            db.session.rollback()
        return render_template('filterassets.html', form=form, results=results)
return render_template('filterassets.html', form=form)

因为它是允许用户过滤而不是显式搜索的页面的后端,所以某些表单数据为空。这导致查询搜索一个或多个表单==无的结果。

示例:用户以马力形式输入100,并将RPM和电压留空。即使存在100马力的行,查询也不会返回结果,因为它正在查找Horsepower == 100和RPM以及Voltage == None的行。

我尝试使用filter(None, object)但没有成功,并认为解决方案可能在某个地方使用kwargs

我知道我需要将所有表单数据传递给将删除NoneNull条目的内容,然后将其传递给最终的Motor.query.filter_by参数,我只是不知道如何。

1 个答案:

答案 0 :(得分:3)

您可以使用过滤器数据创建dict

filter_data = {'horsepower': horsepower, 'voltage': voltage, 'rpm': rpm}

然后创建另一个dict,只包含存在值的键:

filter_data = {key: value for (key, value) in filter_data.items()
               if value}

最后将这个词作为kwargs用于你的查询:

results = Motor.query.filter_by(**filter_data).all()