Question

我尝试使用Sqlalchemy和Postges有效地进行SELECT ... WHERE id IN (1,2,3)。

如果我做一个简单的选择：

s.query(Model).filter(Model.id.in_([1,2,3])).all()

Sqlalchemy运行此查询：

SELECT model.id AS model_id FROM model 
WHERE model.id IN (%(id_1)s, %(id_2)s, %(id_3)s)
{'id_1': 1, 'id_2': 2, 'id_3': 3}

当数组变长时，效率不高。这也不适用于烘焙查询。

知道Postgres支持元组作为参数，我试图通过使用bind参数将我的Array / Tuple直接放入参数部分：

s.query(Model)
 .filter(Model.id.in_(bindparam('my_tuple')))
 .params(my_tuple=(1,2,3)).all()

不幸的是，Sqlalchemy不接受bindparam中的in_：

sqlalchemy.exc.InvalidRequestError: 
    in_() accepts either a list of expressions or a selectable:
        BindParameter('my_tuple', None, type_=NullType())

所以我试图以某种方式欺骗Sqlalchemy接受bindparam 扩展BindParam课我能够这样做：

class TupleBindParameter(BindParameter, Selectable):
    pass

s.query(Model)
 .filter(Model.id.in_(TupleBindParameter('my_tuple')))
 .params(my_tuple=(1,2,3)).all()

现在我得到了我想要的东西：

SELECT model.id AS model_id FROM model 
WHERE model.id IN %(my_tuple)s
{'my_tuple': (1, 2, 3)}

这个解决方案对我来说似乎有点不好意思。是否有正式的方法让Sqlalchemy做同样的事情？

-

重现我的示例的设置非常简单：

Base = declarative_base()    
class Model(Base):
    __tablename__ = 'model'
    id = Column(Integer, primary_key=True)
    def __init__(self, id): self.id = id

engine = create_engine('postgres://x:x@localhost/x')    
Base.metadata.create_all(bind=engine)

Session = sessionmaker(bind=engine)
s = Session()
s.add_all([Model(1), Model(2), Model(4)])
s.commit()

Answer 1

使用op('IN')

s.query(Model)
 .filter(Model.id.op('IN')(bindparam('my_tuple')))
 .params(my_tuple=(1,2,3)).all()

请参阅this issue

在Sqlalchemy / Postgres

1 个答案: