如何用Django Reversion恢复关联模型的ImageField文件？

我之前也踩过类似的关联模型文件恢复的坑，结合Django Reversion和PostgreSQL外键约束的特性，给你几个可行的解决方案：

方案一：用软删除替代真实删除，保留外键与文件 #

核心思路是不让Model1被真实删除，而是标记为"已删除"，这样Model2的外键始终有效，文件也不会被清理：

• 给Model1添加软删除字段，修改Model2的外键删除行为为DO_NOTHING（避免CASCADE删除Model2）：

class Model1(models.Model):
    is_deleted = models.BooleanField(default=False)
    # 你的其他字段

class Model2(models.Model):
    fk = models.ForeignKey("Model1", on_delete=models.DO_NOTHING)
    image = models.ImageField(upload_to="images/")
    # 你的其他字段

• 通过信号拦截Model1的真实删除，改为标记软删除：

from django.db.models.signals import pre_delete
from django.dispatch import receiver

@receiver(pre_delete, sender=Model1)
def soft_delete_model1(sender, instance, **kwargs):
    instance.is_deleted = True
    instance.save()
    # 抛出异常阻止真实删除，也可以用其他方式拦截
    raise ValueError("Model1 使用软删除，无法直接删除")

• 在Admin中过滤掉已标记删除的Model1实例，让用户操作体验和真实删除一致：

class Model1Admin(VersionAdmin):
    def get_queryset(self, request):
        queryset = super().get_queryset(request)
        return queryset.filter(is_deleted=False)

这样Reversion恢复Model1时，只需把is_deleted改回False，关联的Model2和文件都完好无损，直接就能正常显示。

方案二：自定义Reversion恢复逻辑，同步恢复Model2与文件 #

如果不想改软删除，可利用Reversion保存的版本数据，在恢复Model1后手动恢复关联的Model2实例，确保文件被还原：

• 确保两个模型都已被Reversion注册：

import reversion
reversion.register(Model1)
reversion.register(Model2)

• 编写自定义恢复函数，先恢复Model1，再找回并恢复关联的Model2版本：

from reversion.models import Version
import os

def restore_model1_with_related_images(model1_id):
    # 恢复Model1实例
    with reversion.create_revision():
        model1_version = Version.objects.get_for_object_reference(Model1, model1_id).latest('revision__date_created')
        model1 = model1_version.object_version.object
        model1.save()
    
    # 恢复所有关联的Model2实例及文件
    model2_versions = Version.objects.get_for_model(Model2).filter(object__fk_id=model1_id)
    for version in model2_versions:
        with reversion.create_revision():
            model2 = version.object_version.object
            # 检查文件是否存在，不存在则从版本数据中恢复
            if not os.path.exists(model2.image.path):
                file_content = version.field_dict['image']
                with open(model2.image.path, 'wb') as f:
                    f.write(file_content)
            model2.save()

• 可以把这个函数做成Admin的自定义动作，或者在恢复Model1时自动触发。

方案三：修改ImageField行为，删除实例时不删文件 #

最简单的方案是让ImageField在Model2被CASCADE删除时，不删除物理文件，这样Reversion恢复Model2时能直接关联到原文件：

• 自定义一个不删除文件的ImageField：

class NonDeletingImageField(models.ImageField):
    def delete(self, instance, using=None, keep_parents=False):
        # 重写delete方法，跳过文件删除逻辑
        pass

• 替换Model2中的ImageField：

class Model2(models.Model):
    fk = models.ForeignKey("Model1", on_delete=models.CASCADE)
    image = NonDeletingImageField(upload_to="images/")
    # 你的其他字段

这个方案不用改外键逻辑，但要注意：删除Model1后，文件会留在服务器上，需要定期清理无关联的冗余文件。

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