Fixed possible issue not catching duplicate key errors

2014-01-24 13:50:09 +00:00 · 2014-01-24 13:50:09 +00:00 · 0e5a0661e1
commit 0e5a0661e1
parent a839bd428f
1 changed files with 4 additions and 1 deletions
--- a/mongoengine/queryset/base.py
+++ b/mongoengine/queryset/base.py
@ -302,8 +302,11 @@ class BaseQuerySet(object):
        signals.pre_bulk_insert.send(self._document, documents=docs)
        try:
            ids = self._collection.insert(raw, **write_concern)
+        except pymongo.errors.DuplicateKeyError, err:
+            message = 'Could not save document (%s)';
+            raise NotUniqueError(message % unicode(err))
        except pymongo.errors.OperationFailure, err:
-            message = 'Could not save document (%s)'
+            message = 'Could not save document (%s)';
            if re.match('^E1100[01] duplicate key', unicode(err)):
                # E11000 - duplicate key error index
                # E11001 - duplicate key on update