Django的動態選擇一個外鍵關係模型

Question

我有我一直在使用的News模型Comment模式：

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 
    ) 

我需要從URL轉向news_pk參數的pkNews或Material分別news_or_material = get_object_or_404(...)

Answer 1

如果你正在開發這個。那麼使用Django提供的GenericForeignKey將是明智之舉。

Generic Foreign Key Django