python - 拥有一个与多个不同模型相关的模型

Question

我有一个简单的通知模型:

class Notification(models.Model):
    user = models.ForeignKey(User)
    sender = models.ForeignKey(User)
    model = '''What to put here?'''
    comment = models.CharField(max_length=200)
    created = models.DateTimeField(auto_now=False,auto_now_add=True)

例如，我需要与多个不同模型相关的通知；帖子、用户关注等

在 django 中是否可以关联多个模型，而不是为每个模型创建一个通知模型？

我想避免这样的模型:PostLikeNotification、UserFollowNotification 等

那么django有这个功能吗？我在文档中找不到它。

Answer 1

您可以使用Content Types/Generic Relations

class Notification(models.Model):
    user = models.ForeignKey(User)
    sender = models.ForeignKey(User)
    object_id = models.PositiveIntegerField(default=None, null=True)
    content_type = models.ForeignKey(ContentType, default=None, null=True)
    comment = models.CharField(max_length=200)
    created = models.DateTimeField(auto_now=False,auto_now_add=True)

    @property
    def model_object(self):
        content_type = self.content_type
        object_id = self.object_id
        if content_type is not None and object_id is not None:
            MyClass = content_type.model_class()
            model_object = MyClass.objects.filter(pk=object_id)
            if model_object.exists():
                return model_object.first()
        return None

这里我们将相关对象的模型( Using the Content Types framework )和主键(在本例中必须是整数)存储在通知模型中，然后添加一个属性方法来获取相关对象。

通过此功能，您可以将您的通知与任何其他模型相关联。您还可以使用ForeignKey.limit_choices_to content_type 字段上的参数来验证它是否只接受某些模型。