Wie „Gruppe“ in mongoengine

Question

Angenommen, mein Schema sieht wie dies zu tun:

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

Stellen haben wir diese bestehenden Kategorien: "avengers", "justice-leaguers", "villains", und ich möchte so eine "group by" Abfrage für User.objects.all() auszuführen, dass ich bekommen kann so etwas wie dieses:

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

Oder noch besser:

{ 
"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>] 
} 

Ich habe mir die Dokumente von MongoEngine angesehen und muss noch etwas hilfreiches finden. Danke Leute!

Answer 1

die aggregation framework verwenden, können Sie nur nach Kategorie $group Dokumente benötigen:

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

Mit der $push Aggregationsfunktion finden Sie ein Array mit all den Benutzernamen teilen die gleiche Kategorie bauen.

Gegeben Sie Beispieldaten:

> 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" } 

Dies erzeugt:

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

Answer 2

Da die OP für mongoengine gefragt (und das ist, was ich brauchte) hier ist ein Beispiel für diese Verwendung mongoengine:

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

Dies wird einen Pymongo-Befehl Cursor Iterator zurückgeben.

print list(categories) 

Wir kommen wieder:

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