Django动态选择外键关系模型

时间:2016-12-30 10:34:15

标签: django python-3.x django-models foreign-keys generic-foreign-key

我有一个Comment模型,我一直用于News模型:

class Comment(models.Model):
    news = models.ForeignKey(News, on_delete=models.CASCADE)
    posted_by = models.ForeignKey(Student, on_delete=models.CASCADE)
    content = models.TextField(max_length=2048)
    posted_on = models.DateTimeField(auto_now_add=True)
    edited = models.BooleanField(default=False)
    last_edited_on = models.DateTimeField(auto_now=True)


    def __str__(self):
        return '{} - {}'.format(self.posted_by, self.news)

但是现在我有一个Materials模型,我也想在那里发表评论,但是要使用相同的Comments模型。有没有办法动态选择外键关系(news = models.ForeignKey(...) - > news_or_material = ...或类似的东西。)

当然我可以写一个单独的模型(MaterialComment),但我想重用我的代码。

然后在我的视野中我应该做这样的事情(我正在使用Django REST Framework):

def create(self, request, news_pk=None):
    news = get_object_or_404(News, id=news_pk)
    context = {'request': request, 'news': news}

    serializer = self.get_serializer_class()(
        context=context, data=request.data
    )
    serializer.is_valid(raise_exception=True)
    self.perform_create(serializer)
    headers = self.get_success_headers(serializer.data)

    return Response(
        serializer.validated_data,
        status=status.HTTP_201_CREATED,
        headers=headers
    )

我需要将news_pk参数从网址转为pk NewsMaterialnews_or_material = get_object_or_404(...)

1 个答案:

答案 0 :(得分:2)

如果你正在开发这个。然后明智地使用Django提供的GenericForeignKey。

Generic Foreign Key Django