Adding ability to convert comments to answers.

[osqa.git] / forum / models / meta.py

diff --git a/forum/models/meta.py b/forum/models/meta.py
index 7e8213786f211dacb25bdc6840d10be2d6693d6e..41dc5d67d35fcc23ca7bda0c3943834976deb480 100644 (file)
--- a/forum/models/meta.py
+++ b/forum/models/meta.py
@@ -1,152 +1,99 @@
+from django.utils.translation import ugettext as _

 from base import *
 from base import *

-import re

 
 

-class Vote(MetaContent, UserContent):
-    VOTE_UP = +1
-    VOTE_DOWN = -1
-    VOTE_CHOICES = (
-        (VOTE_UP,   u'Up'),
-        (VOTE_DOWN, u'Down'),
-    )
+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)

 
 

-    vote           = models.SmallIntegerField(choices=VOTE_CHOICES)
-    voted_at       = models.DateTimeField(default=datetime.datetime.now)
-    canceled       = models.BooleanField(default=False)
-
-    active = ActiveObjectManager()
+    class Meta:
+        app_label = 'forum'
+        unique_together = ('user', 'node')

 
 

-    class Meta(MetaContent.Meta):
-        db_table = u'vote'

 
 

-    def __unicode__(self):
-        return '[%s] voted at %s: %s' %(self.user, self.voted_at, self.vote)
+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)

 
 

-    def _update_post_vote_count(self, diff):
-        post = self.node
-        field = self.vote == 1 and 'vote_up_count' or 'vote_down_count'
-        post.__dict__[field] = post.__dict__[field] + diff
-        post.save()
+    class Meta:
+        app_label = 'forum'
+        unique_together = ('user', 'node')

 
 

-    def save(self, *args, **kwargs):
-        super(Vote, self).save(*args, **kwargs)
-        if self._is_new:
-            self._update_post_vote_count(1)
+class BadgesQuerySet(models.query.QuerySet):
+    def get(self, *args, **kwargs):
+        try:
+            pk = [v for (k,v) in kwargs.items() if k in ('pk', 'pk__exact', 'id', 'id__exact')][0]
+        except:
+            return super(BadgesQuerySet, self).get(*args, **kwargs)

 
 

-    def cancel(self):
-        if not self.canceled:
-            self.canceled = True
-            self.save()
-            self._update_post_vote_count(-1)
-            vote_canceled.send(sender=Vote, instance=self)
+        from forum.badges.base import BadgesMeta
+        badge = BadgesMeta.by_id.get(int(pk), None)
+        if not badge:
+            return super(BadgesQuerySet, self).get(*args, **kwargs)
+        return badge.ondb

 
 

-    def is_upvote(self):
-        return self.vote == self.VOTE_UP

 
 

-    def is_downvote(self):
-        return self.vote == self.VOTE_DOWN
+class BadgeManager(models.Manager):
+    use_for_related_fields = True

 
 

-vote_canceled = django.dispatch.Signal(providing_args=['instance'])
+    def get_query_set(self):
+        return BadgesQuerySet(self.model)

 
 

-class FlaggedItem(MetaContent, UserContent):
-    """A flag on a Question or Answer indicating offensive content."""
-    flagged_at     = models.DateTimeField(default=datetime.datetime.now)
-    reason         = models.CharField(max_length=300, null=True)
-    canceled       = models.BooleanField(default=False)
+class Badge(models.Model):
+    GOLD = 1
+    SILVER = 2
+    BRONZE = 3

 
 

-    active = ActiveObjectManager()
+    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')

 
 

-    class Meta(MetaContent.Meta):
-        db_table = u'flagged_item'
+    objects = BadgeManager()

 
 

-    def __unicode__(self):
-        return '[%s] flagged at %s' %(self.user, self.flagged_at)
+    @property
+    def name(self):
+        cls = self.__dict__.get('_class', None)
+        return cls and cls.name or _("Unknown")

 
 

-    def _update_post_flag_count(self, diff):
-        post = self.node
-        post.offensive_flag_count = post.offensive_flag_count + diff
-        post.save()
+    @property
+    def description(self):
+        cls = self.__dict__.get('_class', None)
+        return cls and cls.description or _("No description available")

 
 

-    def save(self, *args, **kwargs):
-        super(FlaggedItem, self).save(*args, **kwargs)
-        if self._is_new:
-            self._update_post_flag_count(1)
-
-    def cancel(self):
-        if not self.canceled:
-            self.canceled = True
-            self.save()
-            self._update_post_flag_count(-1)
-            
-
-class Comment(MetaContent, UserContent, DeletableContent):
-    comment        = models.CharField(max_length=300)
-    added_at       = models.DateTimeField(default=datetime.datetime.now)
-    score          = models.IntegerField(default=0)
-    liked_by       = models.ManyToManyField(User, through='LikedComment', related_name='comments_liked')
-
-    class Meta(MetaContent.Meta):
-        ordering = ('-added_at',)
-        db_table = u'comment'
-
-    def _update_post_comment_count(self, diff):
-        post = self.node.leaf
-        post.comment_count = post.comment_count + diff
-        post.save()
+    @models.permalink
+    def get_absolute_url(self):
+        return ('badge', [], {'id': self.id, 'slug': slugify(self.name)})

 
     def save(self, *args, **kwargs):
 
     def save(self, *args, **kwargs):

-        super(Comment,self).save(*args, **kwargs)
-
-        if self._is_new:
-            self._update_post_comment_count(1)
-
-        try:
-            ping_google()
-        except Exception:
-            logging.debug('problem pinging google did you register you sitemap with google?')
+        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)

 
 

-    def mark_deleted(self, user):
-        if super(Comment, self).mark_deleted(user):
-            self._update_post_comment_count(-1)
-
-    def unmark_deleted(self):
-        if super(Comment, self).unmark_deleted():
-            self._update_post_comment_count(1)
-
-    def is_reply_to(self, user):
-        inreply = re.search('@\w+', self.comment)
-        if inreply is not None:
-            return user.username.startswith(inreply.group(0))
-
-        return False
-
-    def __unicode__(self):
-        return self.comment
-
-
-class LikedComment(models.Model):
-    comment       = models.ForeignKey(Comment)
-    user          = models.ForeignKey(User)
-    added_at      = models.DateTimeField(default=datetime.datetime.now)
-    canceled      = models.BooleanField(default=False)
-
-    active = ActiveObjectManager()

 
     class Meta:
         app_label = 'forum'
 
 
     class Meta:
         app_label = 'forum'
 

-    def _update_comment_score(self, diff):
-        self.comment.score = self.comment.score + diff
-        self.comment.save()

 
 

-    def save(self, *args, **kwargs):
-        super(LikedComment, self).save(*args, **kwargs)
-        if self._is_new:
-            self._update_comment_score(1)
+class Award(models.Model):
+    user = models.ForeignKey(User)
+    badge = models.ForeignKey('Badge', related_name="awards")
+    node = models.ForeignKey(Node, null=True)

 
 

-    def cancel(self):
-        if not self.canceled:
-            self.canceled = True
-            self.save()
-            self._update_comment_score(-1)
+    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