Блог

Я пытаюсь выполнить этот запрос в sqlalchemy

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

Я хотел бы привязать список [123, 456] во время выполнения.

Как насчёт

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

Предполагая, что вы используете декларативный стиль (т. Е. классы ORM), это довольно просто:

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

db_session является ли ваш сеанс базы данных здесь, в то время User как класс ORM __tablename__ равен "users" .

Альтернативный способ-использовать режим raw 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()
 

Надеюсь, у вас все получится.

С помощью 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 были определены соответствующим образом.

Просто хотел поделиться своим решением с использованием sqlalchemy и панд в python 3. Возможно, кто-то сочтет его полезным.

 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)
 

Просто дополнение к ответам выше.

Если вы хотите выполнить SQL с помощью оператора «IN», вы можете сделать это:

 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) «), просто поставьте «…В %с»
• Заставьте список ваших идентификаторов быть одним элементом кортежа. Не забудьте » , » : (список идентификаторов,)

ПРАВКА Обратите внимание, что если длина списка равна единице или нулю, это приведет к ошибке!