python - sqlalchemy の filter および filter_by 関数を 1 つの関数にラップする

Question

関数 filter と filter_by を 1 つの関数にラップし、パラメーターに基づいて filter または filter_by を呼び出します。

この関数は、これら 2 つが同じものを返すように動作する必要があります。

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)