我想使用formencode在我的Pyramid应用程序中验证上传文件的大小。据我了解,我需要创建一个继承自formencode.validators.FormValidator)
的类并将其放到chained_validators
。但我无法找到一种方法来检查validate_python
方法中上传文件的大小。它甚至可能吗?
先谢谢你,伊万。
答案 0 :(得分:2)
你可以在文件对象本身上使用len()(在验证器中)来检查文件大小,因为它计算了字节数。
size = len(fileObj)
答案 1 :(得分:1)
当然有 - 虽然我用turbogears做过,但它也适用于金字塔:
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)
请注意,读取()整个数据不应该是必要的(参见其他答案) - 我这样做是为了进一步的内容验证。
答案 2 :(得分:0)
另一种方法:
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),
]