helpers.py 976 B

12345678910111213141516171819202122232425262728293031323334353637383940
  1. from mongoengine import ValidationError
  2. from wtforms.validators import ValidationError as wtfValidationError
  3. from flask_admin._compat import itervalues, as_unicode
  4. def make_gridfs_args(value):
  5. args = {
  6. 'id': value.grid_id,
  7. 'coll': value.collection_name
  8. }
  9. if value.db_alias != 'default':
  10. args['db'] = value.db_alias
  11. return args
  12. def make_thumb_args(value):
  13. if getattr(value, 'thumbnail', None):
  14. args = {
  15. 'id': value.thumbnail._id,
  16. 'coll': value.collection_name
  17. }
  18. if value.db_alias != 'default':
  19. args['db'] = value.db_alias
  20. return args
  21. else:
  22. return make_gridfs_args(value)
  23. def format_error(error):
  24. if isinstance(error, ValidationError):
  25. return as_unicode(error)
  26. if isinstance(error, wtfValidationError):
  27. return '. '.join(itervalues(error.to_dict()))
  28. return as_unicode(error)