SQLAlchemy IN предложение

237

Я пытаюсь сделать этот запрос в sqlalchemy

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

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

wonzbak
источник

Ответы:

332

Как насчет

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
В настоящее время я не использую ORM-часть sqlachemy, а только 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класс ORM с __tablename__равным "users".

Xion
источник
29
Используйте ~( ~User.id.in_([123,456])) или not_из sqlalchemy.sql.expression( not_(User.id.in_([123,456]))).
Сион
1
Вот ссылка на документ .
Сергей Маркелов
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
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

Просто хотел поделиться своим решением, используя sqlalchemy и pandas в 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)
дмитрий
источник
-2

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

Если вы хотите выполнить 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 (например, "... IN (% s)"), просто введите "... IN% s"
  • Сделайте так, чтобы список ваших идентификаторов был одним из элементов кортежа. Не забывайте ",": (ids_list,)

РЕДАКТИРОВАТЬ Обратите внимание, что если длина списка равна единице или нулю, это вызовет ошибку!

Маджид
источник
2
Я думаю, что вы должны сделать args = [(tuple(ids_list),)], пожалуйста, проверьте дважды.
zs2020
4
ОП спрашивает, как сделать это в sqlalchemy, не используя необработанный вызов dbapi ...
Cowbert
Это не работает с одним списком элементов, так как он производит, (item, )и это неправильно в SQL
Даниус