Added sort method to QuerySet

This commit is contained in:
Harry Marr
2009-12-18 18:41:44 +00:00
parent 44fc9096a4
commit 5e6a6aa886
3 changed files with 37 additions and 1 deletions

View File

@@ -97,6 +97,22 @@ class QuerySet(object):
"""
self._collection.remove(self._query)
def sort(self, *keys):
"""Sort the QuerySet by the keys. The order may be specified by
prepending each of the keys by a + or a -. Ascending order is assumed.
"""
key_list = []
for key in keys:
direction = pymongo.ASCENDING
if key[0] == '-':
direction = pymongo.DESCENDING
if key[0] in ('-', '+'):
key = key[1:]
key_list.append((key, direction))
self._cursor.sort(key_list)
return self
def __iter__(self):
return self