SQLAlchemy WHERE IN单值(原始SQL)

时间:2014-01-31 13:42:16

标签: python mysql sqlalchemy

在执行检查多个值的原始SQL时,我遇到SQLAlchemy问题。

my_sess.execute(
        "SELECT * FROM table WHERE `key`='rating' AND uid IN :uids",
        params=dict(uids=some_list)
    ).fetchall()

此查询有两种情况,一种有效,另一种无效。如果some_list = [1],它会抛出一个SQL错误,我在)附近有语法错误。但是如果some_list = [1, 2],则查询成功执行。

为什么会发生这种情况?

1 个答案:

答案 0 :(得分:6)

不,SQL参数只处理标量值。你必须在这里生成SQL;如果您需要原始SQL,请使用:

statement = "SELECT * FROM table WHERE `key`='rating' AND uid IN ({})".format(
    ', '.join([':i{}'.format(i) for i in range(len(some_list))]))

my_sess.execute(
        statement, 
        params={'i{}'.format(i): v for i, v in enumerate(some_list)})
    ).fetchall()

e.g。生成足够的参数以使用字符串格式保存some_list中的所有值,然后生成匹配参数以填充它们。

最好还是使用literal_column() object为你做所有的生成:

from sqlalchemy.sql import literal_column

uid_in = literal_column('uid').in_(some_list)
statement = "SELECT * FROM able WHERE `key`='rating' AND {}".format(uid_in)

my_sess.execute(
        statement, 
        params={'uid_{}'.format(i): v for i, v in enumerate(some_list)})
    ).fetchall()

但是你可能只需使用`sqlalchemy.sql.expression模块生成整个语句,因为这样可以更容易地支持多个数据库方言。

此外,uid_in对象已经保存了对绑定参数的正确值的引用;而不是像上面的str.format()操作那样将它变成一个字符串,SQLAlchemy将拥有实际的对象加上相关的参数,你将不再需要生成params字典 < / em>的

以下内容应该有效:

from sqlalchemy.sql import table, literal_column, select

tbl = table('table')
key_clause = literal_column('key') == 'rating'
uid_clause = literal_column('uid').in_(some_list)
my_sess.execute(select('*', key_clause & uid_clause, [tbl]))

其中sqlalchemy.sql.select()采用列规范(此处硬编码为*),where子句(从&的两个子句生成以生成SQL {{1} }子句和可选择的列表;这里有一个sqlalchemy.sql.table()值。

快速演示:

AND

但是从这一切生成的实际对象树也包含绑定参数的实际值,因此>>> from sqlalchemy.sql import table, literal_column, select >>> some_list = ['foo', 'bar'] >>> tbl = table('table') >>> key_clause = literal_column('key') == 'rating' >>> uid_clause = literal_column('uid').in_(some_list) >>> print select('*', key_clause & uid_clause, [tbl]) SELECT * FROM "table" WHERE key = :key_1 AND uid IN (:uid_1, :uid_2) 可以直接访问这些值。