Commit 1ce20ca9 authored by Serge S. Koval's avatar Serge S. Koval

Check if value is really grid fs proxy object. Related to #263

parent e26e4185
...@@ -11,7 +11,7 @@ def make_gridfs_args(value): ...@@ -11,7 +11,7 @@ def make_gridfs_args(value):
def make_thumb_args(value): def make_thumb_args(value):
if value.thumbnail: if getattr(value, 'thumbnail', None):
args = { args = {
'id': value.thumbnail._id, 'id': value.thumbnail._id,
'coll': value.collection_name 'coll': value.collection_name
......
...@@ -3,6 +3,8 @@ from wtforms.widgets import HTMLString, html_params ...@@ -3,6 +3,8 @@ from wtforms.widgets import HTMLString, html_params
from jinja2 import escape from jinja2 import escape
from flask import url_for from flask import url_for
from mongoengine.fields import GridFSProxy, ImageGridFsProxy
from . import helpers from . import helpers
...@@ -16,7 +18,7 @@ class MongoFileInput(object): ...@@ -16,7 +18,7 @@ class MongoFileInput(object):
kwargs.setdefault('id', field.id) kwargs.setdefault('id', field.id)
placeholder = '' placeholder = ''
if field.data: if field.data and isinstance(field.data, GridFSProxy):
data = field.data data = field.data
placeholder = self.template % { placeholder = self.template % {
...@@ -41,7 +43,7 @@ class MongoImageInput(object): ...@@ -41,7 +43,7 @@ class MongoImageInput(object):
kwargs.setdefault('id', field.id) kwargs.setdefault('id', field.id)
placeholder = '' placeholder = ''
if field.data: if field.data and isinstance(field.data, ImageGridFsProxy):
args = helpers.make_thumb_args(field.data) args = helpers.make_thumb_args(field.data)
placeholder = self.template % { placeholder = self.template % {
'thumb': url_for('.api_file_view', **args) 'thumb': url_for('.api_file_view', **args)
......
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