from urlparse import urlparse from django.core.urlresolvers import reverse from django.db import models from django.utils.translation import ugettext_lazy as _ from mptt.managers import TreeManager from mptt.models import MPTTModel, TreeForeignKey from misago.acl import version as acl_version from misago.acl.models import BaseRole from misago.conf import settings from misago.core.cache import cache from misago.core.utils import slugify CACHE_NAME = 'misago_forums_tree' FORUMS_TREE_ID = 1 class ForumManager(TreeManager): def private_threads(self): return self.get(special_role='private_threads') def root_category(self): return self.get(special_role='root_category') def all_forums(self, include_root=False): qs = self.filter(tree_id=FORUMS_TREE_ID) if not include_root: qs = self.filter(lft__gt=3) return qs.order_by('lft') def get_cached_forums_dict(self): forums_dict = cache.get(CACHE_NAME, 'nada') if forums_dict == 'nada': forums_dict = self.get_forums_dict_from_db() cache.set(CACHE_NAME, forums_dict) return forums_dict def get_forums_dict_from_db(self): forums_dict = {} for forum in self.all_forums(include_root=True): forums_dict[forum.pk] = forum return forums_dict def clear_cache(self): cache.delete(CACHE_NAME) class Forum(MPTTModel): parent = TreeForeignKey( 'self', null=True, blank=True, related_name='children') special_role = models.CharField(max_length=255, null=True, blank=True) role = models.CharField(max_length=255, null=True, blank=True) name = models.CharField(max_length=255) slug = models.CharField(max_length=255) description = models.TextField(null=True, blank=True) is_closed = models.BooleanField(default=False) redirect_url = models.CharField(max_length=255, null=True, blank=True) redirects = models.PositiveIntegerField(default=0) threads = models.PositiveIntegerField(default=0) posts = models.PositiveIntegerField(default=0) last_post_on = models.DateTimeField(null=True, blank=True) last_thread = models.ForeignKey('misago_threads.Thread', related_name='+', null=True, blank=True, on_delete=models.SET_NULL) last_thread_title = models.CharField(max_length=255, null=True, blank=True) last_thread_slug = models.CharField(max_length=255, null=True, blank=True) last_poster = models.ForeignKey(settings.AUTH_USER_MODEL, related_name='+', 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) prune_started_after = models.PositiveIntegerField(default=0) prune_replied_after = models.PositiveIntegerField(default=0) archive_pruned_in = models.ForeignKey('self', related_name='pruned_archive', null=True, blank=True, on_delete=models.SET_NULL) css_class = models.CharField(max_length=255, null=True, blank=True) objects = ForumManager() def __unicode__(self): if self.special_role == 'root_category': return unicode(_('None (will become top level category)')) elif self.special_role == 'private_threads': return unicode(_('Private Threads')) else: return self.name def save(self, *args, **kwargs): if self.pk: acl_version.invalidate() return super(Forum, self).save(*args, **kwargs) def delete(self, *args, **kwargs): Forum.objects.clear_cache() acl_version.invalidate() return super(Forum, self).delete(*args, **kwargs) def synchronize(self): counted_criteria = {'is_hidden':False, 'is_moderated':False} self.threads = self.thread_set.filter(**counted_criteria).count() self.posts = self.post_set.filter(**counted_criteria).count() if self.threads: last_thread_qs = self.thread_set.filter(**counted_criteria) last_thread = last_thread_qs.order_by('-last_post_id')[:1][0] self.set_last_thread(last_thread) else: self.empty_last_thread() def delete_content(self): from misago.forums.signals import delete_forum_content delete_forum_content.send(sender=self) def move_content(self, new_forum): from misago.forums.signals import move_forum_content move_forum_content.send(sender=self, new_forum=new_forum) @property def redirect_host(self): return urlparse(self.redirect_url).hostname def get_absolute_url(self): if not self.special_role: if self.level == 1: formats = (reverse('misago:index'), self.slug, self.id) return '%s#%s-%s' % formats else: return reverse('misago:%s' % self.role, kwargs={ 'forum_id': self.id, 'forum_slug': self.slug }) else: return None def set_name(self, name): self.name = name self.slug = slugify(name) def set_last_thread(self, thread): self.last_post_on = thread.last_post_on self.last_thread = thread self.last_thread_title = thread.title self.last_thread_slug = thread.slug self.last_poster = thread.last_poster self.last_poster_name = thread.last_poster_name self.last_poster_slug = thread.last_poster_slug def empty_last_thread(self): self.last_post_on = None self.last_thread = None self.last_thread_title = None self.last_thread_slug = None self.last_poster = None self.last_poster_name = None self.last_poster_slug = None def has_child(self, child): return child.lft > self.lft and child.rght < self.rght class ForumRole(BaseRole): pass class RoleForumACL(models.Model): role = models.ForeignKey('misago_acl.Role', related_name='forums_acls') forum = models.ForeignKey('Forum', related_name='forum_role_set') forum_role = models.ForeignKey(ForumRole)