from django.db import models from django.core import validators from django.conf import settings from django.utils.translation import ugettext_lazy as _ from django.core.exceptions import ValidationError from django.db import IntegrityError, transaction from django.core.urlresolvers import reverse from django.utils import timezone from core.models import User, MetaGroup, Group, SithFile class Forum(models.Model): """ The Forum class, made as a tree to allow nice tidy organization """ name = models.CharField(_('name'), max_length=64) description = models.CharField(_('description'), max_length=256, default="") is_category = models.BooleanField(_('is a category'), default=False) parent = models.ForeignKey('Forum', related_name='children', null=True, blank=True) owner_group = models.ForeignKey(Group, related_name="owned_forums", default=settings.SITH_GROUP_COM_ADMIN_ID) edit_groups = models.ManyToManyField(Group, related_name="editable_forums", blank=True) view_groups = models.ManyToManyField(Group, related_name="viewable_forums", blank=True) def check_loop(self): """Raise a validation error when a loop is found within the parent list""" objs = [] cur = self while cur.parent is not None: if cur in objs: raise ValidationError(_('You can not make loops in forums')) objs.append(cur) cur = cur.parent def clean(self): self.check_loop() def __str__(self): return "%s" % (self.name) def get_absolute_url(self): return reverse('forum:view_forum', kwargs={'forum_id': self.id}) def get_parent_list(self): l = [] p = self.parent while p is not None: l.append(p) p = p.parent return l class ForumTopic(models.Model): forum = models.ForeignKey(Forum, related_name='topics') author = models.ForeignKey(User, related_name='forum_topics') title = models.CharField(_("title"), default="", max_length=64) description = models.CharField(_('description'), max_length=256, default="") class Meta: ordering = ['-id'] def get_absolute_url(self): return reverse('forum:view_topic', kwargs={'topic_id': self.id}) class ForumMessage(models.Model): """ "A ForumMessage object is a message in the forum" Cpt. Obvious """ topic = models.ForeignKey(ForumTopic, related_name='messages') author = models.ForeignKey(User, related_name='forum_messages') title = models.CharField(_("title"), default="", max_length=64, blank=True) message = models.TextField(_("message"), default="") date = models.DateTimeField(_('date'), default=timezone.now) class Meta: ordering = ['-id'] def get_absolute_url(self): return self.topic.get_absolute_url()