Django中实现可复用多版本图片生成类求助

我明白你想把重复的头像处理逻辑抽成可复用的类，这样就不用在多个模型的多个字段里反复写相同代码了。咱们来把你的avatar_tamper函数改造成一个灵活的可复用类，同时还能解决原代码里的一些小问题：

第一步：实现可复用的头像处理类 #

这个类允许你自定义尺寸、字段名称，方便在不同模型上复用：

import os
import sys
from io import BytesIO
from django.core.files.uploadedfile import InMemoryUploadedFile
from PIL import Image

class AvatarProcessor:
    def __init__(self, main_field='avatar', thumb_field='avatar_tn', main_height=500, thumb_height=175):
        # 自定义字段名和尺寸，默认和你原来的逻辑一致
        self.main_field = main_field
        self.thumb_field = thumb_field
        self.main_height = main_height
        self.thumb_height = thumb_height

    def process(self, instance):
        # 获取主头像字段的值
        main_avatar = getattr(instance, self.main_field)
        if not main_avatar:
            return

        # 获取缩略图字段的值
        thumb_avatar = getattr(instance, self.thumb_field)

        # 检查缩略图是否存在，如果存在就删除（当主头像更新时）
        if thumb_avatar:
            if os.path.exists(thumb_avatar.path):
                thumb_avatar.delete()
            setattr(instance, self.thumb_field, None)

        # 打开并处理图片
        image = Image.open(main_avatar)
        # 计算主头像的宽度比例
        hpercent_main = self.main_height / image.size[1]
        wsize_main = int(image.size[0] * hpercent_main)
        # 计算缩略图的宽度比例
        hpercent_thumb = self.thumb_height / image.size[1]
        wsize_thumb = int(image.size[0] * hpercent_thumb)

        # 调整尺寸
        resized_main = image.resize((wsize_main, self.main_height))
        resized_thumb = image.resize((wsize_thumb, self.thumb_height))

        # 保存到BytesIO
        main_stream = BytesIO()
        thumb_stream = BytesIO()
        resized_main.save(main_stream, format='PNG')
        resized_thumb.save(thumb_stream, format='PNG')

        # 重置流指针到开头
        main_stream.seek(0)
        thumb_stream.seek(0)

        # 生成新的InMemoryUploadedFile
        filename_base = main_avatar.name.split('.')[0]
        main_file = InMemoryUploadedFile(
            main_stream, 'ImageField', f"{filename_base}.png",
            'image/png', sys.getsizeof(main_stream), None
        )
        thumb_file = InMemoryUploadedFile(
            thumb_stream, 'ImageField', f"{filename_base}.png",
            'image/png', sys.getsizeof(thumb_stream), None
        )

        # 更新实例的字段
        setattr(instance, self.main_field, main_file)
        setattr(instance, self.thumb_field, thumb_file)

第二步：在模型中使用这个类 #

原来的save方法里两次调用super().save()有点冗余，我们可以调整逻辑，先处理头像再保存（或者处理后保存一次即可），下面是优化后的用法：

from django.db import models

class User(models.Model):
    avatar = models.ImageField(upload_to='avatars/')
    avatar_tn = models.ImageField(upload_to='avatars/thumbnails/', blank=True, null=True)

    def save(self, *args, **kwargs):
        # 实例化处理器，这里用默认配置，你也可以自定义参数
        processor = AvatarProcessor()
        # 处理头像
        processor.process(self)
        # 只需要保存一次即可
        super().save(*args, **kwargs)
        return self.user

额外说明 #

• 灵活性：如果其他模型需要不同的尺寸或者字段名，比如Profile模型里用profile_pic和profile_pic_tn，只需要这样实例化：processor = AvatarProcessor(main_field='profile_pic', thumb_field='profile_pic_tn', main_height=600, thumb_height=200)
• 逻辑优化：原代码里的条件判断有点绕，重构后先检查缩略图是否存在，如果存在就删除，然后重新生成，这样逻辑更清晰，避免出现旧缩略图残留的问题
• 注意事项：确保你的模型字段已经正确配置了upload_to路径，并且Django的媒体文件设置是正确的

内容的提问来源于stack exchange，提问作者user14389292