我有一个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
参数从 URL 分别转换为pk
ofNews
或Material
andnews_or_material = get_object_or_404(...)