Django后台添加Media触发IntegrityError，未显示预期红色Bootstrap提示

我明白你遇到的困扰了——你已经通过UniqueConstraint在数据库层面实现了每个产品只能有一条base_image=True的Media记录的逻辑，功能本身是正常的，但在Django Admin的添加页面提交时，没有出现预期的红色Bootstrap错误提示，反而直接抛出了数据库的IntegrityError，体验很不友好。

问题根源 #

Django的UniqueConstraint是数据库级别的约束，当违反规则时，数据库会直接抛出错误，而Django Admin默认不会自动捕获这类数据库错误并转化为表单层面的友好提示。所以我们需要在表单验证环节提前做检查，把错误拦截在前端。

解决方案：添加表单层面的验证 #

我们可以通过自定义Admin表单，在表单的clean方法里添加业务逻辑验证，这样就能在提交数据到数据库前就检查规则，触发Admin的友好提示。

第一步：自定义Media表单 #

在你的项目里创建一个表单文件（比如forms.py），或者直接写在admin.py里，代码如下：

from django import forms
from django.core.exceptions import ValidationError
from .models import Media

class MediaAdminForm(forms.ModelForm):
    class Meta:
        model = Media
        fields = '__all__'

    def clean(self):
        # 先调用父类的clean方法获取清洗后的表单数据
        cleaned_data = super().clean()
        product = cleaned_data.get('product')
        base_image = cleaned_data.get('base_image')

        # 只有当base_image设为True，且产品已选择时才做检查
        if base_image and product:
            # 检查该产品是否已经存在主图
            existing_base = Media.objects.filter(product=product, base_image=True).exists()
            
            # 如果是编辑现有记录，要排除当前正在编辑的实例（避免自己和自己冲突）
            if self.instance.pk:
                existing_base = Media.objects.filter(
                    product=product, 
                    base_image=True
                ).exclude(pk=self.instance.pk).exists()
            
            if existing_base:
                # 抛出ValidationError，这个错误会被Django Admin捕获并显示为红色提示
                raise ValidationError("这个产品已经设置了主图，不能再添加新的主图啦！")
        
        return cleaned_data

第二步：在Admin中使用自定义表单 #

修改你的admin.py，把刚才的自定义表单绑定到Media的Admin类上：

from django.contrib import admin
from .models import Media
from .forms import MediaAdminForm  # 如果表单在单独文件里，记得导入

@admin.register(Media)
class MediaAdmin(admin.ModelAdmin):
    # 指定使用我们自定义的表单
    form = MediaAdminForm

额外说明 #

• 数据库层面的UniqueConstraint依然要保留，它是最后一道防线，可以防止通过API、命令行等绕过Admin表单的方式添加重复主图
• 表单层面的验证会在用户提交数据时立即检查，把错误转化为Django Admin自带的红色Bootstrap提示，让用户能直观看到问题所在

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