diff --git a/keg_auth/views.py b/keg_auth/views.py index fe15819..a803588 100644 --- a/keg_auth/views.py +++ b/keg_auth/views.py @@ -193,6 +193,9 @@ def add_edit(self, meth, obj=None): form=form ) + def fetch_orm_obj(self, obj_id): + return self.orm_cls.query.get(obj_id) + def init_object(self, obj_id, action=None): """Load record from ORM for edit/delete cases. @@ -204,7 +207,7 @@ def init_object(self, obj_id, action=None): """ if obj_id is None: flask.abort(400) - self.objinst = self.orm_cls.query.get(obj_id) + self.objinst = self.fetch_orm_obj(obj_id) if not self.objinst: flask.abort(404) if action: