Django：創建自定義Slug

Question

這就是我在Django應用中添加子彈的方式，

class Question(models.Model):
    question_text = models.CharField(max_length=250)
    slug = models.SlugField(max_length=255, unique=True)

    def _get_unique_slug(self):
        slug = slugify(self.question_text)
        unique_slug = slug
        num = 1
        while Question.objects.filter(slug=unique_slug).exists():
            unique_slug = '{}-{}'.format(slug, num)
            num += 1

        if unique_slug == '':
        # then simply add an existing number to unique_slug

        return unique_slug

        def save(self, *args, **kwargs):
        if not self.slug:
            self.slug = self._get_unique_slug()
        return super(Question, self).save()

問題是萬一slug是''那么我想自動添加數字1,2,3 ... n

我怎樣才能做到這一點？

Answer 1

您可以將函數重寫為：

def _get_unique_slug(self):
    unique_slug = slug = slugify(self.question_text)
    empty = False if unique_slug else True
    unique_slug = unique_slug or '1'
    for x in itertools.count(1):
        if not Question.objects.filter(slug=unique_slug).exists():
            break
        unique_slug = '{}'.format(x) if empty else '{}-{}'.format(slug, x)

在這里，您可以增加計數並檢入數據庫，直到找到唯一的自由段。