diff options
Diffstat (limited to 'website/views.py')
| -rw-r--r-- | website/views.py | 55 |
1 files changed, 51 insertions, 4 deletions
diff --git a/website/views.py b/website/views.py index 5974433..6e84cfa 100644 --- a/website/views.py +++ b/website/views.py @@ -60,9 +60,8 @@ def get_question(request, question_id=None, pretty_url=None): form = AnswerQuestionForm() thisuserupvote = question.userUpVotes.filter(id=request.user.id).count() thisuserdownvote = question.userDownVotes.filter(id=request.user.id).count() - net_count = question.userUpVotes.count() - question.userDownVotes.count() - + context = { 'question': question, 'answers': answers, @@ -307,12 +306,12 @@ def new_question(request): def vote_post(request): - print "post" + post_id = int(request.POST.get('id')) vote_type = request.POST.get('type') vote_action = request.POST.get('action') - print post_id + cur_post = get_object_or_404(Question, id=post_id) thisuserupvote = cur_post.userUpVotes.filter(id=request.user.id).count() @@ -353,6 +352,54 @@ def vote_post(request): return HttpResponse(num_votes) +def ans_vote_post(request): + + + post_id = int(request.POST.get('id')) + + vote_type = request.POST.get('type') + vote_action = request.POST.get('action') + + cur_post = get_object_or_404(Answer, id=post_id) + + userupvote = cur_post.userUpVotes.filter(id=request.user.id).count() + userdownvote = cur_post.userDownVotes.filter(id=request.user.id).count() + + initial_votes = cur_post.userUpVotes.count() - cur_post.userDownVotes.count() + + # print "User Initial Upvote and Downvote: %d %d %s " % (thisuserupvote, thisuserdownvote, vote_action) + + #This loop is for voting + if vote_action == 'vote': + if (userupvote == 0) and (userdownvote == 0): + if vote_type == 'up': + cur_post.userUpVotes.add(request.user) + elif vote_type == 'down': + cur_post.userDownVotes.add(request.user) + else: + return HttpResponse("Error: Unknown vote-type passed.") + else: + return HttpResponse(initial_votes) + #This loop is for canceling vote + elif vote_action == 'recall-vote': + if (vote_type == 'up') and (userupvote == 1): + cur_post.userUpVotes.remove(request.user) + elif (vote_type == 'down') and (userdownvote == 1): + cur_post.userDownVotes.remove(request.user) + else: + # "Error - Unknown vote type or no vote to recall" + return HttpResponse(initial_votes) + else: + return HttpResponse("Error: Bad Action.") + + num_votes = cur_post.userUpVotes.count() - cur_post.userDownVotes.count() + cur_post.num_votes = num_votes + cur_post.save() + + print "Num Votes: %s" % num_votes + + return HttpResponse(num_votes) + # Notification Section @login_required def user_questions(request, user_id): |