Fix votes (#34)

- make sure we always use the message_id_hash
- factor some code

1 files changed, 17 insertions, 0 deletions

diff --git a/hyperkitty/lib/__init__.py b/hyperkitty/lib/__init__.py
index 52b6bdc..b3fa6c4 100644
--- a/hyperkitty/lib/__init__.py
+++ b/hyperkitty/lib/__init__.py
@@ -25,6 +25,8 @@ import datetime
 
 from django.conf import settings
 
+from hyperkitty.models import Rating
+
 
 def gravatar_url(email):
     '''Return a gravatar url for an email address'''
@@ -86,3 +88,18 @@ def get_display_dates(year, month, day):
         end_date = begin_date + datetime.timedelta(days=1)
 
     return begin_date, end_date
+
+
+def get_votes(message_id_hash):
+    """Extract all the votes for this message"""
+    likes = dislikes = 0
+    try:
+        votes = Rating.objects.filter(messageid=message_id_hash)
+    except Rating.DoesNotExist:
+        votes = {}
+    for vote in votes:
+        if vote.vote == 1:
+            likes += 1
+        elif vote.vote == -1:
+            dislikes += 1
+    return likes, dislikes