如何在Django预保存时避免模型编辑及仅首次保存触发事件？

嘿，我来帮你搞定这两个Django模型的问题，都是和保存逻辑挂钩的，刚好我之前也踩过类似的坑，给你分享靠谱的解决办法～

问题1：如何在预保存阶段避免对模型进行编辑？ #

这里分两种场景，看你是要完全禁止更新实例，还是只禁止修改特定字段：

场景1：完全禁止编辑，只允许创建新实例 #

直接重写模型的save()方法，判断实例是否已经存在（通过self.pk是否为None），如果是已存在的实例就抛出异常阻止保存：

from django.core.exceptions import PermissionDenied

class Image(models.Model):
    file_name = models.CharField(max_length=200)
    file_path = models.CharField(max_length=500)  # 修正你原代码里的拼写错误哦

    def save(self, *args, **kwargs):
        if self.pk is not None:
            # 抛出权限异常，阻止编辑操作
            raise PermissionDenied("该模型实例不允许进行编辑操作")
        super().save(*args, **kwargs)

场景2：禁止修改特定字段（比如file_name） #

用pre_save信号来监听保存前的事件，对比数据库里的原始实例和当前实例的字段值，如果被修改了就重置或者抛出异常：

from django.db.models.signals import pre_save
from django.dispatch import receiver
from django.core.exceptions import PermissionDenied

class Image(models.Model):
    file_name = models.CharField(max_length=200)
    file_path = models.CharField(max_length=500)

@receiver(pre_save, sender=Image)
def prevent_file_name_edit(sender, instance, **kwargs):
    if instance.pk is not None:
        # 从数据库获取原始实例
        original = sender.objects.get(pk=instance.pk)
        # 检查file_name是否被修改
        if instance.file_name != original.file_name:
            # 两种选择：要么重置为原始值
            instance.file_name = original.file_name
            # 要么抛出异常阻止保存
            # raise PermissionDenied("file_name字段不允许修改")

问题2：仅在首次保存时触发start_event，更新时不触发？ #

你现在的代码每次调用save()都会执行start_event，不管是创建还是更新，所以只要加个判断——只有当实例是首次创建（self.pk为None）的时候才调用事件就行，修正后的代码如下：

class Image(models.Model):
    file_name = models.CharField(max_length=200)
    file_path = models.CharField(max_length=500)  # 记得修正拼写错误哦

    def save(self, *args, **kwargs):
        # 只有首次创建实例时（还没有主键）才触发事件
        if self.pk is None:
            start_event(self.file_path)
        # 调用父类的save方法完成保存
        super(Image, self).save(*args, **kwargs)

另外也可以用pre_save信号来实现，逻辑是一样的：

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

class Image(models.Model):
    file_name = models.CharField(max_length=200)
    file_path = models.CharField(max_length=500)

@receiver(pre_save, sender=Image)
def trigger_event_on_create(sender, instance, **kwargs):
    if instance.pk is None:
        start_event(instance.file_path)

这样不管是通过Image.objects.create()还是实例化后调用save()创建新对象，都会触发start_event；而更新现有实例时（比如img = Image.objects.get(pk=1); img.file_name = "new.jpg"; img.save()）就不会再触发啦。

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