fix style and ids need to be an array

2018-05-20 14:41:25 +08:00 · 2018-05-20 14:41:25 +08:00 · 0fc55451c2
commit 0fc55451c2
parent 5c0bd8a810
1 changed files with 1 additions and 2 deletions
--- a/mongoengine/queryset/base.py
+++ b/mongoengine/queryset/base.py
@ -357,8 +357,7 @@ class BaseQuerySet(object):

        try:
            inserted_result = insert_func(raw, **write_concern)
-            ids = return_one and inserted_result.inserted_id or\
-                  inserted_result.inserted_ids
+            ids = return_one and [inserted_result.inserted_id] or inserted_result.inserted_ids
        except pymongo.errors.DuplicateKeyError as err:
            message = 'Could not save document (%s)'
            raise NotUniqueError(message % six.text_type(err))