Commit a524661d authored by Serge S. Koval's avatar Serge S. Koval

Merge pull request #217 from mostafar/master

Cast to ObjectId is not needed in pymongo backend
parents df42cd14 d6cd59d0
import logging import logging
import pymongo import pymongo
from bson.objectid import ObjectId
from flask import flash from flask import flash
from jinja2 import contextfunction from jinja2 import contextfunction
...@@ -250,7 +249,7 @@ class ModelView(BaseModelView): ...@@ -250,7 +249,7 @@ class ModelView(BaseModelView):
Model ID Model ID
""" """
# TODO: Validate if it is valid ID # TODO: Validate if it is valid ID
return self.coll.find_one({'_id': ObjectId(id)}) return self.coll.find_one({'_id': id})
def edit_form(self, obj): def edit_form(self, obj):
""" """
...@@ -337,7 +336,7 @@ class ModelView(BaseModelView): ...@@ -337,7 +336,7 @@ class ModelView(BaseModelView):
# TODO: Optimize me # TODO: Optimize me
for pk in ids: for pk in ids:
self.coll.remove({'_id': ObjectId(pk)}) self.coll.remove({'_id': pk})
count += 1 count += 1
flash(ngettext('Model was successfully deleted.', flash(ngettext('Model was successfully deleted.',
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment