如何使用SqlAlchemy在Postgres中查询JSON数组?

时间:2018-09-05 16:54:53

标签: python python-3.x postgresql sqlalchemy flask-sqlalchemy

我定义了一个SqlAlchemy模型

from sqlalchemy.dialects.postgresql import JSONB

class User(db.Model):
    __tablename__ = "user"
    id = db.Column(db.Integer, primary_key=True)
    nickname = db.Column(db.String(255), nullable=False)
    city = db.Column(db.String(255))
    contact_list = db.Column(JSONB)
    created_at = db.Column(db.DateTime, default=datetime.utcnow)

def add_user():
    user = User(nickname="Mike")
    user.contact_list = [{"name": "Sam", "phone": ["123456", "654321"]}, 
                         {"name": "John", "phone": ["159753"]},
                         {"name": "Joe", "phone": ["147889", "98741"]}]
    db.session.add(user)
    db.session.commit()

if __name__ == "__main__":
    add_user()

如何使用电话从我的联系人列表中检索名称?例如,我有 147889 ,如何检索 Joe

我已经尝试过了

User.query.filter(User.contact_list.contains({"phone": ["147889"]})).all()

但是,它返回了一个空列表[]

我该怎么做?

1 个答案:

答案 0 :(得分:3)

您只是忘记了JSON路径也应该包含最外面的数组:

User.query.filter(User.contact_list.contains([{"phone": ["147889"]}])).all()

将返回您要查找的用户。如果您的JSON包含带有键“ phone”等的对象,则原始查询将匹配。请注意,这将返回所讨论的User对象,而不是JSON结构中的特定对象/名称。如果您希望这样做(这似乎是最终目标),则可以扩展每个用户的数组元素,根据结果记录进行过滤,然后选择名称:

val = db.column('value', type_=JSONB)
db.session.query(val['name'].astext).\
    select_from(User,
                db.func.jsonb_array_elements(User.contact_list).alias()).\
    filter(val.contains({"phone": ["147889"]})).\
    all()

另一方面,上面的查询并不像第一个查询那样对索引友好,因为它必须在过滤之前扩展所有数组,因此最好先查找联系人中包含电话的用户在子查询或CTE中列出,然后展开并过滤。