Sith/forum/models.py

262 lines
9.2 KiB
Python
Raw Normal View History

# -*- coding:utf-8 -*
#
# Copyright 2016,2017
# - Skia <skia@libskia.so>
#
# Ce fichier fait partie du site de l'Association des Étudiants de l'UTBM,
# http://ae.utbm.fr.
#
# This program is free software; you can redistribute it and/or modify it under
# the terms of the GNU General Public License a published by the Free Software
# Foundation; either version 3 of the License, or (at your option) any later
# version.
#
# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
# FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
# details.
#
# You should have received a copy of the GNU General Public License along with
# this program; if not, write to the Free Sofware Foundation, Inc., 59 Temple
# Place - Suite 330, Boston, MA 02111-1307, USA.
#
#
2017-01-21 02:42:06 +00:00
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
2017-02-24 00:50:09 +00:00
from django.utils.functional import cached_property
2017-01-21 02:42:06 +00:00
2017-01-28 23:16:41 +00:00
from datetime import datetime
import pytz
2017-01-21 02:42:06 +00:00
from core.models import User, MetaGroup, Group, SithFile
from club.models import Club
2017-01-21 02:42:06 +00:00
class Forum(models.Model):
"""
The Forum class, made as a tree to allow nice tidy organization
2017-02-24 00:50:09 +00:00
owner_club allows club members to moderate there own topics
edit_groups allows to put any group as a forum admin
view_groups allows some groups to view a forum
2017-01-21 02:42:06 +00:00
"""
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_club = models.ForeignKey(Club, related_name="owned_forums", verbose_name=_("owner club"),
default=settings.SITH_MAIN_CLUB_ID)
edit_groups = models.ManyToManyField(Group, related_name="editable_forums", blank=True,
default=[settings.SITH_GROUP_OLD_SUBSCRIBERS_ID])
view_groups = models.ManyToManyField(Group, related_name="viewable_forums", blank=True,
default=[settings.SITH_GROUP_PUBLIC_ID])
2017-03-12 17:06:01 +00:00
number = models.IntegerField(_("number to choose a specific forum ordering"), default=1)
class Meta:
ordering = ['number']
def clean(self):
self.check_loop()
def save(self, *args, **kwargs):
copy_rights = False
if self.id is None:
copy_rights = True
super(Forum, self).save(*args, **kwargs)
if copy_rights:
self.copy_rights()
def apply_rights_recursively(self):
children = self.children.all()
for c in children:
c.copy_rights()
c.apply_rights_recursively()
def copy_rights(self):
"""Copy, if possible, the rights of the parent folder"""
if self.parent is not None:
self.owner_club = self.parent.owner_club
self.edit_groups = self.parent.edit_groups.all()
self.view_groups = self.parent.view_groups.all()
self.save()
def is_owned_by(self, user):
if user.is_in_group(settings.SITH_GROUP_FORUM_ADMIN_ID):
return True
m = self.owner_club.get_membership_for(user)
if m:
return m.role > settings.SITH_MAXIMUM_FREE_ROLE
return False
2017-01-21 02:42:06 +00:00
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 __str__(self):
return "%s" % (self.name)
def get_absolute_url(self):
return reverse('forum:view_forum', kwargs={'forum_id': self.id})
2017-02-24 00:50:09 +00:00
@cached_property
def parent_list(self):
return self.get_parent_list()
2017-01-21 02:42:06 +00:00
def get_parent_list(self):
l = []
p = self.parent
while p is not None:
l.append(p)
p = p.parent
return l
2017-02-24 00:50:09 +00:00
@cached_property
def topic_number(self):
return self.get_topic_number()
2017-01-21 21:47:30 +00:00
def get_topic_number(self):
number = self.topics.all().count()
for c in self.children.all():
2017-02-24 00:50:09 +00:00
number += c.topic_number
2017-01-21 21:47:30 +00:00
return number
2017-02-24 00:50:09 +00:00
@cached_property
def last_message(self):
return self.get_last_message()
2017-01-21 21:47:30 +00:00
def get_last_message(self):
last_msg = None
2017-02-24 00:50:09 +00:00
for m in ForumMessage.objects.select_related('topic__forum', 'author').order_by('-id'):
2017-01-21 21:47:30 +00:00
forum = m.topic.forum
if self in (forum.parent_list + [forum]) and not m.deleted:
2017-01-21 21:47:30 +00:00
return m
2017-01-21 02:42:06 +00:00
class ForumTopic(models.Model):
forum = models.ForeignKey(Forum, related_name='topics')
author = models.ForeignKey(User, related_name='forum_topics')
description = models.CharField(_('description'), max_length=256, default="")
class Meta:
2017-01-21 03:19:15 +00:00
ordering = ['-id'] # TODO: add date message ordering
def is_owned_by(self, user):
return self.forum.is_owned_by(user)
def can_be_edited_by(self, user):
return user.can_edit(self.forum)
def can_be_viewed_by(self, user):
return user.can_view(self.forum)
2017-01-21 03:19:15 +00:00
def __str__(self):
return "%s" % (self.title)
2017-01-21 02:42:06 +00:00
def get_absolute_url(self):
return reverse('forum:view_topic', kwargs={'topic_id': self.id})
2017-03-12 19:24:16 +00:00
def get_first_unread_message(self, user):
try:
msg = self.messages.exclude(readers=user).filter(date__gte=user.forum_infos.last_read_date).order_by('id').first()
return msg
except:
return None
@cached_property
def last_message(self):
for msg in self.messages.order_by('id').select_related('author').order_by('-id').all():
if not msg.deleted:
return msg
@property
def title(self):
return self.messages.order_by('date').first().title
2017-01-21 02:42:06 +00:00
class ForumMessage(models.Model):
"""
"A ForumMessage object represents a message in the forum" -- Cpt. Obvious
2017-01-21 02:42:06 +00:00
"""
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)
2017-01-28 23:16:41 +00:00
readers = models.ManyToManyField(User, related_name="read_messages", verbose_name=_("readers"))
2017-01-21 02:42:06 +00:00
class Meta:
2017-01-21 03:19:15 +00:00
ordering = ['id']
2017-01-21 02:42:06 +00:00
2017-01-28 23:16:41 +00:00
def __str__(self):
2017-02-24 00:50:09 +00:00
return "%s - %s" % (self.id, self.title)
2017-01-28 23:16:41 +00:00
2017-02-24 00:50:09 +00:00
def is_owned_by(self, user): # Anyone can create a topic: it's better to
# check the rights at the forum level, since it's more controlled
return self.topic.forum.is_owned_by(user) or user.id == self.author.id
def can_be_edited_by(self, user):
2017-02-24 00:50:09 +00:00
return user.can_edit(self.topic.forum)
def can_be_viewed_by(self, user):
return (not self.deleted and user.can_view(self.topic))
2017-02-24 00:50:05 +00:00
def can_be_moderated_by(self, user):
return self.topic.forum.is_owned_by(user) or user.id == self.author.id
2017-02-24 00:50:05 +00:00
2017-01-21 02:42:06 +00:00
def get_absolute_url(self):
2017-03-12 19:24:16 +00:00
return self.topic.get_absolute_url() + "?page=" + str(self.get_page()) + "#msg_" + str(self.id)
def get_page(self):
return int(self.topic.messages.filter(id__lt=self.id).count() / settings.SITH_FORUM_PAGE_LENGTH) + 1
2017-01-28 23:16:41 +00:00
def mark_as_read(self, user):
2017-02-24 16:22:13 +00:00
try: # Need the try/except because of AnonymousUser
self.readers.add(user)
except: pass
2017-01-28 23:16:41 +00:00
2017-02-24 00:50:09 +00:00
def is_read(self, user):
return (self.date < user.forum_infos.last_read_date) or (user in self.readers.all())
@cached_property
def deleted(self):
return self.is_deleted()
2017-02-24 00:50:05 +00:00
def is_deleted(self):
meta = self.metas.exclude(action="EDIT").order_by('-date').first()
if meta:
return meta.action == "DELETE"
return False
MESSAGE_META_ACTIONS = [
('EDIT', _("Message edited by")),
('DELETE', _("Message deleted by")),
('UNDELETE', _("Message undeleted by")),
]
class ForumMessageMeta(models.Model):
user = models.ForeignKey(User, related_name="forum_message_metas")
message = models.ForeignKey(ForumMessage, related_name="metas")
date = models.DateTimeField(_('date'), default=timezone.now)
action = models.CharField(_("action"), choices=MESSAGE_META_ACTIONS, max_length=16)
2017-01-28 23:16:41 +00:00
class ForumUserInfo(models.Model):
2017-02-24 00:50:09 +00:00
"""
This currently stores only the last date a user clicked "Mark all as read".
However, this can be extended with lot of user preferences dedicated to a
user, such as the favourite topics, the signature, and so on...
"""
user = models.OneToOneField(User, related_name="_forum_infos") # TODO: see to move that to the User class in order to reduce the number of db queries
2017-01-28 23:16:41 +00:00
last_read_date = models.DateTimeField(_('last read date'), default=datetime(year=settings.SITH_SCHOOL_START_YEAR,
month=1, day=1, tzinfo=pytz.UTC))
2017-01-21 02:42:06 +00:00