我有一个包含两个字段的模型,如下所示
模型.py
class Publisher(models.Model):
name = models.CharField(max_length=200)
slug = models.SlugField(max_length=150, unique=True)
def save(self, *args, **kwargs):
if not self.id and not self.slug:
slug = slugify(self.name)
try:
slug_exits = Publisher.objects.get(slug=slug)
if slug_exits:
self.slug = slug + '_1'
except Publisher.DoesNotExist:
self.slug = slug
super(Publisher, self).save(*args, **kwargs)
在这里,我正在创建一个基于
name
的 slug我们可以在上面看到的字段因此,当我们尝试使用
name already exists
创建发布者时, save
该模型的方法将添加 _1
到最后。当我们再次尝试使用相同的
name
创建新记录时, 一个 Integrity
将引发如下错误>> Publisher.objects.create(name="abc")
result: slug will be "abc"
>> Publisher.objects.create(name="abc")
result: slug will be "abc_1"
>> Publisher.objects.create(name="abc")
result:
.................
.................
34 del cursor
35 del connection
---> 36 raise errorclass, errorvalue
37
38 re_numeric_part = re.compile(r"^(\d+)")
IntegrityError: (1062, "Duplicate entry 'abc_1' for key 'slug'")
所以我想要的是如果标题/slug 已经存在于数据库中并且如果 slug 包含数字(最后像 abc
_1
),我们应该增加它的数字所以我想要的是
increment the number in the slug
如果标题/slug 已存在于数据库中,则如下所示abc
abc_1
abc_2
abc_3
那么任何人都可以让我知道如何实现上述增加 slug 的逻辑吗?
提前致谢......
最佳答案
您将不得不使用循环而不仅仅是一个条件。尝试这个:
class Publisher(models.Model):
name = models.CharField(max_length=200)
slug = models.SlugField(max_length=150, unique=True)
def save(self, *args, **kwargs):
if not self.id and not self.slug:
slug = slugify(self.name)
slug_exists = True
counter = 1
self.slug = slug
while slug_exists:
try:
slug_exits = Publisher.objects.get(slug=slug)
if slug_exits:
slug = self.slug + '_' + str(counter)
counter += 1
except Publisher.DoesNotExist:
self.slug = slug
break
super(Publisher, self).save(*args, **kwargs)
关于django - 通过避免 django 模型保存方法中的完整性错误来增加 slug,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/18182001/