1 files changed, 89 insertions, 0 deletions
diff --git a/forum/models/meta.py b/forum/models/meta.py
new file mode 100755
index 00000000..3dfd3e86
--- /dev/null
+++ b/forum/models/meta.py
@@ -0,0 +1,89 @@
+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