Fix duplicate error check

2014-01-14 10:01:39 +00:00 · 2014-01-14 10:01:39 +00:00 · 1b62dd5c40
commit 1b62dd5c40
parent 4911545843
1 changed files with 3 additions and 1 deletions
--- a/mongoengine/document.py
+++ b/mongoengine/document.py
@ -280,7 +280,9 @@ class Document(BaseDocument):
                    kwargs.update(cascade_kwargs)
                kwargs['_refs'] = _refs
                self.cascade_save(**kwargs)
-
+        except pymongo.errors.DuplicateKeyError, err:
+            message = u'Tried to save duplicate unique keys (%s)'
+            raise NotUniqueError(message % unicode(err))
        except pymongo.errors.OperationFailure, err:
            message = 'Could not save document (%s)'
            if re.match('^E1100[01] duplicate key', unicode(err)):