Как запросить массив JSON в Postgres с помощью SqlAlchemy? - PullRequest
Новая
       8

Как запросить массив JSON в Postgres с помощью SqlAlchemy?

0 голосов
/ 05 сентября 2018

У меня определена модель 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 , как я могу получить Джо ?

Я пробовал это

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

Но мне возвращается пустой список, []

Как я могу это сделать?

1 Ответ

0 голосов
/ 05 сентября 2018

Вы просто забыли, что ваш JSON-путь должен включать и самый внешний массив: 

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

вернет искомого пользователя. Исходный запрос соответствовал бы, если бы ваш JSON содержал объект с ключом «телефон» и т. Д. Обратите внимание, что это возвращает рассматриваемый объект 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, а затем разверните и отфильтруйте.

...