python - 尝试在 Django 中创建缩略图时获取无法识别图像文件

Question

<分区>

Answer 1

我看到的问题是您设计Thumbnail 类的方式。它使用类属性来存储实例变量，这意味着当您尝试多次使用该类时会发生冲突。

不需要静态 load 方法，因为一旦您将属性移动到实例，它就会做与类的构造函数完全相同的事情。通过在构造函数中要求 source，您可以确保 generate 稍后在查找空字符串值时不会发生崩溃。

此外，我认为您面临的主要问题之一是，当您使用 Django 模型为 ImageField 返回的类文件对象包装器时。如果您传入字符串路径，您将看不到这一点，但当您传入文件对象时，generate 方法会将其读取到最后。然后您使用相同的源对象第二次调用 generate，但它在最后并且您得到一个 IOError。现在，一种方法是确保在使用它再次调用 Thumbnail 之前将源搜索回 0，但是您可以省去麻烦，只需拥有您的 Thumbnail 类在构造函数中打开并缓存 PIL 图像一次。然后generate就不用每次都不断地重新读取了。

# Example from your code #
def generate(self, size=None, fit=True):
    ...
    # The first time you do this, it will read
    # self.source to the end, because in Django, you
    # are passing a file-like object.
    image = Image.open(self.source)

# this will work the first time
generate()
# uh oh. self.source was a file object that is at the end
generate() # crash

重写缩略图类

from cStringIO import StringIO
from PIL import Image

class Thumbnail(object):

    SIZE = (50, 50)

    def __init__(self, source):
        self.source = source
        self.output = None

        self._pilImage = Image.open(self.source)

    def generate(self, size=None, fit=True):
        if not size:
            size = self.SIZE

        if not isinstance(size, tuple):
            raise TypeError('Thumbnail class: The size parameter must be an instance of a tuple.')

        # resize properties
        box = size
        factor = 1
        image = self._pilImage.copy()

        # Convert to RGB if necessary
        if image.mode not in ('L', 'RGB'): 
            image = image.convert('RGB')
        while image.size[0]/factor > 2*box[0] and image.size[1]*2/factor > 2*box[1]:
            factor *=2
        if factor > 1:
            image.thumbnail((image.size[0]/factor, image.size[1]/factor), Image.NEAREST)

        #calculate the cropping box and get the cropped part
        if fit:
            x1 = y1 = 0
            x2, y2 = image.size
            wRatio = 1.0 * x2/box[0]
            hRatio = 1.0 * y2/box[1]
            if hRatio > wRatio:
                y1 = int(y2/2-box[1]*wRatio/2)
                y2 = int(y2/2+box[1]*wRatio/2)
            else:
                x1 = int(x2/2-box[0]*hRatio/2)
                x2 = int(x2/2+box[0]*hRatio/2)
            image = image.crop((x1,y1,x2,y2))

        #Resize the image with best quality algorithm ANTI-ALIAS
        image.thumbnail(box, Image.ANTIALIAS)

        # save image to memory
        temp_handle = StringIO()
        image.save(temp_handle, 'png')
        temp_handle.seek(0)

        self.output = temp_handle

        return self

    def get_output(self):
        self.output.seek(0)
        return self.output.read()

用法:缩略图(src).generate((200, 200)).get_output()

source 和output 对于每个实例都必须是唯一的。但是在您的版本中，您会将 output 设置为类级别，这意味着 Thumbnail 的两个实例使用共享的最新版本的 output .

# your code #
    # this is assigning the most recently processed
    # object to the class level. shared among all.
    self.output = temp_handle

    return self

def get_output(self):
    # always read the shared class level
    return self.output.read()

此外，我觉得有一种更简单的方法来执行调整大小/适合/裁剪。如果您解释要对图像执行的确切转换，我也可以简化它。

更新

我忘了特别提到，根据我对一次保存源图像的建议，您的用法应该如下所示:

def save(self, commit=True):
    instance = super(ImageForm, self).save(commit=True)

    thumb = Thumbnail(instance.src)

    instance.r128 = SimpleUploadedFile(
        instance.src.name,
        thumb.generate((128, 128)).get_output(),
        content_type='image/png'
    )

    instance.r200 = SimpleUploadedFile(
        instance.src.name,
        thumb.generate((200, 200)).get_output(),
        content_type='image/png'
    )

请注意，我们只使用源代码创建了一个 Thumbnail 实例，它只会在 PIL 中打开一次。然后您可以从中生成任意数量的图像。