Sqlalchemy查询dict列表

时间:2017-08-07 05:47:51

标签: python json list dictionary sqlalchemy

我正在尝试运行一个根据dict列表过滤结果的查询。

让我们更清楚......

我有一个表格标题,每个标题都有一个版本字段

class Heading(Base):
    id = Column(Integer, primary_key=True)
    name = Column(String(50), nullable=False)
    version = Column(Integer)

另一方面,我有一个看起来像这样的dict列表(Json格式化):

"headings": [
    {
        "id": 1,
        "version": 1
    },
    {
        "id": 2,
        "version": 1
    }
]

我想在Heading表上执行一个查询,该查询排除数据库中的版本等于(json)列表中的版本的行。

我尝试执行这样的事情:

res = Heading.query.filter(~Heading.id.in_([d['id'] for d in headings if d['version'] == Heading.version])).all()

数据库看起来像这样(Json再次形成,抱歉):

[
{
    "id": 1,
    "name": "Heading1",
    "version": 2,
},
{
    "id": 2,
    "name": "Heading2",
    "version": 1,
}
]

但它仍然会返回Heading2行。基本上,我希望查询返回Json中未定义的所有行以及DB中具有较新版本的行。

我不知道我的问题是否足够明确。 也许这不是正确的方法,请让我知道

提前感谢您提供任何帮助

修改 生成的Raw SQL是这样的:

SELECT heading.id, heading.name, heading.version
FROM heading
WHERE heading.id = heading.id

因此sqlalchemy查询存在明显问题,但我无法弄清楚...

我认为它来自

部分
if d['version'] == Heading.version

但我不知道如何在查询的其他地方重新制定这种情况

1 个答案:

答案 0 :(得分:3)

您怀疑问题出在

[... for ... if d['version'] == Heading.version]

虽然d['version'] == Heading.version生成了一个SQL表达式对象,但Python然后使用其真值进行过滤:

In [9]: bool(headings[0]['version'] == Heading.version)
Out[9]: False

所以你的列表理解会产生一个空列表。 Heading.id永远不会出现在空列表中,所以

~Heading.id.in_([])

只会评估为true:

In [10]: print(~Heading.id.in_([]))
1 = 1

鉴于您的数据库支持composite IN queries,解决方案是与(id,version)元组进行比较:

In [9]: from sqlalchemy import tuple_

In [10]: session.query(Heading).\
    ...:     filter(~tuple_(Heading.id, Heading.version).in_(
    ...:         [(d['id'], d['version']) for d in headings])).\
    ...:     all()
Out[10]: [<__main__.Heading at 0x7f840cc8ae48>]

In [11]: _[0].name
Out[11]: 'Heading1'