Only raise FileLength validation error on files

This commit is contained in:
dandds 2019-06-10 19:26:44 -04:00
parent 62c794e7b8
commit 49332c5d6e
2 changed files with 8 additions and 2 deletions

View File

@ -1,7 +1,10 @@
from datetime import datetime
import re
from werkzeug.datastructures import FileStorage
from wtforms.validators import ValidationError, StopValidation
import pendulum
from datetime import datetime
from atst.utils.localization import translate
@ -103,7 +106,7 @@ def RequiredIf(criteria_function, message=translate("forms.validators.is_require
def FileLength(max_length=50000000, message=None):
def _file_length(_form, field):
if field.data is None:
if field.data is None or not isinstance(field.data, FileStorage):
return True
content = field.data.read()

View File

@ -100,3 +100,6 @@ class TestFileLength:
with pytest.raises(ValidationError):
validator(dummy_form, dummy_field)
dummy_field.data = "random string"
assert validator(dummy_form, dummy_field)