use insert_one, insert_many and remove deprecated one #1491

This commit is contained in:
Erdenezul Batmunkh
2018-05-20 13:41:20 +08:00
parent d245ea3eaa
commit 1aebc95145
2 changed files with 8 additions and 1 deletions

View File

@@ -350,8 +350,14 @@ class BaseQuerySet(object):
documents=docs, **signal_kwargs)
raw = [doc.to_mongo() for doc in docs]
insert_func = self._collection.insert_many
if return_one:
raw = raw[0]
insert_func = self._collection.insert_one
try:
ids = self._collection.insert(raw, **write_concern)
inserted_result = insert_func(raw, **write_concern)
ids = inserted_result.inserted_id if return_one else inserted_result.inserted_ids
except pymongo.errors.DuplicateKeyError as err:
message = 'Could not save document (%s)'
raise NotUniqueError(message % six.text_type(err))