在Django中添加独特的Slug

时间:2017-12-24 15:00:21

标签: python django

我试图在我的学校项目中添加一个slug字段,这是我在模型中尝试的内容,

def pre_save_receiver(sender, instance, *args, **kwargs):

   slug = slugify(instance.title)
   exists = Data.objects.filter(slug=slug).exists()
   if exists:
      instance.slug = "%s-%s" % (slug, instance.id)
   else:
      instance.slug = slug



pre_save.connect(pre_save_receiver, sender=Data)

一切都运转良好,但问题是即使它是独特的,它也会在slug字段中添加ID。 我该如何解决这个问题?

2 个答案:

答案 0 :(得分:0)

模型至少具有以下内容:

class YourModel(models.Model):
    title   = models.CharField(max_length=120)
    slug  = models.SlugField(blank=True, null=True)

您需要在app文件夹中创建一个名为utils的文件,其中包含以下代码:

<强> utils.py

from django.utils.text import slugify

def random_string_generator(size=10, chars=string.ascii_lowercase + string.digits):
    return ''.join(random.choice(chars) for _ in range(size))

def unique_slug_generator(instance, new_slug=None):
    if new_slug is not None:
        slug = new_slug
    else:
        slug = slugify(instance.title)

    Klass = instance.__class__
    qs_exists = Klass.objects.filter(slug=slug).exists()
    if qs_exists:
        new_slug = "{slug}-{randstr}".format(
                    slug=slug,
                    randstr=random_string_generator(size=4)
                )
        return unique_slug_generator(instance, new_slug=new_slug)
    return slug

<强> Models.py

from django.db.models.signals import pre_save, post_save

from .utils import unique_slug_generator

def pre_save_receiver(sender, instance, *args, **kwargs):
    if not instance.slug:
    instance.slug = unique_slug_generator(instance)

pre_save.connect(pre_save_receiver, sender=YourModel)

答案 1 :(得分:0)

您正在尝试获取不存在的模型的ID。模型保存前 pre_save 信号调用。而不是 pre_save 使用 post_save 信号