Commit d6cd59d0 authored by Mostafa Rokooie's avatar Mostafa Rokooie

Remove ObjectId cast of _id in get and save.

parent df42cd14
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