python - SQLAlchemy IN 子句

Question

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

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

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

Answer 1

怎么样

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您可以通过(or columns) 属性访问表对象上的所有字段。

Answer 2

假设您使用声明式风格（即 ORM 类），这很容易：

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

db_session是您的数据库会话，而UserORM 类是__tablename__等于"users".

Answer 3

另一种方法是在 SQLAlchemy 中使用原始 SQL 模式，我使用 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()

希望一切都适合你。

Answer 4

只是想在 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)

Answer 5

使用基于评论的表达式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已经被适当地定义。

Answer 6

只是对上述答案的补充。

如果您想使用“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,)

编辑请 注意，如果列表的长度为 1 或 0，这将引发错误！