web-dev-qa-db-fra.com

Clause SQLAlchemy IN

J'essaie de faire cette requête dans sqlalchemy

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

Je voudrais relier la liste [123, 456] au moment de l'exécution.

pythonsqlalchemy
194
wonzbak

Que diriez-vous

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

edit : Sans l'ORM, ce serait

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

select() prend deux paramètres, le premier est une liste de champs à récupérer, le second est la condition where. Vous pouvez accéder à tous les champs d'un objet table via la propriété c (ou columns).

259
Simon

En supposant que vous utilisiez le style déclaratif (c'est-à-dire les classes ORM), c'est assez simple:

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

db_session est votre session de base de données ici, tandis que User est la classe ORM avec __tablename__ égal à "users".

117
Xion

Une autre solution consiste à utiliser le mode SQL brut avec SQLAlchemy. J'utilise SQLAlchemy 0.9.8, python 2.7, MySQL 5.X et MySQL-Python comme connecteur. Dans ce cas, un Tuple est nécessaire. Mon code énuméré ci-dessous:

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

J'espère que tout fonctionne pour vous.

31
Jet Yang

Avec cette API d’expression, qui repose sur les commentaires, vous pouvez utiliser la méthode in_ de la colonne correspondante.

Pour interroger

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

utilisation

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)

Cela suppose que user_table et conn ont été définis de manière appropriée.

6
Carl

Juste un ajout aux réponses ci-dessus.

Si vous voulez exécuter un SQL avec une instruction "IN", vous pouvez le faire:

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)

Deux points:

  • La parenthèse n'est pas nécessaire pour l'instruction IN (comme "... IN (% s)"), il suffit de mettre "... IN% s"
  • Forcer la liste de vos identifiants à être l’un des éléments d’un tuple. N'oubliez pas le ",": (ids_list,)

EDIT Attention, si la longueur de la liste est un ou zéro, cela provoquera une erreur!

1
Majid