summaryrefslogtreecommitdiffstats
path: root/models.py
diff options
context:
space:
mode:
Diffstat (limited to 'models.py')
-rw-r--r--models.py20
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))