SQLAlchemy IN 子句

Posted 2023-02-26

【中文标题】SQLAlchemy IN 子句

【英文标题】：SQLAlchemy IN clause

【发布时间】：2012-01-26 01:09:17

【问题描述】：

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

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

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

【参考方案1】：

怎么样

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

edit：没有 ORM，它将是

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

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

【参考方案2】：

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

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

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

【讨论】：

使用~ (~User.id.in_([123,456])) 或not_ 来自sqlalchemy.sql.expression (not_(User.id.in_([123,456])))。

这是link to the doc。

【参考方案3】：

只是对上述答案的补充。

如果您想使用“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，则会引发错误！

【讨论】：

我认为你应该这样做args = [(tuple(ids_list),)]，请仔细检查。

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

这不适用于单个项目列表，因为它会生成 (item, ) 并且在 SQL 中不正确

【参考方案4】：

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

【参考方案5】：

另一种方法是在 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()

希望一切都适合你。

【讨论】：

如果您对此类简单查询使用原始 SQL，则最好直接使用 psycopg2 或其他连接器。

【参考方案6】：

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