向文章添加输入标签,并在数据库中不存在时创建新的标签记录



我有多对多关系的TagArticle数据模型,

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()
  1. 创建在记录中找不到的标签

    #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)
    
  2. 检索输入标签对象

    tag_objs = []
    for input_tag in input_tags:
    try:
    qualified_tab = Tab.object.filter(name=input_tag)[0]
    tag_objs.append(qualified_tab)
    
  3. 将标签附加到文章

    article.tags.add(*tag_objs)
    

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

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

get_or_create(默认值=无,**kwargs(

一种方便的方法,用于查找具有给定kwargs的对象(如果您的模型对所有字段都有默认值,则可能为空(,并在必要时创建一个。

返回(object,created(的元组,其中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

最新更新