Jak zrobić „grupa przez” w mongoengine

Question

Załóżmy mój Schemat wygląda tak:

class User(Document): 
    username = StringField() 
    password = StringField() 
    category = StringField() 

Wyobraź mamy te istniejące kategorie: "avengers", "justice-leaguers", "villains" i chcę wykonać "group by" zapytanie do User.objects.all() tak, że mogę dostać coś takiego:

[ 
[<User: IronMan object>, <User: Thor object>, <User: Hulk object>], 
[<User: Superman object>,<User: Batman object>], 
[<User: Ultron object>, <User: Joker object>, <User: LexLuthor object>] 
] 

Albo jeszcze lepiej:

{ 
"avengers": [<User: IronMan object>, <User: Thor object>, <User: Hulk object>], 
"justice-leaguers": [<User: Superman object>,<User: Batman object>], 
"villains": [<User: Ultron object>, <User: Joker object>, <User: LexLuthor object>] 
} 

Zajrzałem do dokumentacji MongoEngine i jeszcze nie znalazłem nic przydatnego. Dzięki chłopaki!

Answer 1

Używanie aggregation framework, trzeba tylko $group dokumentów według kategorii:

db.User.aggregate([ 
    { 
    $group: { _id: "$category", username: { $push: "$username" }} 
    } 
]) 

Korzystanie z funkcji $push agregacji, można zbudować tablicę zawierającą wszystkie nazwy użytkownika dzielenie tej samej kategorii.

wam dane przykładowe dane:

> db.User.find({},{category:1,username:1,_id:0}) 
{ "category" : "avengers", "username" : "IronMan" } 
{ "category" : "avengers", "username" : "Thor" } 
{ "category" : "avengers", "username" : "Hulk" } 
{ "category" : "justice-leagers", "username" : "Superman" } 
{ "category" : "justice-leagers", "username" : "Batman" } 
{ "category" : "villains", "username" : "Ultron" } 
{ "category" : "villains", "username" : "Joker" } 
{ "category" : "villains", "username" : "LexLuthor" } 

To będzie produkują:

{ "_id" : "villains", "username" : [ "Ultron", "Joker", "LexLuthor" ] } 
{ "_id" : "justice-leagers", "username" : [ "Superman", "Batman" ] } 
{ "_id" : "avengers", "username" : [ "IronMan", "Thor", "Hulk" ] } 

Answer 2

Ponieważ PO poprosił o mongoengine (i to, co potrzebne) Oto przykład tego, używając mongoengine:

categories = User.objects.aggregate([{ 
    '$group': { '_id': '$category', 'username': { '$push': '$username' }} 
}]) 

Spowoduje to wyświetlenie iteratora kursora poleceń pymongo.

print list(categories) 

Wrócimy:

[{ "_id": "villains", "username": ["Ultron", "Joker", "LexLuthor"]}, 
{ "_id": "justice-leagers", "username": ["Superman", "Batman"]}, 
{ "_id": "avengers", "username": ["IronMan", "Thor", "Hulk"]}]