Cláusula SQLAlchemy IN

237

Estou tentando fazer esta consulta no sqlalchemy

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

Eu gostaria de vincular a lista [123, 456]no momento da execução.

wonzbak
fonte

Respostas:

332

E se

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

editar : Sem o ORM, seria

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

select()usa dois parâmetros, o primeiro é uma lista de campos a serem recuperados, o segundo é a wherecondição. Você pode acessar todos os campos em um objeto de tabela através da propriedade c(ou columns).

Simon
fonte
6
No momento, não estou usando a parte ORM do sqlachemy, mas apenas a API de expressão SQL.
22411 Wonzbak
136

Supondo que você use o estilo declarativo (ou seja, classes ORM), é muito fácil:

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

db_sessioné a sua sessão de banco de dados aqui, enquanto Useré a classe ORM com __tablename__igual a "users".

Xion
fonte
29
Use ~( ~User.id.in_([123,456])) ou not_de sqlalchemy.sql.expression( not_(User.id.in_([123,456]))).
Xion
1
Aqui está um link para o documento .
Sergey Markelov
38

Uma maneira alternativa é usar o modo SQL bruto com SQLAlchemy, eu uso SQLAlchemy 0.9.8, python 2.7, MySQL 5.X e MySQL-Python como conector; nesse caso, é necessária uma tupla. Meu código listado abaixo:

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

Espero que tudo funcione para você.

Jet Yang
fonte
5
Se você usa SQL bruto para consultas simples, é melhor usar psycopg2 ou outro conector diretamente.
Omikron
6

Com a expressão API, que com base nos comentários é o que esta pergunta está solicitando, você pode usar o in_método da coluna relevante.

Para consulta

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

usar

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)

Isso pressupõe isso user_tablee connfoi definido adequadamente.

Carl
fonte
1

Só queria compartilhar minha solução usando sqlalchemy e pandas no python 3. Talvez alguém ache útil.

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
fonte
-2

Apenas uma adição às respostas acima.

Se você deseja executar um SQL com uma instrução "IN", faça o seguinte:

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)

Dois pontos:

  • Não há necessidade de parênteses para a instrução IN (como "... IN (% s)"), basta colocar "... IN% s"
  • Force a lista de seus IDs a ser um elemento de uma tupla. Não esqueça o ",": (ids_list,)

EDITAR Cuidado que se o comprimento da lista for um ou zero, isso causará um erro!

Majid
fonte
2
Eu acho que você deve fazer args = [(tuple(ids_list),)], por favor, verifique.
zs2020
4
O OP é perguntando como fazer isso em sqlalchemy, não usando a chamada dbapi cru ...
cowbert
Isso não funciona com um único item da lista, uma vez que produz (item, )e é incorreto no SQL
Danius