X-Git-Url: https://git.openstreetmap.org./osqa.git/blobdiff_plain/a9eef437702d5df7a2f97010e6798c689371808c..8b123b1f0d204c972d5a2554f9b27e4f7d8ae33f:/forum/models/meta.py

diff --git a/forum/models/meta.py b/forum/models/meta.py
old mode 100755
new mode 100644
index 3dfd3e8..2b790d8
--- a/forum/models/meta.py
+++ b/forum/models/meta.py
@@ -1,89 +1,82 @@
-from base import *
-
-class VoteManager(models.Manager):
-    def get_up_vote_count_from_user(self, user):
-        if user is not None:
-            return self.filter(user=user, vote=1).count()
-        else:
-            return 0
-
-    def get_down_vote_count_from_user(self, user):
-        if user is not None:
-            return self.filter(user=user, vote=-1).count()
-        else:
-            return 0
-
-    def get_votes_count_today_from_user(self, user):
-        if user is not None:
-            today = datetime.date.today()
-            return self.filter(user=user, voted_at__range=(today, today + datetime.timedelta(1))).count()
-
-        else:
-            return 0
-
-
-class Vote(MetaContent):
-    VOTE_UP = +1
-    VOTE_DOWN = -1
-    VOTE_CHOICES = (
-        (VOTE_UP,   u'Up'),
-        (VOTE_DOWN, u'Down'),
-    )
-
-    vote           = models.SmallIntegerField(choices=VOTE_CHOICES)
-    voted_at       = models.DateTimeField(default=datetime.datetime.now)
-
-    objects = VoteManager()
-
-    class Meta(MetaContent.Meta):
-        unique_together = ('content_type', 'object_id', 'user')
-        db_table = u'vote'
-
-    def __unicode__(self):
-        return '[%s] voted at %s: %s' %(self.user, self.voted_at, self.vote)
-
-    def is_upvote(self):
-        return self.vote == self.VOTE_UP
-
-    def is_downvote(self):
-        return self.vote == self.VOTE_DOWN
-
-
-class FlaggedItemManager(models.Manager):
-    def get_flagged_items_count_today(self, user):
-        if user is not None:
-            today = datetime.date.today()
-            return self.filter(user=user, flagged_at__range=(today, today + datetime.timedelta(1))).count()
-        else:
-            return 0
-
-class FlaggedItem(MetaContent):
-    """A flag on a Question or Answer indicating offensive content."""
-    flagged_at     = models.DateTimeField(default=datetime.datetime.now)
-
-    objects = FlaggedItemManager()
-
-    class Meta(MetaContent.Meta):
-        unique_together = ('content_type', 'object_id', 'user')
-        db_table = u'flagged_item'
-
-    def __unicode__(self):
-        return '[%s] flagged at %s' %(self.user, self.flagged_at)
-
-class Comment(MetaContent):
-    comment        = models.CharField(max_length=300)
-    added_at       = models.DateTimeField(default=datetime.datetime.now)
-
-    class Meta(MetaContent.Meta):
-        ordering = ('-added_at',)
-        db_table = u'comment'
-
-    def save(self,**kwargs):
-        super(Comment,self).save(**kwargs)
-        try:
-            ping_google()
-        except Exception:
-            logging.debug('problem pinging google did you register you sitemap with google?')
-
-    def __unicode__(self):
-        return self.comment
\ No newline at end of file
+from django.utils.translation import ugettext as _
+from base import *
+
+class Vote(models.Model):
+    user = models.ForeignKey(User, related_name="votes")
+    node = models.ForeignKey(Node, related_name="votes")
+    value = models.SmallIntegerField()
+    action = models.OneToOneField(Action, related_name="vote")
+    voted_at = models.DateTimeField(default=datetime.datetime.now)
+
+    class Meta:
+        app_label = 'forum'
+        unique_together = ('user', 'node')
+
+
+class Flag(models.Model):
+    user = models.ForeignKey(User, related_name="flags")
+    node = models.ForeignKey(Node, related_name="flags")
+    reason = models.CharField(max_length=300)
+    action = models.OneToOneField(Action, related_name="flag")
+    flagged_at = models.DateTimeField(default=datetime.datetime.now)
+
+    class Meta:
+        app_label = 'forum'
+        unique_together = ('user', 'node')
+
+
+class Badge(BaseModel):
+    GOLD = 1
+    SILVER = 2
+    BRONZE = 3
+
+    type        = models.SmallIntegerField()
+    cls         = models.CharField(max_length=50, null=True)
+    awarded_count = models.PositiveIntegerField(default=0)
+    
+    awarded_to    = models.ManyToManyField(User, through='Award', related_name='badges')
+
+    def get_handler(self):
+        from forum.badges import BadgesMeta
+        return BadgesMeta.by_id.get(self.id, None)
+
+    @property
+    def name(self):
+        cls = self.get_handler()
+        return cls and cls.name or _("Unknown")
+
+    @property
+    def description(self):
+        cls = self.get_handler()
+        return cls and cls.description or _("No description available")
+
+    @models.permalink
+    def get_absolute_url(self):
+        return ('badge', [], {'id': self.id, 'slug': slugify(self.name)})
+
+    def save(self, *args, **kwargs):
+        if isinstance(self.awarded_count, models.expressions.ExpressionNode):
+            super(Badge, self).save(*args, **kwargs)
+            self.awarded_count = self.__class__.objects.filter(id=self.id).values_list('awarded_count', flat=True)[0]
+        else:
+            super(Badge, self).save(*args, **kwargs)
+
+
+    class Meta:
+        app_label = 'forum'
+
+
+class Award(models.Model):
+    user = models.ForeignKey(User)
+    badge = models.ForeignKey('Badge', related_name="awards")
+    node = models.ForeignKey(Node, null=True)
+
+    awarded_at = models.DateTimeField(default=datetime.datetime.now)
+
+    trigger = models.ForeignKey(Action, related_name="awards", null=True)
+    action = models.OneToOneField(Action, related_name="award")
+
+
+    class Meta:
+        unique_together = ('user', 'badge', 'node')
+        app_label = 'forum'
\ No newline at end of file