Question

问题 - 我有一个使用django-rest-framework（django v1.7.7，django-rest-framework v3.1.1）的REST服务器。在通知中，我让用户知道他们是否收到了新朋友请求，或者是否获得了新徽章。还有其他通知类型，但这个简单的例子可以解释我的问题。

在我的GET响应中，我想通过动态相关对象获取通知，该对象由类型决定。如果类型是friendreq，那么我希望relatedObject是User实例，具有UserSerializer。如果类型是badge，我想让relatedObject成为带有BadgeSerializer的Badge实例。

注意 ：我已经拥有这些其他序列化程序（UserSerializer，BadgeSerializer）。

以下是我希望在回复中实现的目标：

{
    "id": 1,
    "title": "Some Title",
    "type": "friendreq"
    "relatedObject": {
        // this is the User instance. For badge it would be a Badge instance
        "id": 1,
        "username": "foo",
        "email": "foo@bar.com",
    }
}

以下是我对模型和序列化器的看法：

# models.py
class Notification(models.Model):
    """
    Notifications are sent to users to let them know about something. The 
    notifications will be about earning a badge, receiving friend request,
    or a special message from the site admins.
    """
    TYPE_CHOICES = (
        ('badge', 'badge'),
        ('friendreq', 'friend request'),
        ('system', 'system'),
    )
    title = models.CharField(max_length=30)
    type = models.CharField(max_length=10, choices=TYPE_CHOICES)
    user = models.ForeignKey(settings.AUTH_USER_MODEL, related_name="user")
    related_id = models.PositiveIntegerField(null=True, blank=True)


# serializers.py
class NotificationSerializer(serializers.ModelSerializer):
    if self.type == "badge":
        related_object = BadgeSerializer(
            read_only=True,
            queryset=Badge.objects.get(id=self.related_id)
        )
    elif self.type == "friendreq":
        related_object = FriendRequestSerializer(
            read_only=True,
            queryset=FriendRequest.objects.get(id=self.related_id)
        )

    class Meta:
        model = Notification

这段代码不起作用，但希望它能解释我正在努力实现的目标和我想要的方向。也许这个方向是完全错误的，我应该尝试通过使用其他方法来实现这一点。

我尝试的另一个选项是使用SerializerMethodField并在方法中执行此操作，但对于尝试根据其他字段返回Serialized对象的情况，这似乎不那么干净。

Answer 1

我相信您要使用的是DRF文档中提到的.to_representation()方法：http://www.django-rest-framework.org/api-guide/relations/#generic-relationships

django rest框架动态相关对象

1 个答案: