Added __unicode__ function for models

1 files changed, 19 insertions, 1 deletions

diff --git a/models.py b/models.py
index 42f1205..4a8f35c 100644
--- a/models.py
+++ b/models.py
@@ -7,11 +7,26 @@ logger = logging.getLogger(__name__)
 
 
 class Rating(models.Model):
+	# @TODO: instead of list_address, user list model from kittystore?
 	list_address = models.CharField(max_length=50)
+
+	# @TODO: instead of messsageid, use message model from kittystore?
 	messageid = models.CharField(max_length=100)
+
 	user = models.ForeignKey(User)
+
 	vote = models.SmallIntegerField()
 
+	def __unicode__(self):
+		"""Unicode representation"""
+		if self.vote == 1:
+	  		return u'id = %s : %s voted up %s' % (self.id, unicode(self.user), self.messageid)
+		else:
+			return u'id = %s : %s voted down %s' % (self.id, unicode(self.user), self.messageid)
+
+	
+
+
 class UserProfile(models.Model):
 	# User Object
 	user = models.OneToOneField(User)
@@ -28,5 +43,8 @@ class UserProfile(models.Model):
 
      	    return votes
 
-
 	votes = property(_get_votes)
+
+	def __unicode__(self):
+		"""Unicode representation"""
+		return u'%s' % (unicode(self.user))