将sqlalchemy的filter和filter_by函数包装在一个函数中

Question

我想将功能filter和filter_by包装到一个函数中，并根据参数调用filter或filter_by。

该函数的行为必须使这两个函数返回相同的结果：

MyModel.get(MyModel.mycolumn == 'value')

MyModel.get(mycolumn='value')

有没有办法做到这一点？谢谢！

Answer 1

我认为您可以根据是否以关键字给出参数来做到这一点：def get(*args, **kwargs):    if kwargs:        return query.filter_by(**kwargs)    else:        return query.filter(*args)