Commit 44554259 authored by Matteo Danieli's avatar Matteo Danieli

Only replace MongoDB Image if a new one is submitted

parent dc8d8fc3
...@@ -6,6 +6,13 @@ from wtforms.fields.core import _unset_value ...@@ -6,6 +6,13 @@ from wtforms.fields.core import _unset_value
from . import widgets from . import widgets
def is_empty(file_object):
file_object.seek(0)
first_char = file_object.read(1)
file_object.seek(0)
return not bool(first_char)
class ModelFormField(fields.FormField): class ModelFormField(fields.FormField):
""" """
Customized ModelFormField for MongoEngine EmbeddedDocuments. Customized ModelFormField for MongoEngine EmbeddedDocuments.
...@@ -54,7 +61,7 @@ class MongoFileField(fields.FileField): ...@@ -54,7 +61,7 @@ class MongoFileField(fields.FileField):
field.delete() field.delete()
return return
if isinstance(self.data, FileStorage): if isinstance(self.data, FileStorage) and not is_empty(self.data.stream):
if not field.grid_id: if not field.grid_id:
func = field.put func = field.put
else: else:
......
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