2011-12-22 11 views
133

estoy tratando de hacer esta consulta en sqlalchemysqlalchemy en la cláusula

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

quisiera obligar a la lista [123456] en tiempo de ejecución.

Respuesta

166

¿Qué tal

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

edición: Sin el ORM, sería

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

select() toma dos parámetros, el primero es una lista de campos para recuperar, el segundo es la condición where. Puede acceder a todos los campos en un objeto de tabla a través de la propiedad c (o columns).

+4

Actualmente no estoy usando la parte ORM de sqlachemy, pero solo la API de expresión SQL. – wonzbak

94

Asumiendo que utiliza el estilo declarativa (es decir, las clases ORM), que es bastante fácil:

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

db_session es su sesión de base de aquí, mientras User es la clase ORM con __tablename__ igual a "users".

+1

en la cláusula funciona bien, cómo usar la cláusula notin_ (NOT IN) – sunny

+16

Use '~' ('~ User.id.in _ ([123,456])') o 'not_' from' sqlalchemy.sql.expression' (' not_ (User.id.in _ ([123,456])) '). – Xion

+1

muchas gracias @Xion – sunny

2

Simplemente una adición a las respuestas anteriores.

Si desea ejecutar un SQL con una sentencia "en" usted puede hacer esto:

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) 

dos puntos:

  • No hay necesidad de paréntesis para la declaración EN (como "... IN (% s)"), simplemente ponga "... IN% s"
  • Fuerce que la lista de sus identificadores sea un elemento de una tupla. No se olvide del "": (ids_list,)

EDITAR cuidado de que si la longitud de la lista es uno o cero, esto generará un error!

+2

Creo que deberías hacer 'args = [(tuple (ids_list),)]', por favor revisa. – zsong

+1

El OP pregunta cómo hacer esto en sqlalchemy, sin utilizar la llamada de dbapi sin procesar ... – cowbert

4

Con la expresión API, que en función de los comentarios es lo que esta pregunta está pidiendo, puede utilizar el método in_ de la columna correspondiente.

Para consultar

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) 

Esto supone que user_table y conn se han definido adecuadamente.

13

Una forma alternativa es usar el modo SQL sin procesar con SQLAlchemy, utilizo SQLAlchemy 0.9.8, python 2.7, MySQL 5.X y MySQL-Python como conector, en este caso, se necesita una tupla. Mi código se enumera a continuación:

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 todo funcione para usted.

+3

Si usa SQL sin formato para consultas tan simples, es mejor que use psycopg2 u otro conector directamente. – omikron