SQLAlchemy过滤器在数组内嵌套jsonb

时间:2019-02-13 12:10:35

标签: python postgresql sqlalchemy jsonb

我有一个Postgres JSONB字段,其中包含一些嵌套数组和其他对象。

from sqlalchemy.dialects.postgresql import JSONB


class Family(db.Model):
   meta = db.Column(JSONB)


joes = Family(meta=[
    {
        "name": "Joe",
        "children": [
            {
                "name": "Jane"
            },
            {
                "name": "Kate"
            }
        ]
    },
    {
        "name": "Lisa",
        "children": [
            {
                "name": "Mary"
            },
            {
                "name": "David"
            }
        ]
    },
])

有没有一种方法可以查询所有名字中带有特定子字符串的孩子?

如果我想查询'a',它应该让我Mary, David, Kate, Jane

我在想,也许像

Family.query.filter(
    Family.meta.contains([{"children": [{"name": func.contains("a")}]}])
)

1 个答案:

答案 0 :(得分:1)

技巧是使用jsonb_array_elements()取消嵌套一个或多个数组,然后过滤:

meta_value = literal_column('meta.value', type_=JSONB)
children_value = literal_column('children.value', type_=JSONB)

Family.query.\
    with_entities(children_value['name'].astext).\
    select_from(
        Family,
        func.jsonb_array_elements(Family.meta).alias('meta'),
        func.jsonb_array_elements(
            meta_value['children']).alias('children')).\
    filter(children_value['name'].astext.contains('a'))

请注意使用literal_column()来引用返回的集合函数jsonb_array_elements()的值。