我想使用 formencode 在我的 Pyramid 应用程序中验证上传文件的大小。据我了解,我需要创建一个继承自的类formencode.validators.FormValidator)
并将其放入chained_validators
. 但是我想不出一种方法来检查方法中上传文件的大小validate_python
。甚至可能吗?
在此先感谢,伊万。
我想使用 formencode 在我的 Pyramid 应用程序中验证上传文件的大小。据我了解,我需要创建一个继承自的类formencode.validators.FormValidator)
并将其放入chained_validators
. 但是我想不出一种方法来检查方法中上传文件的大小validate_python
。甚至可能吗?
在此先感谢,伊万。
您可以在文件对象本身上使用 len() (在验证器中)来检查文件大小,因为它会计算字节数。
size = len(fileObj)
当然有——虽然我是用涡轮齿轮做的,但它也应该适用于金字塔:
class MyFileValidator(FancyValidator):
def _to_python(self, value, state):
max_size = 10*1024*1024
payload = value.file.read(max_size+1)
# rewind so that the application can access the content
value.file.seek(0)
if len(payload) == max_size:
raise Invalid(u"The file is too big (>10MB)", value, state)
return value
class MySchema(Schema):
my_file = MyFileValidator(not_empty=True)
请注意,不需要读取整个数据(请参阅其他答案) - 我这样做是为了进一步验证内容。
另一种方法:
class CheckFileSize(formencode.validators.FormValidator):
__unpackargs__ = ('upload_field', 'max_file_size')
def validate_python(self, value_dict, state):
log.info('test')
if value_dict.get(self.upload_field) is None:
return value_dict
fileobj = getattr(value_dict.get(self.upload_field), 'file', None)
fileobj.seek(0, os.SEEK_END)
if int(fileobj.tell()) > int(self.max_file_size):
raise formencode.Invalid(
_('File too big'),
value_dict, state,
error_dict={self.upload_field:
formencode.Invalid(_('File too big'), value_dict, state)})
return value_dict
class CreateNewCaseForm(formencode.Schema):
...
chained_validators = [
CheckFileSize('file', max_upload_size),
]