如果不存在，如何在 Django 中为模型自动创建对象？

Question

我有两个模型。 一个用于 UserProfile，另一个用于 Company。

class UserProfile(models.Model):
     company_name = models.ForeignKey(Company, on_delete = models.CASCADE, related_name = 'company')

class Company(models.Model):
     name = models.CharField(max_length = 100)

我正在使用 Django Rest Framework 来创建用户。 我想要实现的是当我创建一个新用户时，我想为该用户分配一个 company_name 。 如果该 company_name 不存在于数据库中，那么我想随时随地创建它。 但它抛出一个错误。 “无效的超链接 - 没有 URL 匹配。”

Answer 1

首先，您需要将您的 UserProfile 模型与用户相关联。 它应该是一对一的关系，因为我猜用户应该只有一家公司。

在您的序列化程序中，您应该添加 Company 模型并从 API 的输入中保存公司名称，然后将其连接到正在创建的新用户。

Answer 2

您可以使用 python 的@property以简洁明了的方式解决这个问题。 它也适用于创建和更新对象。 请注意， UserPorifle的字段已重命名为company 。 这是你如何做到的：

class Company(models.Model):
    name = models.CharField(max_length=100)


class UserProfile(models.Model):
    company = models.ForeignKey(Company, on_delete=models.CASCADE, related_name='company')

    @property
    def company_name(self):
         return self.company.name

    @company_name.setter
    def company_name(self, value):
        self.company, _ = Company.objects.get_or_create(name=value)

现在您可以使用以下方法创建对象：

UserProfile.objects.create(company_name='Some name')