SQLAlchemy IN子句


237

我正在尝试在sqlalchemy中执行此查询

SELECT id, name FROM user WHERE id IN (123, 456)

我想[123, 456]在执行时绑定列表。

Answers:


331

怎么样

session.query(MyUserClass).filter(MyUserClass.id.in_((123,456))).all()

编辑:没有ORM,它将是

session.execute(
    select(
        [MyUserTable.c.id, MyUserTable.c.name], 
        MyUserTable.c.id.in_((123, 456))
    )
).fetchall()

select()需要两个参数,第一个是要检索的字段列表,第二个是where条件。您可以通过c(或columns)属性访问表对象上的所有字段。


6
我目前不使用sqlachemy的ORM部分,而仅使用SQL Expression API。
wonzbak

136

假设您使用声明式样式(即ORM类),则非常简单:

query = db_session.query(User.id, User.name).filter(User.id.in_([123,456]))
results = query.all()

db_session是您的数据库会话,User而是__tablename__等于的ORM类"users"


29
使用~~User.id.in_([123,456]))或not_sqlalchemy.sql.expressionnot_(User.id.in_([123,456])))。
Xion 2013年


38

另一种方法是将原始SQL模式与SQLAlchemy结合使用,我使用SQLAlchemy 0.9.8,python 2.7,MySQL 5.X和MySQL-Python作为连接器,在这种情况下,需要一个元组。我的代码如下:

id_list = [1, 2, 3, 4, 5] # in most case we have an integer list or set
s = text('SELECT id, content FROM myTable WHERE id IN :id_list')
conn = engine.connect() # get a mysql connection
rs = conn.execute(s, id_list=tuple(id_list)).fetchall()

希望一切对您有用。


5
如果将原始SQL用于此类简单查询,则最好直接使用psycopg2或其他连接器。
omikron 2015年

6

使用表达式API(基于注释就是这个问题的要求),您可以使用in_相关列的方法。

查询

SELECT id, name FROM user WHERE id in (123,456)

myList = [123, 456]
select = sqlalchemy.sql.select([user_table.c.id, user_table.c.name], user_table.c.id.in_(myList))
result = conn.execute(select)
for row in result:
    process(row)

这假定user_table并且conn已经适当定义。


1

只是想与我在python 3中使用sqlalchemy和pandas共享我的解决方案。也许,有人会觉得它有用。

import sqlalchemy as sa
import pandas as pd
engine = sa.create_engine("postgresql://postgres:my_password@my_host:my_port/my_db")
values = [val1,val2,val3]   
query = sa.text(""" 
                SELECT *
                FROM my_table
                WHERE col1 IN :values; 
""")
query = query.bindparams(values=tuple(values))
df = pd.read_sql(query, engine)

-2

只是上述答案的补充。

如果要使用“ IN”语句执行SQL,则可以执行以下操作:

ids_list = [1,2,3]
query = "SELECT id, name FROM user WHERE id IN %s" 
args = [(ids_list,)] # Don't forget the "comma", to force the tuple
conn.execute(query, args)

两点:

  • IN语句不需要括号(例如“ ... IN(%s)”),只需输入“ ... IN%s”
  • 强制将ID列表作为元组的一个元素。别忘了“,”:(ids_list,)

编辑请 注意,如果列表的长度为一或零,则将引发错误!


2
我认为您应该这样做args = [(tuple(ids_list),)],请仔细检查。
zs2020 2014年

4
OP正在询问如何在sqlalchemy中执行此操作,而不是使用原始dbapi调用...
Cowbert

这不适用于单个项目列表,因为它会产生(item, )并且在SQL中是不正确的
danius
By using our site, you acknowledge that you have read and understand our Cookie Policy and Privacy Policy.
Licensed under cc by-sa 3.0 with attribution required.