X-Git-Url: https://git.openstreetmap.org./osqa.git/blobdiff_plain/133a8f939cdfa69dd6a7caeafed158aa38d20092..6006a10232650fcd622af648036f2e8260ed2e17:/forum/models/question.py diff --git a/forum/models/question.py b/forum/models/question.py index da8083f..ef4a37d 100644 --- a/forum/models/question.py +++ b/forum/models/question.py @@ -2,76 +2,49 @@ from base import * from tag import Tag from django.utils.translation import ugettext as _ -question_view = django.dispatch.Signal(providing_args=['instance', 'user']) +class QuestionManager(NodeManager): + def search(self, keywords): + return False, self.filter(models.Q(title__icontains=keywords) | models.Q(body__icontains=keywords)) -class Question(QandA): - accepted_answer = models.OneToOneField('Answer', null=True, related_name="question_accepting") - closed = models.BooleanField(default=False) - closed_by = models.ForeignKey(User, null=True, blank=True, related_name='closed_questions') - closed_at = models.DateTimeField(null=True, blank=True) - close_reason = models.SmallIntegerField(choices=CLOSE_REASONS, null=True, blank=True) - subscribers = models.ManyToManyField(User, related_name='subscriptions', through='QuestionSubscription') +class Question(Node): + class Meta(Node.Meta): + proxy = True + + answer_count = DenormalizedField("children", ~models.Q(state_string__contains="(deleted)"), node_type="answer") + accepted_count = DenormalizedField("children", ~models.Q(state_string__contains="(deleted)"), node_type="answer", marked=True) + favorite_count = DenormalizedField("actions", action_type="favorite", canceled=False) - # Denormalised data - answer_count = models.PositiveIntegerField(default=0) - view_count = models.IntegerField(default=0) - favourite_count = models.IntegerField(default=0) - last_activity_at = models.DateTimeField(default=datetime.datetime.now) - last_activity_by = models.ForeignKey(User, related_name='last_active_in_questions', null=True) + friendly_name = _("question") + objects = QuestionManager() - favorited_by = models.ManyToManyField(User, through='FavoriteQuestion', related_name='favorite_questions') + @property + def closed(self): + return self.nis.closed - class Meta(QandA.Meta): - db_table = u'question' + @property + def view_count(self): + return self.extra_count @property def headline(self): - if self.closed: - return _('[closed] ') + self.title - - if self.deleted: + if self.nis.deleted: return _('[deleted] ') + self.title + if self.nis.closed: + return _('[closed] ') + self.title + return self.title @property - def answer_accepted(self): - return self.accepted_answer is not None - - def save(self, *args, **kwargs): - if not self.last_activity_by: - self.last_activity_by = self.author - super(Question, self).save(*args, **kwargs) - - def update_last_activity(self, user): - self.last_activity_by = user - self.last_activity_at = datetime.datetime.now() - self.save() - - def activate_revision(self, user, revision): - super(Question, self).activate_revision(user, revision) - self.update_last_activity(user) + def accepted_answers(self): + return self.answers.filter(~models.Q(state_string__contains="(deleted)"), marked=True) @models.permalink def get_absolute_url(self): return ('question', (), {'id': self.id, 'slug': django_urlquote(slugify(self.title))}) - - def get_answer_count_by_user(self, user_id): - from answer import Answer - query_set = Answer.objects.filter(author__id=user_id) - return query_set.filter(question=self).count() - - def get_question_title(self): - if self.closed: - attr = CONST['closed'] - elif self.deleted: - attr = CONST['deleted'] - else: - attr = None - if attr is not None: - return u'%s %s' % (self.title, attr) - else: - return self.title + + def meta_description(self): + return self.summary def get_revision_url(self): return reverse('question_revisions', args=[self.id]) @@ -81,50 +54,32 @@ class Question(QandA): related_list = cache.get(cache_key) if related_list is None: - related_list = Question.objects.values('id').filter(tags__id__in=[t.id for t in self.tags.all()] - ).exclude(id=self.id).exclude(deleted=True).annotate(frequency=models.Count('id')).order_by('-frequency')[:count] + related_list = Question.objects.filter_state(deleted=False).values('id').filter(tags__id__in=[t.id for t in self.tags.all()] + ).exclude(id=self.id).annotate(frequency=models.Count('id')).order_by('-frequency')[:count] cache.set(cache_key, related_list, 60 * 60) return [Question.objects.get(id=r['id']) for r in related_list] + + def get_active_users(self): + active_users = set() + + active_users.add(self.author) + + for answer in self.answers: + active_users.add(answer.author) + + for comment in answer.comments: + active_users.add(comment.author) + + for comment in self.comments: + active_users.add(comment.author) + + return active_users - def __unicode__(self): - return self.title - -def question_viewed(instance, **kwargs): - instance.view_count += 1 - instance.save() - -question_view.connect(question_viewed) - -class FavoriteQuestion(models.Model): - question = models.ForeignKey('Question') - user = models.ForeignKey(User, related_name='user_favorite_questions') - added_at = models.DateTimeField(default=datetime.datetime.now) - - class Meta: - unique_together = ('question', 'user') - app_label = 'forum' - db_table = u'favorite_question' - - def __unicode__(self): - return '[%s] favorited at %s' %(self.user, self.added_at) - - def _update_question_fav_count(self, diff): - self.question.favourite_count = self.question.favourite_count + diff - self.question.save() - - def save(self, *args, **kwargs): - super(FavoriteQuestion, self).save(*args, **kwargs) - if self._is_new: - self._update_question_fav_count(1) - - def delete(self): - self._update_question_fav_count(-1) - super(FavoriteQuestion, self).delete() class QuestionSubscription(models.Model): user = models.ForeignKey(User) - question = models.ForeignKey(Question) + question = models.ForeignKey(Node) auto_subscription = models.BooleanField(default=True) last_view = models.DateTimeField(default=datetime.datetime.now())