diff options
Diffstat (limited to 'models.py')
-rw-r--r-- | models.py | 20 |
1 files changed, 19 insertions, 1 deletions
@@ -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)) |