SQLAlchemy IN clause

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()

edit:如果没有ORM，它将是

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

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

Answer 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"。

Answer 3

另一种方法是使用SQLAlchemy的原始SQL模式，我使用SQLAlchemy 0.9.8, python 2.7, MySQL 5。和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_(list)，类似于@Carl已经建议的

 stmt = select(
        id,
         name
      ).where(
        id.in_(idlist),
      )

完整的代码，假设在User类中有数据模型:

def fetch_name_ids(engine, idlist):
    # create an empty dataframe
    df = pd.DataFrame()
    try:
        # create session with engine
        session = Session(engine, future=True)
         stmt = select(
            User.id,
            User.name
          ).where(
            User.id.in_(idlist),
          )
        data = session.execute(stmt)
        df = pd.DataFrame(data.all())
        if len(df) > 0:
            df.columns = data.keys()
        else:
            columns = data.keys()
            df = pd.DataFrame(columns=columns)
    except SQLAlchemyError as e:
        error = str(e.__dict__['orig'])
        session.rollback()
        raise error
     else:
        session.commit()
     finally:
        engine.dispose()
        session.close()

      return df

Answer 7

这个问题发布了一个选择查询的解决方案，不幸的是，它不适用于update查询。使用这种解决方案，即使在select条件下也会有所帮助。

更新查询方案:

id_list = [1, 2, 3, 4, 5] # in most cases we have an integer list or set
query = 'update myTable set content = 1 WHERE id IN {id_list}'.format(id_list=tuple(id_list))
conn.execute(query)

注意:使用一个元组而不是列表。

Answer 8

只是对上述答案的补充。

如果你想执行一个带有"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语句不需要括号(如"…")(% s))，只需输入"…"IN %s"
• 强制你的id列表是元组的一个元素。别忘了"，":(ids_list，)

编辑注意，如果list的长度是1或0，这将引发一个错误!