いいね機能をGenericRelationで実装したいのですが

ArticleモデルとCommentモデルにいいね(一般化リレーション)を付けれる様にしたいのですが、困ったことがあるのでこちらで質問させて頂きます。

Python
1# model
2class Like(models.Model):
3    user = models.ForeignKey(settings.AUTH_USER_MODEL)
4    content_type = models.ForeignKey(ContentType, on_delete=models.CASCADE)
5    object_id = models.PositiveIntegerField()
6    content_object = GenericForeignKey()
7    created_at = models.DateTimeField(auto_now_add=True)
8
9# views
10def like(request, id):
11    user = request.user
12    ct = Article.objects.get(id=id)
13    if request.method == 'POST':
14        like = Like.objects.create(user=user,
15                                   content_object=ct)
16        like.save()
17    ...
18    ...

上記のコードでArticleにいいねを付けれる様になったのですが、当然これではCommentにいいねは付けれないので、
どの様にすればcontent_objectを判定し、Article, Commentそれぞれにいいねを付けれるのでしょうか。

また、この様な実装ではなく、ArticleLike, CommentLikeといった形で実装するべきなのでしょうか。

どなたか回答宜しくお願い致します。

行動規範の内容に同意します

回答1件

ベストアンサー

どの様にすればcontent_objectを判定し、Article, Commentそれぞれにいいねを付けれるのでしょうか。

ここの部分がよくわかりませんでした。 GenericForeignKeyを使ってるのであれば、Article や Commentのオブジェクトをそのまま content_object に渡して save できないでしょうか？

python
1# content_objectはモデルオブジェクトであれば、なんでもセットできる
2
3article = Article.objects.get(pk=1)
4Like.objects.create(user=request.user, content_object=article)
5
6comment = Comment.objects.get(pk=1)
7Like.objects.create(user=request.user, content_object=comment)

単純にviewの作り方で迷ってるのであれば、 view に content_type を情報を渡せば良いと思いました。

python
1/api/like/(?P<content_type>\w+)/(?P<id>[0-9]+)/
2
3def like(request, content_type, id):
4    user = request.user
5    if content_type == 'article':
6        obj = Article.objects.get(id=id)
7    elif content_type == 'comment':
8        obj = Comment.objects.get(id=id)
9
10    if request.method == 'POST':
11        like = Like.objects.create(user=user, content_object=obj)