When using autogenerated document ids in a sharded collection, do set that id back into the Document

2014-05-05 16:39:55 +02:00
parent f099dc6a37
commit babbc8bcd6
2 changed files with 5 additions and 1 deletions
--- a/mongoengine/document.py
+++ b/mongoengine/document.py
@@ -293,7 +293,7 @@ class Document(BaseDocument):
                raise NotUniqueError(message % unicode(err))
            raise OperationError(message % unicode(err))
        id_field = self._meta['id_field']
-        if id_field not in self._meta.get('shard_key', []):
+        if created or id_field not in self._meta.get('shard_key', []):
            self[id_field] = self._fields[id_field].to_python(object_id)

        self._clear_changed_fields()