pub/mongoengine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added get_or_create-method

Browse Source
This commit is contained in:
Florian Schlachter 2010-01-30 21:12:46 +01:00
parent 5cac52720c
commit 8d953f0bcb
1 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
mongoengine/queryset.py
View File

@ -10,6 +10,9 @@ __all__ = ['queryset_manager', 'Q', 'InvalidQueryError',
# The maximum number of items to display in a QuerySet.__repr__ # The maximum number of items to display in a QuerySet.__repr__
REPR_OUTPUT_SIZE = 20 REPR_OUTPUT_SIZE = 20
class MultipleObjectsReturned(Exception):
pass
class InvalidQueryError(Exception): class InvalidQueryError(Exception):
pass pass
@ -292,6 +295,23 @@ class QuerySet(object):
return mongo_query return mongo_query
def get_or_create(self, **kwargs):
"""Retreive unique object or create with paras, if it doesn't exist
"""
dataset = self.filter(**kwargs)
cnt = dataset.count()
if cnt == 0:
if kwargs.has_key('defaults'):
kwargs.update(kwargs.get('defaults'))
del kwargs['defaults']
doc = self._document(**kwargs)
doc.save()
return doc
elif cnt == 1:
return dataset.first()
else:
raise MultipleObjectsReturned(u'%d items returned, instead of 1' % cnt)
def first(self): def first(self):
"""Retrieve the first object matching the query. """Retrieve the first object matching the query.
""" """