Django中如何重写模型的update方法以避开取消课程的唯一约束报错

• 你自定义的update方法不属于Django模型的原生生命周期方法，完全不会被自动触发：
  1. 单实例更新时调用的是save()方法，不会执行你写的update逻辑
  2. 批量更新时调用的是Django原生QuerySet的update()方法，该操作直接生成SQL执行，不会调用模型实例上的自定义方法
• 唯一约束触发的原因：当前你要更新的课程，已经存在相同class_id、calendar_date、is_cancelled=True、cancellation_count=0的记录，违反了你配置的唯一键规则。

方案1：单实例更新场景，重写save()方法做异常捕获 #

from django.db import IntegrityError

class YourClassModel(models.Model):
    # 你的字段定义省略
    class Meta:
        constraints = [
            models.UniqueConstraint(fields=['class_id', 'calendar_date', 'is_cancelled', 'cancellation_count'],
                                    name='unique-class')
        ]

    def save(self, *args, **kwargs):
        # 仅对状态变为已取消的更新操作做异常捕获
        if not self._state.adding and self.is_cancelled:
            # 检查是否是首次变更为取消状态
            original_instance = YourClassModel.objects.get(pk=self.pk)
            if not original_instance.is_cancelled:
                try:
                    return super().save(*args, **kwargs)
                except IntegrityError as e:
                    if "unique-class" in str(e):
                        print("Classes are already cancelled.")
                        return
                    # 非唯一键异常正常抛出
                    raise
        # 其他场景正常走保存逻辑
        super().save(*args, **kwargs)

方案2：批量更新场景，自定义QuerySet实现带异常捕获的update方法 #

from django.db import IntegrityError, models

class ClassQuerySet(models.QuerySet):
    def update(self, *args, **kwargs):
        try:
            return super().update(*args, **kwargs)
        except IntegrityError as e:
            if "unique-class" in str(e):
                print("部分课程已处于取消状态，跳过重复操作")
                # 返回0表示无记录被更新，可根据业务需求调整
                return 0
            raise

class YourClassModel(models.Model):
    # 你的字段定义省略
    objects = ClassQuerySet.as_manager()

    class Meta:
        constraints = [
            models.UniqueConstraint(fields=['class_id', 'calendar_date', 'is_cancelled', 'cancellation_count'],
                                    name='unique-class')
        ]

如果你配置唯一约束的目标是「同一课程同一天只能有一个未取消的有效记录」，可以修改为条件唯一约束，从根源上避免取消操作触发唯一键报错：

from django.db.models import Q

class Meta:
    constraints = [
        models.UniqueConstraint(
            fields=['class_id', 'calendar_date'],
            condition=Q(is_cancelled=False),
            name='unique_active_class'
        )
    ]

内容的提问来源于stack exchange，提问作者raj-kapil