Подтвердить что ты не робот

"SELECT... WHERE... IN" с неизвестным количеством параметров

Я пытаюсь выполнить запрос в форме...

SELECT col2 FROM tab WHERE col1 IN (val1, val2, val3...)

... где значения хранятся в списке/кортеже Python произвольной длины. Кажется, я не могу найти "чистый" способ сделать это.

>>> db = connect(":memory:")
>>> db.execute("CREATE TABLE tab (col1 INTEGER, col2 TEXT)")
>>> db.execute("INSERT INTO tab VALUES(1,'one')")
>>> db.execute("INSERT INTO tab VALUES(2,'two')")
>>> db.execute("INSERT INTO tab VALUES(3,'three')")
>>> db.execute("INSERT INTO tab VALUES(4,'four')")
>>> db.execute("INSERT INTO tab VALUES(5,'five')")
>>> db.commit()

# Expected result
>>> db.execute("SELECT col2 FROM tab WHERE col1 IN (1,3,4)").fetchall()
[(u'one',), (u'three',), (u'four',)]

>>> vals = (1,3,4)

>>> db.execute("SELECT col2 FROM tab WHERE col1 IN (?)", vals).fetchall()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
sqlite3.ProgrammingError: Incorrect number of bindings supplied. The current statement uses 1, and there are 3 supplied.

>>> db.execute("SELECT col2 FROM tab WHERE col1 IN (?)", (vals,)).fetchall()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
sqlite3.InterfaceError: Error binding parameter 0 - probably unsupported type.

>>> db.execute("SELECT col2 FROM tab WHERE col1 IN (?)", (','.join(str(val) for val in vals),)).fetchall()
[]

>>> 

Теперь я могу сделать следующее, которое (я думаю..., пожалуйста, поправьте меня, если я ошибаюсь) сохраняет безопасность встроенной замены параметров, но все еще немного уродливо:

>>> db.execute("SELECT col2 FROM tab WHERE col1 IN (" + ",".join("?"*len(vals)) + ")", vals).fetchall()
[(u'one',), (u'three',), (u'four',)]
>>> 

Это мой лучший вариант, или есть ли лучший способ обойти это?

4b9b3361

Ответ 1

Это ваш лучший вариант без использования дополнительных библиотек. Я, конечно, выступал за эту технику в прошлом не раз.

Вы также можете переключиться на использование SQLAlchemy, которая генерирует SQL для вас, но для этого вам нужно подняться на свою кривую обучения и переписать большую часть вашего приложения.