Clausola SQLAlchemy IN


237

Sto provando a fare questa query in sqlalchemy

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

Vorrei legare l'elenco [123, 456]al momento dell'esecuzione.

Risposte:


332

Che ne dite di

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

modifica : Senza l'ORM, lo sarebbe

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

select()accetta due parametri, il primo è un elenco di campi da recuperare, il secondo è la wherecondizione. È possibile accedere a tutti i campi di un oggetto tabella tramite la proprietà c(o columns).


6
Attualmente non sto usando la parte ORM di sqlachemy, ma solo l'API SQL Expression.
wonzbak,

136

Supponendo che tu usi lo stile dichiarativo (cioè le classi ORM), è abbastanza facile:

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

db_sessionè la sessione del database qui, mentre Userè la classe ORM con __tablename__uguale a "users".


29
Usa ~( ~User.id.in_([123,456])) o not_da sqlalchemy.sql.expression( not_(User.id.in_([123,456]))).
Xion,


38

Un modo alternativo è usare la modalità SQL raw con SQLAlchemy, io uso SQLAlchemy 0.9.8, python 2.7, MySQL 5.X e MySQL-Python come connettore, in questo caso è necessaria una tupla. Il mio codice è elencato di seguito:

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

Spero che tutto funzioni per te.


5
Se si utilizza SQL non elaborato per query così semplici, è meglio utilizzare direttamente psycopg2 o altri connettori.
omikron,

6

Con l'API di espressione, che in base ai commenti è ciò che questa domanda richiede, è possibile utilizzare il in_metodo della colonna pertinente.

Per interrogare

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

uso

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)

Questo presuppone che user_tablee connsiano stati definiti in modo appropriato.


1

Volevo solo condividere la mia soluzione usando sqlalchemy e panda in Python 3. Forse lo troveremmo utile.

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

Solo un'aggiunta alle risposte sopra.

Se si desidera eseguire un SQL con un'istruzione "IN", è possibile:

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)

Due punti:

  • Non è necessaria parentesi per l'istruzione IN (come "... IN (% s)"), basta inserire "... IN% s"
  • Forza l'elenco dei tuoi ID come elemento di una tupla. Non dimenticare ",": (ids_list,)

EDIT Fai attenzione che se la lunghezza della lista è una o zero questo genererà un errore!


2
Penso che dovresti farlo args = [(tuple(ids_list),)], per favore ricontrolla.
zs2020,

4
L'OP sta chiedendo come farlo in sqlalchemy, non usando la chiamata raw dbapi ...
cowbert

Questo non funziona con un singolo elenco di elementi in quanto produce (item, )ed è errato in SQL
danius
Utilizzando il nostro sito, riconosci di aver letto e compreso le nostre Informativa sui cookie e Informativa sulla privacy.
Licensed under cc by-sa 3.0 with attribution required.