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

Fix last issue with binary field as primary key and skipped new test

Browse Source
This commit is contained in:
mrigal
2015-04-16 11:45:04 +02:00
committed by Matthieu Rigal
parent 3421fffa9b
commit 571a7dc42d
2 changed files with 23 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
mongoengine/document.py
View File

@@ -312,8 +312,9 @@ class Document(BaseDocument):
# In PyMongo 3.0, the save() call calls internally the _update() call
# but they forget to return the _id value passed back, therefore getting it back here
if not object_id and pymongo.version_tuple == (3, 0):
object_id = self._qs.filter(**self._object_key).first() and \
self._qs.filter(**self._object_key).first().pk
pk_as_mongo_obj = self._fields.get(self._meta['id_field']).to_mongo(self.pk)
object_id = self._qs.filter(pk=pk_as_mongo_obj).first() and \
self._qs.filter(pk=pk_as_mongo_obj).first().pk
else:
object_id = doc['_id']
updates, removals = self._delta()