Пункт SQLAlchemy IN


237

Я намагаюся зробити цей запит у sqlalchemy

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

Я хотів би прив’язати список [123, 456]під час виконання.

Відповіді:


332

Як щодо

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

редагувати : Без ОРМ було б

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

select()приймає два параметри, перший - список полів для отримання, другий - whereумова. Ви можете отримати доступ до всіх полів об’єкта таблиці через властивість c(або columns).


6
Наразі я не використовую частину ORM sqlachemy, а лише API SQL Expression.
winzbak

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


29
Використовуйте ~( ~User.id.in_([123,456])) або not_від sqlalchemy.sql.expression( not_(User.id.in_([123,456]))).
Сіон


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 або інший роз'єм безпосередньо.
омікрон

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 та панди в 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, не використовуючи сирої дзбапі дзвінки ...
cowbert

Це не працює з одним списком елементів, оскільки він створює, (item, )і це неправильно в SQL
danius
Використовуючи наш веб-сайт, ви визнаєте, що прочитали та зрозуміли наші Політику щодо файлів cookie та Політику конфіденційності.
Licensed under cc by-sa 3.0 with attribution required.