在 Django 中同时保存对象及其相关对象

Question

在 Django 中，有没有办法创建一个对象，创建它的相关对象，然后一次性保存它们？

例如，在下面的代码中：

from django.db import models


class Post(models.Model):
    title = models.CharField(max_length=255)
    body = models.CharField(max_length=255)


class Tag(models.Model):
    post = models.ForeignKey(Post)
    title = models.CharField(max_length=255)

post = Post(title='My Title', body='My Body')
post.tag_set = [Tag(post=post, title='test tag'), Tag(post=post, title='second test tag')]
post.save()

我创建了一个 Post 对象。然后我还想创建和关联我的 Tag 对象。我想避免保存帖子然后保存标签，因为如果 post.save() 成功，那么 tag.save() 失败，我会留下一个没有标签的帖子。

在 Django 中有没有办法一次保存所有这些，或者至少可以加强数据完整性？

Answer 1

transactions 救援！

from django.db import transaction

with transaction.atomic():
   post = Post.objects.create('My Title', 'My Body')
   post.tag_set = [Tag(post, 'test tag'), Tag(post, 'second test tag')]

附带说明：我认为您真的想要Post 和Tag 之间的多对多关系...

Answer 2

覆盖保存...

class Post(models.Model):

    ...

    def save(self, *args, **kwargs):
        super(Post, self).save(*args, **kwargs)
        for tag in self.tag_set:
            tag.save()

这样您就不必一遍又一遍地编写事务。如果你想使用事务，只需在 save 方法中实现它，而不是循环。