Fixed #566. PyMongo fix

2a5b7db2 · Serge S. Koval · 2e36143a · 2a5b7db2
Commit 2a5b7db2 authored Jun 09, 2014 by Serge S. Koval
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

view.py flask_admin/contrib/pymongo/view.py +2 -2

No files found.
--- a/flask_admin/contrib/pymongo/view.py
+++ b/flask_admin/contrib/pymongo/view.py
@@ -334,8 +334,8 @@ class ModelView(BaseModelView):

            # TODO: Optimize me
            for pk in ids:
-                self.coll.remove({'_id': self._get_valid_id(pk)})
-                count += 1
+                if self.delete_model(self.get_one(pk)):
+                    count += 1

            flash(ngettext('Model was successfully deleted.',
                           '%(count)s models were successfully deleted.',