Django-用于评论上下投票系统的投票实现

from django.db import models
from django.contrib.auth.models import User

class Comment(models.Model):
    user = models.ForeignKey(User, on_delete=models.CASCADE)
    content = models.TextField()
    created_at = models.DateTimeField(auto_now_add=True)

class Vote(models.Model):
    user = models.ForeignKey(User, on_delete=models.CASCADE)
    comment = models.ForeignKey(Comment, on_delete=models.CASCADE)
    vote_type = models.CharField(max_length=1, choices=[('U', 'Upvote'), ('D', 'Downvote')])

from django.shortcuts import get_object_or_404, redirect
from django.http import JsonResponse
from .models import Comment, Vote

def vote(request, comment_id):
    comment = get_object_or_404(Comment, id=comment_id)
    vote_type = request.POST.get('vote_type')
    
    if vote_type not in ['U', 'D']:
        return JsonResponse({'error': 'Invalid vote type'}, status=400)
    
    # Check if the user has already voted
    existing_vote = Vote.objects.filter(user=request.user, comment=comment).first()
    if existing_vote:
        existing_vote.vote_type = vote_type
        existing_vote.save()
    else:
        Vote.objects.create(user=request.user, comment=comment, vote_type=vote_type)
    
    return JsonResponse({'success': True})

from django.urls import path
from .views import vote

urlpatterns = [
    path('vote/<int:comment_id>/', vote, name='vote'),
]

<script src="https://code.jquery.com/jquery-3.6.0.min.js"></script>
<script>
$(document).ready(function() {
    $('.vote-button').click(function() {
        var commentId = $(this).data('comment-id');
        var voteType = $(this).data('vote-type');
        $.post('/vote/' + commentId + '/', {vote_type: voteType}, function(data) {
            if (data.success) {
                alert('Vote submitted!');
            } else {
                alert('Error: ' + data.error);
            }
        });
    });
});
</script>

<button class="vote-button" data-comment-id="1" data-vote-type="U">Upvote</button>
<button class="vote-button" data-comment-id="1" data-vote-type="D">Downvote</button>