如何无需重写save()方法自动填充Django的SlugField？

嘿，完全懂你的痛点——给每个模型复制粘贴那段save()代码真的太繁琐了！其实Django有几种优雅的方式实现你要的效果，不用每个模型都写一遍逻辑：

方法一：自定义带自动填充的SlugField（最贴合你的需求） #

你可以自己封装一个AutoSlugField，继承自models.SlugField，添加populate_from参数来指定要生成slug的源字段，然后在字段的pre_save方法里自动处理slug生成逻辑。这样每个模型只需要用这个自定义字段，指定populate_from='name'就行：

from django.db import models
from django.utils.text import slugify
from django.core.exceptions import ObjectDoesNotExist

class AutoSlugField(models.SlugField):
    def __init__(self, populate_from=None, unique=True, **kwargs):
        self.populate_from = populate_from
        super().__init__(unique=unique, **kwargs)

    def pre_save(self, model_instance, add):
        # 如果实例还没slug，从指定字段生成
        if add and not getattr(model_instance, self.attname):
            if not self.populate_from:
                raise ValueError("必须指定populate_from参数来设置slug的源字段")
            
            source_value = getattr(model_instance, self.populate_from)
            slug = slugify(source_value)
            
            # 处理slug唯一的逻辑：如果存在就加序号
            model_class = model_instance.__class__
            counter = 1
            while True:
                try:
                    model_class.objects.get(**{self.attname: slug})
                    slug = f"{slugify(source_value)}-{counter}"
                    counter += 1
                except ObjectDoesNotExist:
                    break
            
            setattr(model_instance, self.attname, slug)
        return super().pre_save(model_instance, add)

然后在你的模型里这样用：

class YourModel(models.Model):
    name = models.CharField(max_length=255)
    slug = AutoSlugField(populate_from='name')

这样每次创建实例时，slug会自动从name字段生成，还能保证唯一性，完全不用重写save()！

方法二：使用Django信号（全局统一处理） #

如果不想自定义字段，也可以用pre_save信号来全局监听模型的保存事件，自动生成slug。你可以写一个通用的信号接收器，只给带有slug字段和指定源字段的模型处理：

from django.db.models.signals import pre_save
from django.dispatch import receiver
from django.utils.text import slugify
from django.core.exceptions import ObjectDoesNotExist

@receiver(pre_save)
def auto_generate_slug(sender, instance, **kwargs):
    # 只处理有slug字段和name字段的模型，且实例slug为空
    if hasattr(instance, 'slug') and hasattr(instance, 'name') and not instance.slug:
        slug = slugify(instance.name)
        model_class = sender
        counter = 1
        while True:
            try:
                model_class.objects.get(slug=slug)
                slug = f"{slugify(instance.name)}-{counter}"
                counter += 1
            except ObjectDoesNotExist:
                break
        instance.slug = slug

把这段代码放在你的models.py或者专门的signals.py里（记得在apps.py里注册信号），这样所有符合条件的模型都会自动生成slug，不用每个模型写逻辑。

额外提示：Django Admin的自动填充原理 #

其实Django Admin的自动填充是用了prepopulated_fields配置，但那只是Admin界面的前端自动生成，后端还是需要保证唯一性。上面的两种方法都是后端层面的处理，比Admin的前端填充更可靠，能确保不管是通过Admin还是API、脚本创建的实例，slug都能正确生成。

这样以后你再也不用复制粘贴那段save()代码啦！

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