Include votes in user_profile

1 files changed, 14 insertions, 1 deletions

diff --git a/models.py b/models.py
index 95db0f1..42f1205 100644
--- a/models.py
+++ b/models.py
@@ -15,5 +15,18 @@ class Rating(models.Model):
 class UserProfile(models.Model):
 	# User Object
 	user = models.OneToOneField(User)
-	
+
 	karma = models.IntegerField(default=1)
+
+	def _get_votes(self):
+   	    "Returns all the votes by a user"
+            # Extract all the votes by this user
+     	    try:
+	    	votes = Rating.objects.filter(user = self.user)
+            except Rating.DoesNotExist:
+		votes = {}
+
+     	    return votes
+
+
+	votes = property(_get_votes)