fix style and ids need to be an array

This commit is contained in:
Erdenezul Batmunkh 2018-05-20 14:41:25 +08:00
parent eb2e106871
commit 1d3f20b666

View File

@ -357,8 +357,7 @@ class BaseQuerySet(object):
try: try:
inserted_result = insert_func(raw, **write_concern) inserted_result = insert_func(raw, **write_concern)
ids = return_one and inserted_result.inserted_id or\ ids = return_one and [inserted_result.inserted_id] or inserted_result.inserted_ids
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))