Question

我想在imagefield上传有限扩展名的文件。那么，我怎样才能验证jpg，bmp和gif的图像字段。此外，默认情况下图像字段会占用哪个扩展名？

Answer 1

我就是这样做的：

from django.utils.image import Image

# settings.py
# ALLOWED_UPLOAD_IMAGES = ('gif', 'bmp', 'jpeg')


class ImageForm(forms.Form):

    image = forms.ImageField()

    def clean_image(self):
        image = self.cleaned_data["image"]
        # This won't raise an exception since it was validated by ImageField.
        im = Image.open(image)

        if im.format.lower() not in settings.ALLOWED_UPLOAD_IMAGES:
            raise forms.ValidationError(_("Unsupported file format. Supported formats are %s."
                                          % ", ".join(settings.ALLOWED_UPLOAD_IMAGES)))

        image.seek(0)
        return image

也适用于ModelForm。

单元测试：

from StringIO import StringIO

from django.core.files.uploadedfile import SimpleUploadedFile
from django.test.utils import override_settings
from django.test import TestCase


class ImageFormTest(TestCase):

    def test_image_upload(self):
        """
        Image upload
        """
        content = 'GIF87a\x01\x00\x01\x00\x80\x01\x00\x00\x00\x00ccc,\x00' \
                  '\x00\x00\x00\x01\x00\x01\x00\x00\x02\x02D\x01\x00;'
        image = StringIO(content)
        image.name = 'image.gif'
        image.content_type = 'image/gif'
        files = {'image': SimpleUploadedFile(image.name, image.read()), }

        form = ImageForm(data={}, files=files)
        self.assertTrue(form.is_valid())


    @override_settings(ALLOWED_UPLOAD_IMAGES=['png', ])
    def test_image_upload_not_allowed_format(self):
        image = StringIO('GIF87a\x01\x00\x01\x00\x80\x01\x00\x00\x00\x00ccc,\x00'
                         '\x00\x00\x00\x01\x00\x01\x00\x00\x02\x02D\x01\x00;')
        image.name = 'image'
        files = {'image': SimpleUploadedFile(image.name, image.read()), }
        form = ImageForm(data={}, files=files)
        self.assertFalse(form.is_valid())

Pillow将允许一堆image formats

在django中，哪个格式的models.ImageField默认采用

1 个答案: