Requêtes paramétrées avec psycopg2: SELECT à partir d'une liste Python

Question

Étant donné une grande liste (~ 10 000) d'entiers en Python, comment puis-je efficacement SELECT tous les éléments d'une table dont l'identifiant est dans la liste?

J'ai essayé:

>>> lst2 
[{'id': 97600167}, {'id': 97600168}, {'id': 97611194}] 
>>> cur.executemany("SELECT id, parent_id FROM my_table WHERE id=%(id)s", lst2) 
>>> cur.fetchall() 
[(97611194, 10020688), (None, None), (None, None)] 

Le deuxième et troisième id (97600168, 97611194) existent dans le tableau.

Answer 1

En utilisant votre exemple, utilisez where id in puis passer un paramètre qui est un tuple des valeurs id que vous souhaitez sélectionner:

>>> lst2 
[{'id': 97600167}, {'id': 97600168}, {'id': 97611194}] 
>>> ids = tuple(x['id'] for x in lst2) 
>>> cur.execute("SELECT id, parent_id FROM my_table where id in %s",[ids])