将生成的 PIL 图像保存到 django 中的 ImageField 中

Question

我正在使用qrcode 生成二维码。当买了票，或者当买的是真的时，我想生成一个二维码图像并使用 PIL 进行一些更改。最后将修改后的画布保存到模型的Image字段中。

class Ticket(models.Model):
    booked_at = models.DateTimeField(default=timezone.now)
    qrcode_file = models.ImageField(upload_to='qrcode', blank=True, null=True)
    bought = models.BooleanField(default=False)

    def save(self, *args, **kwargs):
        if self.bought:
            ...
            ...
            qrcode_img = qrcode.make('some data')
            canvas = Image.new('RGB', (total_width, total_height), 'white')
            draw = ImageDraw.Draw(canvas)
            position = (left, top)
            canvas.paste(qrcode_img, position)

            self.qrcode_file = canvas
            self.booked_at = timezone.now()
            super(Ticket, self).save(*args, **kwargs)
            canvas.close()
            qrcode_img.close()
        else:
            self.booked_at = timezone.now()
            super(Ticket, self).save(*args, **kwargs)

但这会引发错误：

AttributeError: 'Image' 对象没有属性 '_committed'

如何将生成的 PIL 图像保存到 django 中的 ImageField 中？

Answer 1

您可以使用 BytesIO 将 Pillow 文件保存到内存中的 blob。然后创建一个File 对象并将其传递给您的模型实例ImageField 的save 方法。

from io import BytesIO
from django.core.files import File

canvas = Image.new('RGB', (total_width, total_height), 'white')
...
blob = BytesIO()
canvas.save(blob, 'JPEG')  
self.qrcode_file.save('ticket-filename.jpg', File(blob), save=False) 

查看File 对象的django 文档。 https://docs.djangoproject.com/en/2.0/ref/files/file/#the-file-object

您必须使用save=False，因为默认的save=True 意味着在保存图像后会调用父模型的save 方法。你不想在这里递归，因为你通常会陷入无限循环。

Answer 2

更改您的代码并使用 Django File，如下所示：

from django.core.files import File


class Ticket(models.Model):
    booked_at = models.DateTimeField(default=timezone.now)
    qrcode_file = models.ImageField(upload_to='qrcode', blank=True, null=True)
    bought = models.BooleanField(default=False)

    def save(self, *args, **kwargs):
        if self.bought:
            ...
            ...
            qrcode_img = qrcode.make('some data')
            canvas = Image.new('RGB', (total_width, total_height), 'white')
            draw = ImageDraw.Draw(canvas)
            position = (left, top)
            canvas.paste(qrcode_img, position)

            canvas.save('path/of/dest.png', 'PNG')
            destination_file = open('path/of/dest.png', 'rb')
            self.qrcode_file.save('dest.png', File(destination_file), save=False)
            destination_file.close()

            self.booked_at = timezone.now()
            super(Ticket, self).save(*args, **kwargs)
            canvas.close()
            qrcode_img.close()
        else:
            self.booked_at = timezone.now()
            super(Ticket, self).save(*args, **kwargs)

您可以将canvas 保存在media_root 和upload_to 路径中，或保存在临时目录中或使用BytesIO 对象。