MongoAlchemy запрос вложенных документов

Question

Я хочу знать, как использовать MongoAlchemy о встраиваемой операции с документами. Но я не нашел никаких документов об этом. Кто-нибудь может дать мне помощь?

Вот демонстрационный код:

#!/usr/bin/python
# -*- coding: utf-8 -*-

from flask import Flask
from flaskext.mongoalchemy import MongoAlchemy

app = Flask(__name__)
app.config['DEBUG'] = True
app.config['MONGOALCHEMY_DATABASE'] = 'book'
db = MongoAlchemy(app)

class Comment(db.Document):
    user_id = db.StringField(db_field='uid')
    posted = db.StringField(db_field='posted')

class Book(db.Document):
    title = db.StringField()
    author = db.StringField()

    comments = db.ListField(db.DocumentField(Comment), db_field='Comments')

from mongoalchemy.session import Session
def test():
    with Session.connect('book') as s:
        s.clear_collection(Book)

    save()
    test_Book()

def save():
    title = "Hello World"
    author = 'me'

    comment_a = Comment(user_id='user_a', posted='post_a')
    comment_b = Comment(user_id='user_b', posted='post_b')
    comments = [comment_a, comment_b]

    book = Book(title=title, author=author, comments=comments)
    book.save()

def test_Book():
    book = Book.query.filter({'author':'me'}).first()
    comment = book.comments[0]
    comment.posted = str(book.comments[0].posted)+'_new'
    book.save()
    print 'change posted: Book.comments[0].posted:', book.comments[0].posted

    comment_c = Comment(user_id='user_c', posted='post_c')
    book.comments.append(comment_c)
    book.save()
    print 'append: Book.comments[2].posted:', book.comments[2].posted

    query = Book.query.filter({Book.comments:{'$elemMatch':{Comment.user_id:'user_c'}}}).limit(1).first()
    print 'query type:', type(query)

if __name__ == '__main__':
    test()

1. Я хочу запросить данные, user_id которых равен "user_c", и просто вернуть обратно один комментарий. Как я могу это сделать? Являются ли эти методы ниже MongoAlchemy рекомендуется? Кстати, эти методы вернут весь документ.

   #query = Book.query.filter({Book.comments:{'uid':'user_c'}}).limit(1).first()
   #query = Book.query_class(Comment).filter(Comment.user_id == 'user_c').limit(1).first()
   #query = Book.query.filter({'comments':{'$elemMatch':{'uid':'user_c'}}}).limit(1).first()
   #query = Book.query.filter({Book.comments:{'$elemMatch':{Comment.user_id:'user_c'}}}).limit(1).first()
   

2. Как я могу изменить "user_c" на "user_c_new", который найти по запросу?

3. Как я могу удалить один комментарий, user_id которого является "user_b"?

Answer 1

Mongo не поддерживает возврат вложенных документов. Вы можете использовать $elemMatch для фильтрации, так что возвращаются только документы с соответствующими атрибутами, но вы должны будете сами получить комментарии. Вы можете немного оптимизировать, только возвращая поле комментариев следующим образом:

query = Book.query.filter({Book.comments:{'$elemMatch':{Comment.user_id:'user_c'}}})
query = query.fields(Book.comments.elem_match({Comment.user_id:'user_c'}))
result = query.limit(1).first()
print 'query result:', result.comments

Обратите внимание, что до 0.14.3 была ошибка с этой ошибкой (которую я только что выпустил несколько минут назад), из-за которой results.comments не работал.

Еще одно очень важное замечание, что elem_match, который я там делаю, возвращает только первый соответствующий элемент. Если вам нужны все подходящие элементы, вы должны отфильтровать их самостоятельно:

query = Book.query.filter({Book.comments:{'$elemMatch':{Comment.user_id:'user_c'}}})
result = query.limit(1).first()
print 'query result:', [c for c in result.comments if c.user_id == 'user_c']