| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465 |
- from werkzeug.datastructures import FileStorage
- from wtforms import fields
- from ...upload import BS3FileUploadFieldWidget, BS3ImageUploadFieldWidget
- try:
- from wtforms.fields.core import _unset_value as unset_value
- except ImportError:
- from wtforms.utils import unset_value
- 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 MongoFileField(fields.FileField):
- """
- GridFS file field.
- """
- widget = BS3FileUploadFieldWidget()
- def __init__(self, label=None, validators=None, **kwargs):
- super().__init__(label, validators, **kwargs)
- self._should_delete = False
- def process(self, formdata, data=unset_value, **kwargs):
- if formdata:
- marker = "_%s-delete" % self.name
- if marker in formdata:
- self._should_delete = True
- return super().process(formdata, data, **kwargs)
- def populate_obj(self, obj, name):
- field = getattr(obj, name, None)
- if field is not None:
- # If field should be deleted, clean it up
- if self._should_delete:
- field.delete()
- return
- if isinstance(self.data, FileStorage) and not is_empty(self.data.stream):
- if not field.grid_id:
- func = field.put
- else:
- func = field.replace
- func(
- self.data.stream,
- filename=self.data.filename,
- content_type=self.data.content_type,
- )
- class MongoImageField(MongoFileField):
- """
- GridFS file field.
- """
- widget = BS3ImageUploadFieldWidget()
|
