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