mirror of
https://github.com/ae-utbm/sith.git
synced 2024-11-18 04:03:22 +00:00
80 lines
2.8 KiB
Python
80 lines
2.8 KiB
Python
|
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()
|
||
|
|
|