X-Git-Url: https://git.openstreetmap.org./osqa.git/blobdiff_plain/d77834bde5bef508cf0180bdf90aa62a36671c17..b6f6103a8b578d13047219ef3812cea09001eea2:/forum/models/meta.py diff --git a/forum/models/meta.py b/forum/models/meta.py index 538d6a2..fca763f 100644 --- a/forum/models/meta.py +++ b/forum/models/meta.py @@ -1,152 +1,99 @@ -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'), - ) - - vote = models.SmallIntegerField(choices=VOTE_CHOICES) - voted_at = models.DateTimeField(default=datetime.datetime.now) - canceled = models.BooleanField(default=False) - - active = ActiveObjectManager() - - class Meta(MetaContent.Meta): - db_table = u'vote' - - def __unicode__(self): - return '[%s] voted at %s: %s' %(self.user, self.voted_at, self.vote) - - def _update_post_vote_count(self, diff): - post = self.content_object - field = self.vote == 1 and 'vote_up_count' or 'vote_down_count' - post.__dict__[field] = post.__dict__[field] + diff - post.save() - - def save(self, *args, **kwargs): - super(Vote, self).save(*args, **kwargs) - if self._is_new: - self._update_post_vote_count(1) - - 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) - - def is_upvote(self): - return self.vote == self.VOTE_UP - - def is_downvote(self): - return self.vote == self.VOTE_DOWN - -vote_canceled = django.dispatch.Signal(providing_args=['instance']) - -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) - - active = ActiveObjectManager() - - class Meta(MetaContent.Meta): - db_table = u'flagged_item' - - def __unicode__(self): - return '[%s] flagged at %s' %(self.user, self.flagged_at) - - def _update_post_flag_count(self, diff): - post = self.content_object - post.offensive_flag_count = post.offensive_flag_count + diff - post.save() - - 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.content_object - post.comment_count = post.comment_count + diff - post.save() - - 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?') - - 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' - - 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) - - def cancel(self): - if not self.canceled: - self.canceled = True - self.save() - self._update_comment_score(-1) - - +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 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) + + 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 + + +class BadgeManager(models.Manager): + use_for_related_fields = True + + def get_query_set(self): + return BadgesQuerySet(self.model) + +class Badge(models.Model): + 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') + + objects = BadgeManager() + + @property + def name(self): + cls = self.__dict__.get('_class', None) + return cls and cls.name or _("Unknown") + + @property + def description(self): + cls = self.__dict__.get('_class', None) + 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