SQLAlchemy IN-Klausel

234

Ich versuche, diese Abfrage in sqlalchemy durchzuführen

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

Ich möchte die Liste [123, 456]zur Ausführungszeit binden .

Wonzbak
quelle

Antworten:

328

Wie wäre es mit

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

edit : Ohne das ORM wäre es

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

select()Es werden zwei Parameter verwendet, der erste ist eine Liste der abzurufenden Felder, der zweite ist die whereBedingung. Sie können über die Eigenschaft c(oder columns) auf alle Felder eines Tabellenobjekts zugreifen .

Simon
quelle
6
Ich verwende derzeit nicht den ORM-Teil von sqlachemy, sondern nur die SQL Expression-API.
Wonzbak
136

Angenommen, Sie verwenden den deklarativen Stil (dh ORM-Klassen), ist dies ziemlich einfach:

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

db_sessionist Ihre Datenbanksitzung hier, während Userdie ORM-Klasse __tablename__gleich ist "users".

Xion
quelle
29
Verwenden Sie ~( ~User.id.in_([123,456])) oder not_from sqlalchemy.sql.expression( not_(User.id.in_([123,456]))).
Xion
1
Hier ist ein Link zum Dokument .
Sergey Markelov
37

Eine alternative Möglichkeit ist die Verwendung des Raw-SQL-Modus mit SQLAlchemy. Ich verwende SQLAlchemy 0.9.8, Python 2.7, MySQL 5.X und MySQL-Python als Connector. In diesem Fall wird ein Tupel benötigt. Mein Code unten aufgeführt:

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

Hoffe alles funktioniert für dich.

Jet Yang
quelle
5
Wenn Sie für solche einfachen Abfragen Raw SQL verwenden, ist es besser, psycopg2 oder einen anderen Connector direkt zu verwenden.
Omikron
6

Mit der Ausdrucks-API, nach der diese Frage basierend auf den Kommentaren fragt, können Sie die in_Methode der entsprechenden Spalte verwenden.

Abfragen

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

verwenden

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)

Dies setzt dies voraus user_tableund connwurde entsprechend definiert.

Carl
quelle
1

Ich wollte nur meine Lösung mit SQLalchemie und Pandas in Python 3 teilen. Vielleicht würde man sie nützlich finden.

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)
dmitry
quelle
-2

Nur eine Ergänzung zu den obigen Antworten.

Wenn Sie eine SQL mit einer "IN" -Anweisung ausführen möchten, können Sie dies tun:

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)

Zwei Punkte:

  • Für die IN-Anweisung ist keine Klammer erforderlich (wie "... IN (% s)"). Geben Sie einfach "... IN% s" ein.
  • Erzwingen Sie, dass die Liste Ihrer IDs ein Element eines Tupels ist. Vergessen Sie nicht das ",": (ids_list,)

BEARBEITEN Beachten Sie, dass bei einer Länge der Liste von eins oder null ein Fehler auftritt!

Majid
quelle
2
Ich denke, Sie sollten es tun args = [(tuple(ids_list),)], bitte überprüfen Sie es noch einmal.
zs2020
4
Das OP fragt, wie dies in sqlalchemy zu tun ist, ohne rohen dbapi-Aufruf zu verwenden ...
cowbert
Dies funktioniert nicht mit einer einzelnen (item, )
Elementliste,