Raise NotUniqueError in Document.update() on pymongo.errors.DuplicateKeyError

2014-04-09 14:25:53 +04:00
parent d4b3649640
commit 803caddbd4
1 changed files with 2 additions and 0 deletions
--- a/mongoengine/queryset/base.py
+++ b/mongoengine/queryset/base.py
@@ -443,6 +443,8 @@ class BaseQuerySet(object):
                return result
            elif result:
                return result['n']
+        except pymongo.errors.DuplicateKeyError, err:
+            raise NotUniqueError(u'Update failed (%s)' % unicode(err))
        except pymongo.errors.OperationFailure, err:
            if unicode(err) == u'multi not coded yet':
                message = u'update() method requires MongoDB 1.1.3+'