SQLALchemy动态过滤器_by

Posted 2021-04-30

我知道你可以为SQLAlchemy的查询建立动态过滤器，通过提供 **kwargs 到 filter_by.

例如

    filters = {'id': '123456', 'amount': '232'}
    db.session.query(Transaction).filter_by(**filters)

下面是我的问题。

如果我需要用 "大于 "或 "小于 "子句来查询怎么办？ 例如（原始SQL）。

 select * from transaction t 
 where t.amount > 10 and t.amount < 100;

答案

而不是使用 filter_by 我建议使用 filter它给你更多的选择。

例如（从手册上看）。

db.session.query(MyClass).filter(
    MyClass.name == 'some name',
    MyClass.id > 5,
)

根据你的情况

filters = (
    Transaction.amount > 10,
    Transaction.amount < 100,
)
db.session.query(Transaction).filter(*filters)

另一答案

def get_filter_by_args(dic_args: dict):
    filters = []
    for key, value in dic_args.items():  # type: str, any
        if key.endswith('___min'):
            key = key[:-6]
            filters.append(getattr(model_class, key) > value)
        elif key.endswith('___max'):
            key = key[:-6]
            filters.append(getattr(model_class, key) < value)
        elif key.endswith('__min'):
            key = key[:-5]
            filters.append(getattr(model_class, key) >= value)
        elif key.endswith('__max'):
            key = key[:-5]
            filters.append(getattr(model_class, key) <= value)
        else:
            filters.append(getattr(model_class, key) == value)
    return filters

dic_args = {
    'is_created': 1,
    'create_time__min': 1588125484029,
}
filters = get_filter_by_args(dic_args)
lst = session.query(model_class).filter(*filters)