如何在Django中建模“一个或多个”关系？

Question

我有一个模型Owner 。 Owner的类型可以是“ Individual或“ Business 。 根据类型， Owner可能与Property模型具有一对一或一对多的关系。 如果为Individual ，则只能具有一个Property但是如果是Business ，则可能具有多个Property 。 您如何在Django中对此建模？ 如何强制Individual仅拥有一个Property 。

Answer 1

如果您使用的是PostgreSQL或SQLite，则可以使用部分唯一索引来强制实施。 从Django 2.2开始，您可以声明性地执行此操作，例如：

from django.db.models import Model, Q, UniqueConstraint

class Property(Model):
    ...
    class Meta:
        constraints = [UniqueConstraint(fields=["owner"], condition=Q(type="Individual"))]

在Django 2.2之前，您可以通过迁移进行此操作。 就像是：

class Migration(migrations.Migration):

    dependencies = [ ... ]

    operations = [
        migrations.RunSQL("CREATE UNIQUE INDEX property_owner 
                           ON property(owner_id) 
                           WHERE type = 'Individual'"),
    ]

Answer 2

我建议创建一个带有ForeignKey的Owner属性模型，然后通过覆盖类的save函数添加一些自定义验证。

from django.db import IntegrityError

class Property(models.Model):
    owner = models.ForeignKey(Owner)

    def save(self, *args, **kwargs):
        if self.owner.type == "Individual":
            if Property.objects.filter(owner=self.owner).exists():
                raise IntegrityError
        super(Property, self).save(*args, **kwargs)

这不会涵盖所有情况，例如手动将记录添加到数据库或使用某些绕过保存功能的django方法，但它将涵盖大多数情况。