コード例 #1
0
ファイル: message.py プロジェクト: olasd/hyperkitty
def vote(request, mlist_fqdn, message_id_hash):
    """ Add a rating to a given message identified by messageid. """
    if request.method != 'POST':
        raise SuspiciousOperation

    if not request.user.is_authenticated():
        return HttpResponse('You must be logged in to vote',
                            content_type="text/plain",
                            status=403)

    store = get_store(request)
    message = store.get_message_by_hash_from_list(mlist_fqdn, message_id_hash)
    if message is None:
        raise Http404

    value = int(request.POST['vote'])
    if value not in [-1, 0, 1]:
        raise SuspiciousOperation

    # Checks if the user has already voted for a this message.
    try:
        v = Rating.objects.get(user=request.user,
                               messageid=message_id_hash,
                               list_address=mlist_fqdn)
        if v.vote == value:
            return HttpResponse("You've already cast this vote",
                                content_type="text/plain",
                                status=403)
    except Rating.DoesNotExist:
        if value != 0:
            v = Rating(list_address=mlist_fqdn,
                       messageid=message_id_hash,
                       vote=value)
            v.user = request.user
        else:
            return HttpResponse("There is no vote to cancel",
                                content_type="text/plain",
                                status=500)

    if value == 0:
        v.delete()
    else:
        v.vote = value
        v.save()

    # Extract all the votes for this message to refresh it
    set_message_votes(message, request.user)
    t = loader.get_template('messages/like_form.html')
    html = t.render(
        RequestContext(request, {
            "object": message,
            "message_id_hash": message_id_hash,
        }))

    result = {
        "like": message.likes,
        "dislike": message.dislikes,
        "html": html,
    }
    return HttpResponse(json.dumps(result), mimetype='application/javascript')
コード例 #2
0
ファイル: test_views.py プロジェクト: TomasTomecek/hyperkitty
 def test_vote_cancel(self):
     v = Rating(list_address="*****@*****.**", messageid="m1", vote=1)
     v.user = self.user
     v.save()
     v = Rating(list_address="*****@*****.**", messageid="m2", vote=-1)
     v.user = self.user
     v.save()
     for msg in ["m1", "m2"]:
         request = self.factory.post("/vote", {"vote": "0"})
         request.user = self.user
         resp = vote(request, '*****@*****.**', msg)
         self.assertEqual(resp.status_code, 200)
         try:
             Rating.objects.get(user=self.user, messageid=msg,
                                list_address='*****@*****.**')
         except Rating.DoesNotExist:
             pass
         else:
             self.fail("Vote for msg %s should have been deleted" % msg)
         result = json.loads(resp.content)
         self.assertEqual(result["like"], 0)
         self.assertEqual(result["dislike"], 0)