重写 Django 模型 save() 方法：轻松实现图片缩放与格式转换

python

2024-03-19 09:10:18

利用 Django 模型重写 save() 方法实现图片缩放与格式转换

简介

随着现代网络应用对图片处理需求的不断增长，优化图像处理流程变得至关重要。本文将深入探讨如何在 Django 模型中重写 save() 方法，从而在图片上传时自动对其进行缩放和格式转换。

问题

在许多场景下，开发者需要在用户上传图片时对其进行处理，比如缩放或格式转换。传统上，这些操作是在视图函数中实现的。然而，这会导致不必要的带宽消耗，因为未处理的原始图片也会被上传到服务器。

解决方案：重写模型 save() 方法

为了解决这一问题，我们可以重写模型的 save() 方法。这样一来，图片的处理操作将在数据库保存之前完成，从而避免了对未处理图片的上传。

步骤详解

以下是重写模型 save() 方法的详细步骤：

调用父类方法保存图片： 首先，调用父类的 save() 方法，将原始图片保存到数据库。
打开图片文件： 使用 Image.open() 方法打开保存的图片文件。
缩放图片： 使用 image.resize() 方法将图片缩放为所需的尺寸。
格式转换： 使用 image.save() 方法将图片转换为所需的格式，如 JPEG。
覆盖原图片： 将转换后的图片覆盖原图片，从而完成图片处理操作。

代码示例

from PIL import Image
from io import BytesIO

def save(self, *args, **kwargs):
    # 调用父类方法保存图片
    super().save(*args, **kwargs)

    # 打开图片文件
    image = Image.open(self.image.path)

    # 缩放图片
    image = image.resize((128, 128))

    # 将图片转换成 JPEG 格式
    output = BytesIO()
    image.save(output, format="JPEG")
    output.seek(0)

    # 覆盖原图片
    self.image.save(self.image.name, output, save=False)