from django.core.urlresolvers import reverse from django.db import models, transaction from django.dispatch import receiver from misago.conf import settings from misago.core.shortcuts import paginate from misago.core.utils import slugify __all__ = [ 'Thread', 'ThreadParticipant' ] class PrivateThreadMixin(object): pass class Thread(models.Model, PrivateThreadMixin): forum = models.ForeignKey('misago_forums.Forum') label = models.ForeignKey('misago_threads.Label', null=True, blank=True, on_delete=models.SET_NULL) title = models.CharField(max_length=255) slug = models.CharField(max_length=255) replies = models.PositiveIntegerField(default=0, db_index=True) has_reported_posts = models.BooleanField(default=False) has_moderated_posts = models.BooleanField(default=False) has_hidden_posts = models.BooleanField(default=False) has_events = models.BooleanField(default=False) started_on = models.DateTimeField(db_index=True) first_post = models.ForeignKey('misago_threads.Post', related_name='+', null=True, blank=True, on_delete=models.SET_NULL) starter = models.ForeignKey(settings.AUTH_USER_MODEL, null=True, blank=True, on_delete=models.SET_NULL) starter_name = models.CharField(max_length=255) starter_slug = models.CharField(max_length=255) last_post_on = models.DateTimeField(db_index=True) last_post = models.ForeignKey('misago_threads.Post', related_name='+', null=True, blank=True, on_delete=models.SET_NULL) last_poster = models.ForeignKey(settings.AUTH_USER_MODEL, related_name='last_poster_set', null=True, blank=True, on_delete=models.SET_NULL) last_poster_name = models.CharField(max_length=255, null=True, blank=True) last_poster_slug = models.CharField(max_length=255, null=True, blank=True) is_pinned = models.BooleanField(default=False, db_index=True) is_poll = models.BooleanField(default=False) is_moderated = models.BooleanField(default=False, db_index=True) is_hidden = models.BooleanField(default=False) is_closed = models.BooleanField(default=False) participants = models.ManyToManyField(settings.AUTH_USER_MODEL, related_name='private_thread_set', through='ThreadParticipant', through_fields=('thread', 'user')) class Meta: index_together = [ ['forum', 'id'], ['forum', 'last_post_on'], ['forum', 'replies'], ] def __unicode__(self): return self.title def lock(self): return Thread.objects.select_for_update().get(id=self.id) def delete(self, *args, **kwargs): from misago.threads.signals import delete_thread delete_thread.send(sender=self) super(Thread, self).delete(*args, **kwargs) def merge(self, other_thread): if self.pk == other_thread.pk: raise ValueError("thread can't be merged with itself") from misago.threads.signals import merge_thread merge_thread.send(sender=self, other_thread=other_thread) def move(self, new_forum): from misago.threads.signals import move_thread self.forum = new_forum move_thread.send(sender=self) def synchronize(self): self.replies = self.post_set.filter(is_moderated=False).count() if self.replies > 0: self.replies -= 1 reported_post_qs = self.post_set.filter(is_reported=True) self.has_reported_posts = reported_post_qs.exists() moderated_post_qs = self.post_set.filter(is_moderated=True) self.has_moderated_posts = moderated_post_qs.exists() hidden_post_qs = self.post_set.filter(is_hidden=True)[:1] self.has_hidden_posts = hidden_post_qs.exists() self.has_events = self.event_set.exists() first_post = self.post_set.order_by('id')[:1][0] self.set_first_post(first_post) self.is_moderated = first_post.is_moderated self.is_hidden = first_post.is_hidden last_post_qs = self.post_set.filter(is_moderated=False).order_by('-id') last_post = last_post_qs[:1] if last_post: self.set_last_post(last_post[0]) else: self.set_last_post(first_post) @property def thread_type(self): return self.forum.thread_type def get_absolute_url(self): return self.thread_type.get_thread_absolute_url(self) def get_post_url(self, post, page): try: post_id = post.id except AttributeError: post_id = post return self.thread_type.get_thread_post_url(self, post_id, page) def get_last_reply_url(self): return self.thread_type.get_thread_last_reply_url(self) def get_new_reply_url(self): return self.thread_type.get_thread_new_reply_url(self) def get_moderated_url(self): return self.thread_type.get_thread_moderated_url(self) def get_reported_url(self): return self.thread_type.get_thread_reported_url(self) def get_reply_api_url(self): return self.thread_type.get_reply_url(self) def set_title(self, title): self.title = title self.slug = slugify(title) def set_first_post(self, post): self.started_on = post.posted_on self.first_post = post self.starter = post.poster self.starter_name = post.poster_name if post.poster: self.starter_slug = post.poster.slug else: self.starter_slug = slugify(post.poster_name) self.is_moderated = post.is_moderated self.is_hidden = post.is_hidden def set_last_post(self, post): self.last_post_on = post.posted_on self.last_post = post self.last_poster = post.poster self.last_poster_name = post.poster_name if post.poster: self.last_poster_slug = post.poster.slug else: self.last_poster_slug = slugify(post.poster_name) class ThreadParticipantManager(models.Manager): def remove_participant(self, thread, user): ThreadParticipant.objects.filter(thread=thread, user=user).delete() @transaction.atomic def set_owner(self, thread, user): thread_owner = ThreadParticipant.objects.filter( thread=thread, is_owner=True) thread_owner.update(is_owner=False) self.remove_participant(thread, user) ThreadParticipant.objects.create( thread=thread, user=user, is_owner=True) @transaction.atomic def add_participant(self, thread, user): self.remove_participant(thread, user) ThreadParticipant.objects.create( thread=thread, user=user,) class ThreadParticipant(models.Model): thread = models.ForeignKey(Thread) user = models.ForeignKey(settings.AUTH_USER_MODEL) is_owner = models.BooleanField(default=False) objects = ThreadParticipantManager()