有没有办法使用python formencode验证文件大小?

时间:2012-02-15 17:00:51

标签: python pyramid formencode

我想使用formencode在我的Pyramid应用程序中验证上传文件的大小。据我了解,我需要创建一个继承自formencode.validators.FormValidator)的类并将其放到chained_validators。但我无法找到一种方法来检查validate_python方法中上传文件的大小。它甚至可能吗?

先谢谢你,伊万。

3 个答案:

答案 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),
    ]