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

fix inserted_ids

Browse Source
This commit is contained in:
Erdenezul Batmunkh 2018-05-20 14:33:12 +08:00
parent 3e0d84383e
commit f9a887c8c6
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mongoengine/queryset/base.py
View File

@ -357,7 +357,8 @@ class BaseQuerySet(object):
try: try:
inserted_result = insert_func(raw, **write_concern) inserted_result = insert_func(raw, **write_concern)
ids = inserted_result.inserted_id if return_one else inserted_result.inserted_ids ids = return_one and inserted_result.inserted_id or\
inserted_result.inserted_ids
except pymongo.errors.DuplicateKeyError as err: except pymongo.errors.DuplicateKeyError as err:
message = 'Could not save document (%s)' message = 'Could not save document (%s)'
raise NotUniqueError(message % six.text_type(err)) raise NotUniqueError(message % six.text_type(err))
@ -374,7 +375,6 @@ class BaseQuerySet(object):
signals.post_bulk_insert.send( signals.post_bulk_insert.send(
self._document, documents=docs, loaded=False, **signal_kwargs) self._document, documents=docs, loaded=False, **signal_kwargs)
return return_one and ids[0] or ids return return_one and ids[0] or ids
documents = self.in_bulk(ids) documents = self.in_bulk(ids)
results = [] results = []
for obj_id in ids: for obj_id in ids: