Question

我有多对多关系的Tag和Article数据模型，

class Tag(models.Model):
    owner = models.ForeignKey(User,on_delete=models.CASCADE)
    name = models.CharField(max_length=100)
    def __str__(self):
        return self.name
    class Meta:
        ordering = ("id",)

class Article(models.Model):
    tags = models.ManyToManyField(Tag, blank=True)
    owner = models.ForeignKey(User, on_delete=models.CASCADE)
    ...

当接收到从表单输入的标签时，我首先检查标签是否存在，以确定是否在数据库中创建一个新标签，
然后将所有输入标签附加到文章上。

#save the new article in database   
if article_form.is_valid():
    article = article_form.save(commit=False)
    article.owner = self.request.user
    article.save()

创建在记录中找不到的标签

#compare tags
input_tags = self.request.POST["tags"]
input_tags = tags.split(",")
query_tags = Tag.objects.all()
#get name list of tab objects
query_tags = [tag.name for tag in query_tags]
#create new tabs in TabTable
for tag in input_tags:
    if tag not in query_tags:
        Tag.objects.create(name=tag, owner=self.request.user)

检索输入标签对象

tag_objs = []
for input_tag in input_tags:
    try:
        qualified_tab = Tab.object.filter(name=input_tag)[0]
        tag_objs.append(qualified_tab)

将标签附加到文章
```
article.tags.add(*tag_objs)
```

这是一项常规任务，但解决方案很麻烦。

如何优雅地完成这样的任务？

Answer 1

get_or_create（默认设置为None，** kwargs）

一种方便的方法，用于使用给定的kwarg查找对象（如果模型的所有字段均具有默认值，则为空），必要时创建一个对象。

返回（创建对象）的元组，其中object是已检索或创建的对象，并且created是一个布尔值，指定是否创建了新对象。

if article_form.is_valid():
    article = article_form.save(commit=False)
    article.owner = self.request.user
    article.save()

   # tag list
   input_tags = self.request.POST["tags"]
   input_tags = tags.split(",")

   #create new tabs in TabTable
   for tag in input_tags:
       tag, created = Tag.objects.get_or_created(
                           name=tag, 
                           defaults={"owner":self.request.user})
       tag.article_set.add(article)
       #article.tags.add(tag) test

如果数据库中不存在输入标签，则将其添加到文章中并创建新的标签记录

1 个答案: