Django 模型中文件字段的预处理与保存最佳实践

在 django 中，直接在 `save()` 方法中访问未保存文件的本地路径会导致 filenotfounderror；正确做法是读取 `filefield` 的字节流进行内存处理，再写回或生成新文件，避免依赖尚未创建的磁盘路径。

Django 的 FileField 在模型实例保存前并不会将上传文件写入磁盘——它仅在调用 super().save() 时才触发文件存储（如 FileSystemStorage 的 save()），因此你在 save() 中尝试通过 f"media/upload/{self.filename}" 构造路径并用 Image.open() 打开，必然失败：该路径此时根本不存在。

✅ 正确思路是：绕过文件系统路径，直接操作文件内容字节流。self.file 是一个类文件对象（InMemoryUploadedFile 或 TemporaryUploadedFile），支持 .read()、.seek(0) 等操作。你应在内存中完成图像处理（如缩放、格式转换、ThumbHash 生成等），再决定如何持久化结果。

以下是重构后的 Media.save() 方法示例（含关键修复与健壮性增强）：

import os
from io import BytesIO
from PIL import Image
from django.core.files.base import ContentFile
from django.conf import settings

class Media(models.Model):
    title = models.CharField(max_length=255, null=True, blank=True)
    file = models.FileField(upload_to="upload/")
    filename = models.CharField(max_length=255, null=True, blank=True)
    mime_type = models.CharField(max_length=255, null=True, blank=True)
    thumbnail = models.JSONField(null=True, blank=True)
    size = models.FloatField(null=True, blank=True)
    url = models.CharField(max_length=300, null=True, blank=True)
    thumbhash = models.CharField(max_length=255, blank=True, null=True)
    is_public = models.BooleanField(default=False)

    def save(self, *args, **kwargs):
        # ✅ 1. 确保 filename 已设置（例如来自 serializer 或 upload handler）
        if not self.filename:
            self.filename = self.file.name

        # ✅ 2. 读取原始文件字节（必须 seek(0) 防止多次读取为空）
        self.file.seek(0)
        file_bytes = self.file.read()
        if not file_bytes:
            raise ValueError("Uploaded fil
e is empty.")

        # ✅ 3. 使用 BytesIO 在内存中打开图像
        try:
            image = Image.open(BytesIO(file_bytes))
            image_format = image.format or "JPEG"
            mime_type = Image.MIME.get(image_format, "image/jpeg")
        except Exception as e:
            raise ValueError(f"Invalid image file: {e}")

        # ✅ 4. 处理缩略图（同样在内存中操作）
        sizes = [(150, 150), (256, 256)]
        thumbnail_data = {}
        cache_dir = os.path.join(settings.MEDIA_ROOT, "cache")
        os.makedirs(cache_dir, exist_ok=True)  # ✅ 使用 settings.MEDIA_ROOT 更安全

        for i, (w, h) in enumerate(sizes):
            resized = image.resize((w, h), Image.Resampling.LANCZOS)
            index = "small" if i == 0 else "medium"
            ext = image_format.lower()
            if ext == "jpg":
                ext = "jpeg"
            filename_base = f"{self.id}-resized-{self.filename.rsplit('.', 1)[0]}-{index}.{ext}"
            cache_path = os.path.join(cache_dir, filename_base)

            # 保存到磁盘（此时 MEDIA_ROOT 已确保存在）
            resized.save(cache_path, format=image_format)
            thumbnail_data[f"{w}*{h}"] = f"cache/{filename_base}"  # 相对 URL 路径

        # ✅ 5. 设置字段值
        self.mime_type = mime_type
        self.size = len(file_bytes)
        self.thumbnail = thumbnail_data
        self.url = f"{settings.MEDIA_URL}upload/{self.filename}"
        self.thumbhash = image_to_thumbhash(image)  # 假设该函数接受 PIL.Image

        # ✅ 6. 关键：重置 file 字段指针，并可选覆盖原始文件（如需修改）
        # 若仅需保存原始上传文件，跳过此步；若需保存处理后图像，用 ContentFile 替换：
        # self.file = ContentFile(file_bytes_processed, name=self.filename)

        # ✅ 7. 调用父类 save —— 此时文件才真正写入 media/upload/
        super().save(*args, **kwargs)

⚠️ 注意事项：

• 不要硬编码 "media/upload/"：始终使用 settings.MEDIA_ROOT 和 settings.MEDIA_URL，确保跨环境兼容。
• self.file.name vs self.filename：self.file.name 是上传时的原始文件名（含扩展名），建议优先使用；若需自定义命名，应在 upload_to 函数或 serializer 中统一处理。
• Serializer 需正确调用 save()：你当前的 MediaSerializer.create() 仅返回实例而未保存，应改为：

  def create(self, validated_data):
      return Media.objects.create(**validated_data)  # ✅ 触发 save()

• 大文件风险：self.file.read() 将整个文件加载进内存。对 >10MB 文件，建议改用流式处理或异步任务（如 Celery）。
• 事务与异常安全：若缩略图生成失败，super().save() 不会执行，避免脏数据；但已写入的缓存文件需手动清理（可结合 try/except/finally）。

总结：Django 文件处理的核心原则是——“先内存，后磁盘；先读取，再保存”。放弃对临时路径的幻想，拥抱 BytesIO 与 ContentFile，你的 save() 方法就能既健壮又高效。