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

Use info from getlasterror whether a document has been updated or created.

Browse Source
This commit is contained in:
Manuel Hermann 2012-08-07 17:04:03 +02:00
parent 12c8b5c0b9
commit 500eb920e4
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
mongoengine/document.py
View File

@ -208,11 +208,20 @@ class Document(BaseDocument):
actual_key = self._db_field_map.get(k, k)
select_dict[actual_key] = doc[actual_key]
def is_new_object(last_error):
if last_error is not None:
updated = last_error.get("updatedExisting")
if updated is not None:
return not updated
return created
upsert = self._created
if updates:
collection.update(select_dict, {"$set": updates}, upsert=upsert, safe=safe, **write_options)
last_error = collection.update(select_dict, {"$set": updates}, upsert=upsert, safe=safe, **write_options)
created = is_new_object(last_error)
if removals:
collection.update(select_dict, {"$unset": removals}, upsert=upsert, safe=safe, **write_options)
last_error = collection.update(select_dict, {"$unset": removals}, upsert=upsert, safe=safe, **write_options)
created = created or is_new_object(last_error)
cascade = self._meta.get('cascade', True) if cascade is None else cascade
if cascade: