mirror of
https://github.com/ae-utbm/sith.git
synced 2024-11-29 12:34:21 +00:00
972 lines
36 KiB
Python
972 lines
36 KiB
Python
from django.db import models
|
||
from django.core.mail import send_mail
|
||
from django.contrib.auth.models import AbstractBaseUser, PermissionsMixin, UserManager, Group as AuthGroup, GroupManager as AuthGroupManager, AnonymousUser as AuthAnonymousUser
|
||
from django.utils.translation import ugettext_lazy as _
|
||
from django.utils import timezone
|
||
from django.core import validators
|
||
from django.core.exceptions import ValidationError
|
||
from django.core.urlresolvers import reverse
|
||
from django.conf import settings
|
||
from django.db import transaction
|
||
from django.contrib.staticfiles.storage import staticfiles_storage
|
||
from django.utils.html import escape
|
||
from django.utils.functional import cached_property
|
||
|
||
import os
|
||
|
||
from phonenumber_field.modelfields import PhoneNumberField
|
||
|
||
from datetime import datetime, timedelta, date
|
||
|
||
import unicodedata
|
||
|
||
class RealGroupManager(AuthGroupManager):
|
||
def get_queryset(self):
|
||
return super(RealGroupManager, self).get_queryset().filter(is_meta=False)
|
||
|
||
class MetaGroupManager(AuthGroupManager):
|
||
def get_queryset(self):
|
||
return super(MetaGroupManager, self).get_queryset().filter(is_meta=True)
|
||
|
||
class Group(AuthGroup):
|
||
is_meta = models.BooleanField(
|
||
_('meta group status'),
|
||
default=False,
|
||
help_text=_('Whether a group is a meta group or not'),
|
||
)
|
||
description = models.CharField(_('description'), max_length=60)
|
||
|
||
class Meta:
|
||
ordering = ['name']
|
||
|
||
def get_absolute_url(self):
|
||
"""
|
||
This is needed for black magic powered UpdateView's children
|
||
"""
|
||
return reverse('core:group_list')
|
||
|
||
class MetaGroup(Group):
|
||
objects = MetaGroupManager()
|
||
class Meta:
|
||
proxy = True
|
||
|
||
def __init__(self, *args, **kwargs):
|
||
super(MetaGroup, self).__init__(*args, **kwargs)
|
||
self.is_meta = True
|
||
|
||
class RealGroup(Group):
|
||
objects = RealGroupManager()
|
||
class Meta:
|
||
proxy = True
|
||
|
||
def validate_promo(value):
|
||
start_year = settings.SITH_SCHOOL_START_YEAR
|
||
delta = (date.today()+timedelta(days=180)).year - start_year
|
||
if value < 0 or delta < value:
|
||
raise ValidationError(
|
||
_('%(value)s is not a valid promo (between 0 and %(end)s)'),
|
||
params={'value': value, 'end': delta},
|
||
)
|
||
|
||
class User(AbstractBaseUser):
|
||
"""
|
||
Defines the base user class, useable in every app
|
||
|
||
This is almost the same as the auth module AbstractUser since it inherits from it,
|
||
but some fields are required, and the username is generated automatically with the
|
||
name of the user (see generate_username()).
|
||
|
||
Added field: nick_name, date_of_birth
|
||
Required fields: email, first_name, last_name, date_of_birth
|
||
"""
|
||
username = models.CharField(
|
||
_('username'),
|
||
max_length=254,
|
||
unique=True,
|
||
help_text=_('Required. 254 characters or fewer. Letters, digits and ./+/-/_ only.'),
|
||
validators=[
|
||
validators.RegexValidator(
|
||
r'^[\w.+-]+$',
|
||
_('Enter a valid username. This value may contain only '
|
||
'letters, numbers ' 'and ./+/-/_ characters.')
|
||
),
|
||
],
|
||
error_messages={
|
||
'unique': _("A user with that username already exists."),
|
||
},
|
||
)
|
||
first_name = models.CharField(_('first name'), max_length=64)
|
||
last_name = models.CharField(_('last name'), max_length=64)
|
||
email = models.EmailField(_('email address'), unique=True)
|
||
date_of_birth = models.DateField(_('date of birth'), blank=True, null=True)
|
||
nick_name = models.CharField(_('nick name'), max_length=64, null=True, blank=True)
|
||
is_staff = models.BooleanField(
|
||
_('staff status'),
|
||
default=False,
|
||
help_text=_('Designates whether the user can log into this admin site.'),
|
||
)
|
||
is_active = models.BooleanField(
|
||
_('active'),
|
||
default=True,
|
||
help_text=_(
|
||
'Designates whether this user should be treated as active. '
|
||
'Unselect this instead of deleting accounts.'
|
||
),
|
||
)
|
||
date_joined = models.DateField(_('date joined'), auto_now_add=True)
|
||
last_update = models.DateTimeField(_('last update'), auto_now=True)
|
||
is_superuser = models.BooleanField(
|
||
_('superuser'),
|
||
default=False,
|
||
help_text=_(
|
||
'Designates whether this user is a superuser. '
|
||
),
|
||
)
|
||
groups = models.ManyToManyField(RealGroup, related_name='users', blank=True)
|
||
home = models.OneToOneField('SithFile', related_name='home_of', verbose_name=_("home"), null=True, blank=True,
|
||
on_delete=models.SET_NULL)
|
||
profile_pict = models.OneToOneField('SithFile', related_name='profile_of', verbose_name=_("profile"), null=True,
|
||
blank=True, on_delete=models.SET_NULL)
|
||
avatar_pict = models.OneToOneField('SithFile', related_name='avatar_of', verbose_name=_("avatar"), null=True,
|
||
blank=True, on_delete=models.SET_NULL)
|
||
scrub_pict = models.OneToOneField('SithFile', related_name='scrub_of', verbose_name=_("scrub"), null=True,
|
||
blank=True, on_delete=models.SET_NULL)
|
||
sex = models.CharField(_("sex"), max_length=10, choices=[("MAN", _("Man")), ("WOMAN", _("Woman"))], default="MAN")
|
||
tshirt_size = models.CharField(_("tshirt size"), max_length=5, choices=[
|
||
("-", _("-")),
|
||
("XS", _("XS")),
|
||
("S", _("S")),
|
||
("M", _("M")),
|
||
("L", _("L")),
|
||
("XL", _("XL")),
|
||
("XXL", _("XXL")),
|
||
("XXXL", _("XXXL")),
|
||
], default="-")
|
||
role = models.CharField(_("role"), max_length=15, choices=[
|
||
("STUDENT", _("Student")),
|
||
("ADMINISTRATIVE", _("Administrative agent")),
|
||
("TEACHER", _("Teacher")),
|
||
("AGENT", _("Agent")),
|
||
("DOCTOR", _("Doctor")),
|
||
("FORMER STUDENT", _("Former student")),
|
||
("SERVICE", _("Service")),
|
||
], blank=True, default="")
|
||
department = models.CharField(_("department"), max_length=15, choices=settings.SITH_PROFILE_DEPARTMENTS,
|
||
default="NA", blank=True)
|
||
dpt_option = models.CharField(_("dpt option"), max_length=32, blank=True, default="")
|
||
semester = models.CharField(_("semester"), max_length=5, blank=True, default="")
|
||
quote = models.CharField(_("quote"), max_length=256, blank=True, default="")
|
||
school = models.CharField(_("school"), max_length=80, blank=True, default="")
|
||
promo = models.IntegerField(_("promo"), validators=[validate_promo], null=True, blank=True)
|
||
forum_signature = models.TextField(_("forum signature"), max_length=256, blank=True, default="")
|
||
second_email = models.EmailField(_('second email address'), null=True, blank=True)
|
||
phone = PhoneNumberField(_("phone"), null=True, blank=True)
|
||
parent_phone = PhoneNumberField(_("parent phone"), null=True, blank=True)
|
||
address = models.CharField(_("address"), max_length=128, blank=True, default="")
|
||
parent_address = models.CharField(_("parent address"), max_length=128, blank=True, default="")
|
||
is_subscriber_viewable = models.BooleanField(_("is subscriber viewable"), default=True)
|
||
godfathers = models.ManyToManyField('User', related_name='godchildren', blank=True)
|
||
|
||
objects = UserManager()
|
||
|
||
USERNAME_FIELD = 'username'
|
||
# REQUIRED_FIELDS = ['email']
|
||
|
||
def has_module_perms(self, package_name):
|
||
return self.is_active
|
||
|
||
def has_perm(self, perm, obj=None):
|
||
return self.is_active and self.is_superuser
|
||
|
||
def get_absolute_url(self):
|
||
"""
|
||
This is needed for black magic powered UpdateView's children
|
||
"""
|
||
return reverse('core:user_profile', kwargs={'user_id': self.pk})
|
||
|
||
def __str__(self):
|
||
return self.get_display_name()
|
||
|
||
def to_dict(self):
|
||
return self.__dict__
|
||
|
||
@cached_property
|
||
def was_subscribed(self):
|
||
return self.subscriptions.exists()
|
||
|
||
@cached_property
|
||
def is_subscribed(self):
|
||
s = self.subscriptions.last()
|
||
return s.is_valid_now() if s is not None else False
|
||
|
||
def is_in_group(self, group_name):
|
||
"""If the user is in the group passed in argument (as string or by id)"""
|
||
group_id = 0
|
||
g = None
|
||
if isinstance(group_name, int): # Handle the case where group_name is an ID
|
||
g = Group.objects.filter(id=group_name).first()
|
||
else:
|
||
g = Group.objects.filter(name=group_name).first()
|
||
if g:
|
||
group_name = g.name
|
||
group_id = g.id
|
||
else:
|
||
return False
|
||
if group_id == settings.SITH_GROUP_PUBLIC_ID:
|
||
return True
|
||
if group_id == settings.SITH_GROUP_SUBSCRIBERS_ID:
|
||
return self.is_subscribed
|
||
if group_id == settings.SITH_GROUP_OLD_SUBSCRIBERS_ID:
|
||
return self.was_subscribed
|
||
if group_name == settings.SITH_MAIN_MEMBERS_GROUP: # We check the subscription if asked
|
||
return self.is_subscribed
|
||
if group_name[-len(settings.SITH_BOARD_SUFFIX):] == settings.SITH_BOARD_SUFFIX:
|
||
from club.models import Club
|
||
name = group_name[:-len(settings.SITH_BOARD_SUFFIX)]
|
||
c = Club.objects.filter(unix_name=name).first()
|
||
mem = c.get_membership_for(self)
|
||
if mem:
|
||
return mem.role > settings.SITH_MAXIMUM_FREE_ROLE
|
||
return False
|
||
if group_name[-len(settings.SITH_MEMBER_SUFFIX):] == settings.SITH_MEMBER_SUFFIX:
|
||
from club.models import Club
|
||
name = group_name[:-len(settings.SITH_MEMBER_SUFFIX)]
|
||
c = Club.objects.filter(unix_name=name).first()
|
||
mem = c.get_membership_for(self)
|
||
if mem:
|
||
return True
|
||
return False
|
||
if group_id == settings.SITH_GROUP_ROOT_ID and self.is_superuser:
|
||
return True
|
||
return self.groups.filter(name=group_name).exists()
|
||
|
||
@cached_property
|
||
def is_root(self):
|
||
return self.is_superuser or self.groups.filter(id=settings.SITH_GROUP_ROOT_ID).exists()
|
||
|
||
@cached_property
|
||
def is_board_member(self):
|
||
from club.models import Club
|
||
return Club.objects.filter(unix_name=settings.SITH_MAIN_CLUB['unix_name']).first().get_membership_for(self)
|
||
|
||
@cached_property
|
||
def is_launderette_manager(self):
|
||
from club.models import Club
|
||
return Club.objects.filter(unix_name=settings.SITH_LAUNDERETTE_MANAGER['unix_name']).first().get_membership_for(self)
|
||
|
||
@cached_property
|
||
def is_banned_alcohol(self):
|
||
return self.is_in_group(settings.SITH_GROUP_BANNED_ALCOHOL_ID)
|
||
|
||
@cached_property
|
||
def is_banned_counter(self):
|
||
return self.is_in_group(settings.SITH_GROUP_BANNED_COUNTER_ID)
|
||
|
||
def save(self, *args, **kwargs):
|
||
create = False
|
||
with transaction.atomic():
|
||
if self.id:
|
||
old = User.objects.filter(id=self.id).first()
|
||
if old and old.username != self.username:
|
||
self._change_username(self.username)
|
||
else:
|
||
create = True
|
||
super(User, self).save(*args, **kwargs)
|
||
if create and settings.IS_OLD_MYSQL_PRESENT: # Create user on the old site: TODO remove me!
|
||
import MySQLdb
|
||
try:
|
||
db = MySQLdb.connect(**settings.OLD_MYSQL_INFOS)
|
||
c = db.cursor()
|
||
c.execute("""INSERT INTO utilisateurs (id_utilisateur, nom_utl, prenom_utl, email_utl, hash_utl, ae_utl) VALUES
|
||
(%s, %s, %s, %s, %s, %s)""", (self.id, self.last_name, self.first_name, self.email, "valid", "0"))
|
||
db.commit()
|
||
except Exception as e:
|
||
with open(settings.BASE_DIR+"/user_fail.log", "a") as f:
|
||
print("FAIL to add user %s (%s %s - %s) to old site" % (self.id, self.first_name, self.last_name,
|
||
self.email), file=f)
|
||
print("Reason: %s" % (repr(e)), file=f)
|
||
db.rollback()
|
||
|
||
def make_home(self):
|
||
if self.home is None:
|
||
home_root = SithFile.objects.filter(parent=None, name="users").first()
|
||
if home_root is not None:
|
||
home = SithFile(parent=home_root, name=self.username, owner=self)
|
||
home.save()
|
||
self.home = home
|
||
self.save()
|
||
|
||
def _change_username(self, new_name):
|
||
u = User.objects.filter(username=new_name).first()
|
||
if u is None:
|
||
if self.home:
|
||
self.home.name = new_name
|
||
self.home.save()
|
||
else:
|
||
raise ValidationError(_("A user with that username already exists"))
|
||
|
||
def get_profile(self):
|
||
return {
|
||
"last_name": self.last_name,
|
||
"first_name": self.first_name,
|
||
"nick_name": self.nick_name,
|
||
"date_of_birth": self.date_of_birth,
|
||
}
|
||
|
||
def get_full_name(self):
|
||
"""
|
||
Returns the first_name plus the last_name, with a space in between.
|
||
"""
|
||
full_name = '%s %s' % (self.first_name, self.last_name)
|
||
return full_name.strip()
|
||
|
||
def get_short_name(self):
|
||
"Returns the short name for the user."
|
||
if self.nick_name:
|
||
return self.nick_name
|
||
return self.first_name + " " + self.last_name
|
||
|
||
def get_display_name(self):
|
||
"""
|
||
Returns the display name of the user.
|
||
A nickname if possible, otherwise, the full name
|
||
"""
|
||
if self.nick_name:
|
||
return "%s (%s)" % (self.get_full_name(), self.nick_name)
|
||
return self.get_full_name()
|
||
|
||
def get_age(self):
|
||
"""
|
||
Returns the age
|
||
"""
|
||
today = timezone.now()
|
||
born = self.date_of_birth
|
||
return today.year - born.year - ((today.month, today.day) < (born.month, born.day))
|
||
|
||
def email_user(self, subject, message, from_email=None, **kwargs):
|
||
"""
|
||
Sends an email to this User.
|
||
"""
|
||
if from_email is None:
|
||
from_email = settings.DEFAULT_FROM_EMAIL
|
||
send_mail(subject, message, from_email, [self.email], **kwargs)
|
||
|
||
def generate_username(self):
|
||
"""
|
||
Generates a unique username based on the first and last names.
|
||
For example: Guy Carlier gives gcarlier, and gcarlier1 if the first one exists
|
||
Returns the generated username
|
||
"""
|
||
def remove_accents(data):
|
||
return ''.join(x for x in unicodedata.normalize('NFKD', data) if \
|
||
unicodedata.category(x)[0] == 'L').lower()
|
||
user_name = remove_accents(self.first_name[0]+self.last_name).encode('ascii', 'ignore').decode('utf-8')
|
||
un_set = [u.username for u in User.objects.all()]
|
||
if user_name in un_set:
|
||
i = 1
|
||
while user_name+str(i) in un_set:
|
||
i += 1
|
||
user_name += str(i)
|
||
self.username = user_name
|
||
return user_name
|
||
|
||
def is_owner(self, obj):
|
||
"""
|
||
Determine if the object is owned by the user
|
||
"""
|
||
if hasattr(obj, "is_owned_by") and obj.is_owned_by(self):
|
||
return True
|
||
if hasattr(obj, "owner_group") and self.is_in_group(obj.owner_group.name):
|
||
return True
|
||
if self.is_superuser or self.is_in_group(settings.SITH_GROUP_ROOT_ID):
|
||
return True
|
||
return False
|
||
|
||
def can_edit(self, obj):
|
||
"""
|
||
Determine if the object can be edited by the user
|
||
"""
|
||
if hasattr(obj, "can_be_edited_by") and obj.can_be_edited_by(self):
|
||
return True
|
||
if hasattr(obj, "edit_groups"):
|
||
for g in obj.edit_groups.all():
|
||
if self.is_in_group(g.name):
|
||
return True
|
||
if isinstance(obj, User) and obj == self:
|
||
return True
|
||
if self.is_owner(obj):
|
||
return True
|
||
return False
|
||
|
||
def can_view(self, obj):
|
||
"""
|
||
Determine if the object can be viewed by the user
|
||
"""
|
||
if hasattr(obj, "can_be_viewed_by") and obj.can_be_viewed_by(self):
|
||
return True
|
||
if hasattr(obj, "view_groups"):
|
||
for g in obj.view_groups.all():
|
||
if self.is_in_group(g.name):
|
||
return True
|
||
if self.can_edit(obj):
|
||
return True
|
||
return False
|
||
|
||
def can_be_edited_by(self, user):
|
||
return user.is_in_group(settings.SITH_MAIN_BOARD_GROUP) or user.is_root
|
||
|
||
def can_be_viewed_by(self, user):
|
||
return (user.was_subscribed and self.is_subscriber_viewable) or user.is_root
|
||
|
||
def get_mini_item(self):
|
||
return """
|
||
<div class="mini_profile_link" >
|
||
<span>
|
||
<img src="%s" alt="%s" />
|
||
</span>
|
||
<em>%s</em>
|
||
</a>
|
||
""" % (
|
||
self.profile_pict.get_download_url() if self.profile_pict else staticfiles_storage.url("core/img/unknown.jpg"),
|
||
_("Profile"),
|
||
escape(self.get_display_name()),
|
||
)
|
||
|
||
@cached_property
|
||
def subscribed(self):
|
||
return self.is_in_group(settings.SITH_MAIN_MEMBERS_GROUP)
|
||
|
||
@cached_property
|
||
def forum_infos(self):
|
||
try:
|
||
return self._forum_infos
|
||
except:
|
||
from forum.models import ForumUserInfo
|
||
infos = ForumUserInfo(user=self)
|
||
infos.save()
|
||
return infos
|
||
|
||
class AnonymousUser(AuthAnonymousUser):
|
||
def __init__(self, request):
|
||
super(AnonymousUser, self).__init__()
|
||
|
||
@property
|
||
def was_subscribed(self):
|
||
return False
|
||
|
||
@property
|
||
def subscribed(self):
|
||
return False
|
||
|
||
@property
|
||
def is_root(self):
|
||
return False
|
||
|
||
@property
|
||
def is_board_member(self):
|
||
return False
|
||
|
||
@property
|
||
def is_launderette_manager(self):
|
||
return False
|
||
|
||
@property
|
||
def is_banned_alcohol(self):
|
||
return False
|
||
|
||
@property
|
||
def is_banned_conuter(self):
|
||
return False
|
||
|
||
def is_in_group(self, group_name):
|
||
"""
|
||
The anonymous user is only the public group
|
||
"""
|
||
group_id = 0
|
||
if isinstance(group_name, int): # Handle the case where group_name is an ID
|
||
g = Group.objects.filter(id=group_name).first()
|
||
if g:
|
||
group_name = g.name
|
||
group_id = g.id
|
||
else:
|
||
return False
|
||
if group_id == settings.SITH_GROUP_PUBLIC_ID:
|
||
return True
|
||
return False
|
||
|
||
def is_owner(self, obj):
|
||
return False
|
||
|
||
def can_edit(self, obj):
|
||
return False
|
||
|
||
def can_view(self, obj):
|
||
if hasattr(obj, 'view_groups') and obj.view_groups.filter(id=settings.SITH_GROUP_PUBLIC_ID).exists():
|
||
return True
|
||
if hasattr(obj, 'can_be_viewed_by') and obj.can_be_viewed_by(self):
|
||
return True
|
||
return False
|
||
|
||
def get_display_name(self):
|
||
return _("Visitor")
|
||
|
||
class Preferences(models.Model):
|
||
user = models.OneToOneField(User, related_name="preferences")
|
||
receive_weekmail = models.BooleanField(
|
||
_('do you want to receive the weekmail'),
|
||
default=False,
|
||
# help_text=_('Do you want to receive the weekmail?'),
|
||
)
|
||
show_my_stats = models.BooleanField(
|
||
_('define if we show a users stats'),
|
||
default=False,
|
||
help_text=_('Show your account statistics to others'),
|
||
)
|
||
|
||
def get_display_name(self):
|
||
return self.user.get_display_name()
|
||
|
||
def get_absolute_url(self):
|
||
return self.user.get_absolute_url()
|
||
|
||
def get_directory(instance, filename):
|
||
return '.{0}/{1}'.format(instance.get_parent_path(), filename)
|
||
|
||
def get_compressed_directory(instance, filename):
|
||
return '.{0}/compressed/{1}'.format(instance.get_parent_path(), filename)
|
||
|
||
def get_thumbnail_directory(instance, filename):
|
||
return '.{0}/thumbnail/{1}'.format(instance.get_parent_path(), filename)
|
||
|
||
class SithFile(models.Model):
|
||
name = models.CharField(_('file name'), max_length=256, blank=False)
|
||
parent = models.ForeignKey('self', related_name="children", verbose_name=_("parent"), null=True, blank=True)
|
||
file = models.FileField(upload_to=get_directory, verbose_name=_("file"), max_length=256, null=True, blank=True)
|
||
compressed = models.FileField(upload_to=get_compressed_directory, verbose_name=_("compressed file"), max_length=256, null=True, blank=True)
|
||
thumbnail = models.FileField(upload_to=get_thumbnail_directory, verbose_name=_("thumbnail"), max_length=256, null=True, blank=True)
|
||
owner = models.ForeignKey(User, related_name="owned_files", verbose_name=_("owner"))
|
||
edit_groups = models.ManyToManyField(Group, related_name="editable_files", verbose_name=_("edit group"), blank=True)
|
||
view_groups = models.ManyToManyField(Group, related_name="viewable_files", verbose_name=_("view group"), blank=True)
|
||
is_folder = models.BooleanField(_("is folder"), default=True)
|
||
mime_type = models.CharField(_('mime type'), max_length=30)
|
||
size = models.IntegerField(_("size"), default=0)
|
||
date = models.DateTimeField(_('date'), default=timezone.now)
|
||
is_moderated = models.BooleanField(_("is moderated"), default=False)
|
||
moderator = models.ForeignKey(User, related_name="moderated_files", verbose_name=_("owner"), null=True, blank=True)
|
||
asked_for_removal = models.BooleanField(_("asked for removal"), default=False)
|
||
is_in_sas = models.BooleanField(_("is in the SAS"), default=False) # Allows to query this flag, updated at each call to save()
|
||
|
||
class Meta:
|
||
verbose_name = _("file")
|
||
|
||
def is_owned_by(self, user):
|
||
if hasattr(self, 'profile_of') and user.is_in_group(settings.SITH_MAIN_BOARD_GROUP):
|
||
return True
|
||
if user.is_in_group(settings.SITH_GROUP_COM_ADMIN_ID):
|
||
return True
|
||
if self.is_in_sas and user.is_in_group(settings.SITH_GROUP_SAS_ADMIN_ID):
|
||
return True
|
||
return user.id == self.owner.id
|
||
|
||
def can_be_viewed_by(self, user):
|
||
if hasattr(self, 'profile_of'):
|
||
return user.can_view(self.profile_of)
|
||
if hasattr(self, 'avatar_of'):
|
||
return user.can_view(self.avatar_of)
|
||
if hasattr(self, 'scrub_of'):
|
||
return user.can_view(self.scrub_of)
|
||
return False
|
||
|
||
def delete(self):
|
||
for c in self.children.all():
|
||
c.delete()
|
||
self.file.delete()
|
||
if self.compressed:
|
||
self.compressed.delete()
|
||
if self.thumbnail:
|
||
self.thumbnail.delete()
|
||
return super(SithFile, self).delete()
|
||
|
||
def clean(self):
|
||
"""
|
||
Cleans up the file
|
||
"""
|
||
super(SithFile, self).clean()
|
||
if '/' in self.name:
|
||
raise ValidationError(_("Character '/' not authorized in name"))
|
||
if self == self.parent:
|
||
raise ValidationError(
|
||
_('Loop in folder tree'),
|
||
code='loop',
|
||
)
|
||
if (self == self.parent or (self.parent is not None and self in self.get_parent_list())):
|
||
raise ValidationError(
|
||
_('Loop in folder tree'),
|
||
code='loop',
|
||
)
|
||
if self.parent and self.parent.is_file:
|
||
raise ValidationError(_('You can not make a file be a children of a non folder file'))
|
||
if ((self.parent is None and SithFile.objects.exclude(id=self.id).filter(parent=None, name=self.name).exists()) or
|
||
(self.parent and self.parent.children.exclude(id=self.id).filter(name=self.name).exists())):
|
||
raise ValidationError(
|
||
_('Duplicate file'),
|
||
code='duplicate',
|
||
)
|
||
if self.is_folder:
|
||
if self.file:
|
||
try:
|
||
import imghdr
|
||
if imghdr.what(None, self.file.read()) not in ['gif', 'png', 'jpeg']:
|
||
self.file.delete()
|
||
self.file = None
|
||
except:
|
||
self.file = None
|
||
self.mime_type = "inode/directory"
|
||
if self.is_file and (self.file is None or self.file == ""):
|
||
raise ValidationError(_("You must provide a file"))
|
||
|
||
def save(self, *args, **kwargs):
|
||
sas = SithFile.objects.filter(id=settings.SITH_SAS_ROOT_DIR_ID).first()
|
||
self.is_in_sas = sas in self.get_parent_list()
|
||
copy_rights = False
|
||
if self.id is None:
|
||
copy_rights = True
|
||
super(SithFile, self).save(*args, **kwargs)
|
||
if copy_rights:
|
||
self.copy_rights()
|
||
|
||
def apply_rights_recursively(self, only_folders=False):
|
||
children = self.children.all()
|
||
if only_folders:
|
||
children = children.filter(is_folder=True)
|
||
for c in children:
|
||
c.copy_rights()
|
||
c.apply_rights_recursively(only_folders)
|
||
|
||
def copy_rights(self):
|
||
"""Copy, if possible, the rights of the parent folder"""
|
||
if self.parent is not None:
|
||
self.edit_groups = self.parent.edit_groups.all()
|
||
self.view_groups = self.parent.view_groups.all()
|
||
self.save()
|
||
|
||
def move_to(self, parent):
|
||
"""Move a file to somewhere else"""
|
||
if not parent.is_folder:
|
||
return
|
||
import shutil
|
||
import os
|
||
with transaction.atomic():
|
||
if self.is_folder:
|
||
old_file_name = self.get_full_path()
|
||
else:
|
||
old_file_name = self.file.name
|
||
self.parent = parent
|
||
self.save()
|
||
if self.is_folder:
|
||
for c in self.children.all():
|
||
c.move_to(self)
|
||
shutil.rmtree(os.path.join(settings.MEDIA_ROOT, old_file_name))
|
||
else:
|
||
self.file.save(name=self.name, content=self.file)
|
||
os.remove(os.path.join(settings.MEDIA_ROOT, old_file_name))
|
||
|
||
def __getattribute__(self, attr):
|
||
if attr == "is_file":
|
||
return not self.is_folder
|
||
else:
|
||
return super(SithFile, self).__getattribute__(attr)
|
||
|
||
@cached_property
|
||
def as_picture(self):
|
||
from sas.models import Picture
|
||
return Picture.objects.filter(id=self.id).first()
|
||
|
||
@cached_property
|
||
def as_album(self):
|
||
from sas.models import Album
|
||
return Album.objects.filter(id=self.id).first()
|
||
|
||
def __str__(self):
|
||
if self.is_folder:
|
||
return _("Folder: ") + self.name
|
||
else:
|
||
return _("File: ") + self.name
|
||
|
||
def get_parent_list(self):
|
||
l = []
|
||
p = self.parent
|
||
while p is not None:
|
||
l.append(p)
|
||
p = p.parent
|
||
return l
|
||
|
||
def get_parent_path(self):
|
||
return '/' + '/'.join([p.name for p in self.get_parent_list()[::-1]])
|
||
|
||
def get_full_path(self):
|
||
return self.get_parent_path() + '/' + self.name
|
||
|
||
def get_display_name(self):
|
||
return self.name
|
||
|
||
def get_download_url(self):
|
||
return reverse('core:download', kwargs={'file_id': self.id})
|
||
|
||
def __str__(self):
|
||
return self.get_parent_path() + "/" + self.name
|
||
|
||
class LockError(Exception):
|
||
"""There was a lock error on the object"""
|
||
pass
|
||
|
||
class AlreadyLocked(LockError):
|
||
"""The object is already locked"""
|
||
pass
|
||
|
||
class NotLocked(LockError):
|
||
"""The object is not locked"""
|
||
pass
|
||
|
||
class Page(models.Model):
|
||
"""
|
||
The page class to build a Wiki
|
||
Each page may have a parent and it's URL is of the form my.site/page/<grd_pa>/<parent>/<mypage>
|
||
It has an ID field, but don't use it, since it's only there for DB part, and because compound primary key is
|
||
awkward!
|
||
Prefere querying pages with Page.get_page_by_full_name()
|
||
|
||
Be careful with the _full_name attribute: this field may not be valid until you call save(). It's made for fast
|
||
query, but don't rely on it when playing with a Page object, use get_full_name() instead!
|
||
"""
|
||
name = models.CharField(_('page unix name'), max_length=30,
|
||
validators=[
|
||
validators.RegexValidator(
|
||
r'^[A-z.+-]+$',
|
||
_('Enter a valid page name. This value may contain only '
|
||
'unaccented letters, numbers ' 'and ./+/-/_ characters.')
|
||
),
|
||
],
|
||
blank=False)
|
||
parent = models.ForeignKey('self', related_name="children", verbose_name=_("parent"), null=True, blank=True, on_delete=models.SET_NULL)
|
||
# Attention: this field may not be valid until you call save(). It's made for fast query, but don't rely on it when
|
||
# playing with a Page object, use get_full_name() instead!
|
||
_full_name = models.CharField(_('page name'), max_length=255, blank=True)
|
||
owner_group = models.ForeignKey(Group, related_name="owned_page", verbose_name=_("owner group"),
|
||
default=settings.SITH_GROUP_ROOT_ID)
|
||
edit_groups = models.ManyToManyField(Group, related_name="editable_page", verbose_name=_("edit group"), blank=True)
|
||
view_groups = models.ManyToManyField(Group, related_name="viewable_page", verbose_name=_("view group"), blank=True)
|
||
lock_user = models.ForeignKey(User, related_name="locked_pages", verbose_name=_("lock user"), blank=True, null=True, default=None)
|
||
lock_timeout = models.DateTimeField(_('lock_timeout'), null=True, blank=True, default=None)
|
||
|
||
class Meta:
|
||
unique_together = ('name', 'parent')
|
||
permissions = (
|
||
("change_prop_page", "Can change the page's properties (groups, ...)"),
|
||
("view_page", "Can view the page"),
|
||
)
|
||
|
||
@staticmethod
|
||
def get_page_by_full_name(name):
|
||
"""
|
||
Quicker to get a page with that method rather than building the request every time
|
||
"""
|
||
return Page.objects.filter(_full_name=name).first()
|
||
|
||
def __init__(self, *args, **kwargs):
|
||
super(Page, self).__init__(*args, **kwargs)
|
||
|
||
def clean(self):
|
||
"""
|
||
Cleans up only the name for the moment, but this can be used to make any treatment before saving the object
|
||
"""
|
||
if '/' in self.name:
|
||
self.name = self.name.split('/')[-1]
|
||
if Page.objects.exclude(pk=self.pk).filter(_full_name=self.get_full_name()).exists():
|
||
raise ValidationError(
|
||
_('Duplicate page'),
|
||
code='duplicate',
|
||
)
|
||
super(Page, self).clean()
|
||
if self.parent is not None and self in self.get_parent_list():
|
||
raise ValidationError(
|
||
_('Loop in page tree'),
|
||
code='loop',
|
||
)
|
||
|
||
def get_parent_list(self):
|
||
l = []
|
||
p = self.parent
|
||
while p is not None:
|
||
l.append(p)
|
||
p = p.parent
|
||
return l
|
||
|
||
def save(self, *args, **kwargs):
|
||
"""
|
||
Performs some needed actions before and after saving a page in database
|
||
"""
|
||
locked = kwargs.pop('force_lock', False)
|
||
if not locked: locked = self.is_locked()
|
||
if not locked:
|
||
raise NotLocked("The page is not locked and thus can not be saved")
|
||
self.full_clean()
|
||
if not self.id:
|
||
super(Page, self).save(*args, **kwargs) # Save a first time to correctly set _full_name
|
||
# This reset the _full_name just before saving to maintain a coherent field quicker for queries than the
|
||
# recursive method
|
||
# It also update all the children to maintain correct names
|
||
self._full_name = self.get_full_name()
|
||
for c in self.children.all():
|
||
c.save()
|
||
super(Page, self).save(*args, **kwargs)
|
||
self.unset_lock()
|
||
|
||
def is_locked(self):
|
||
"""
|
||
Is True if the page is locked, False otherwise
|
||
This is where the timeout is handled, so a locked page for which the timeout is reach will be unlocked and this
|
||
function will return False
|
||
"""
|
||
if self.lock_timeout and (timezone.now() - self.lock_timeout > timedelta(minutes=5)):
|
||
# print("Lock timed out")
|
||
self.unset_lock()
|
||
return self.lock_user and self.lock_timeout and (timezone.now() - self.lock_timeout < timedelta(minutes=5))
|
||
|
||
def set_lock(self, user):
|
||
"""
|
||
Sets a lock on the current page or raise an AlreadyLocked exception
|
||
"""
|
||
if self.is_locked() and self.get_lock() != user:
|
||
raise AlreadyLocked("The page is already locked by someone else")
|
||
self.lock_user = user
|
||
self.lock_timeout = timezone.now()
|
||
super(Page, self).save()
|
||
# print("Locking page")
|
||
|
||
def set_lock_recursive(self, user):
|
||
"""
|
||
Locks recursively all the child pages for editing properties
|
||
"""
|
||
for p in self.children.all():
|
||
p.set_lock_recursive(user)
|
||
self.set_lock(user)
|
||
|
||
def unset_lock_recursive(self):
|
||
"""
|
||
Unlocks recursively all the child pages
|
||
"""
|
||
for p in self.children.all():
|
||
p.unset_lock_recursive()
|
||
self.unset_lock()
|
||
|
||
def unset_lock(self):
|
||
"""Always try to unlock, even if there is no lock"""
|
||
self.lock_user = None
|
||
self.lock_timeout = None
|
||
super(Page, self).save()
|
||
# print("Unlocking page")
|
||
|
||
def get_lock(self):
|
||
"""
|
||
Returns the page's mutex containing the time and the user in a dict
|
||
"""
|
||
if self.lock_user:
|
||
return self.lock_user
|
||
raise NotLocked("The page is not locked and thus can not return its user")
|
||
|
||
def get_absolute_url(self):
|
||
"""
|
||
This is needed for black magic powered UpdateView's children
|
||
"""
|
||
return reverse('core:page', kwargs={'page_name': self._full_name})
|
||
|
||
def __str__(self):
|
||
return self.get_full_name()
|
||
|
||
def get_full_name(self):
|
||
"""
|
||
Computes the real full_name of the page based on its name and its parent's name
|
||
You can and must rely on this function when working on a page object that is not freshly fetched from the DB
|
||
(For example when treating a Page object coming from a form)
|
||
"""
|
||
if self.parent is None:
|
||
return self.name
|
||
return '/'.join([self.parent.get_full_name(), self.name])
|
||
|
||
def get_display_name(self):
|
||
try:
|
||
return self.revisions.last().title
|
||
except:
|
||
return self.name
|
||
|
||
def delete(self):
|
||
self.unset_lock_recursive()
|
||
self.set_lock_recursive(User.objects.get(id=0))
|
||
for child in self.children.all():
|
||
child.parent = self.parent
|
||
child.save()
|
||
child.unset_lock_recursive()
|
||
super(Page, self).delete()
|
||
|
||
|
||
class PageRev(models.Model):
|
||
"""
|
||
This is the true content of the page.
|
||
Each page object has a revisions field that is a list of PageRev, ordered by date.
|
||
my_page.revisions.last() gives the PageRev object that is the most up-to-date, and thus,
|
||
is the real content of the page.
|
||
The content is in PageRev.title and PageRev.content .
|
||
"""
|
||
revision = models.IntegerField(_("revision"))
|
||
title = models.CharField(_("page title"), max_length=255, blank=True)
|
||
content = models.TextField(_("page content"), blank=True)
|
||
date = models.DateTimeField(_('date'), auto_now=True)
|
||
author = models.ForeignKey(User, related_name='page_rev')
|
||
page = models.ForeignKey(Page, related_name='revisions')
|
||
|
||
class Meta:
|
||
ordering = ['date',]
|
||
|
||
def get_absolute_url(self):
|
||
"""
|
||
This is needed for black magic powered UpdateView's children
|
||
"""
|
||
return reverse('core:page', kwargs={'page_name': self.page._full_name})
|
||
|
||
def __str__(self):
|
||
return str(self.__dict__)
|
||
|
||
def __getattribute__(self, attr):
|
||
if attr == "owner_group":
|
||
return self.page.owner_group
|
||
elif attr == "edit_groups":
|
||
return self.page.edit_groups
|
||
elif attr == "view_groups":
|
||
return self.page.view_groups
|
||
elif attr == "unset_lock":
|
||
return self.page.unset_lock
|
||
else:
|
||
return object.__getattribute__(self, attr)
|
||
|
||
def save(self, *args, **kwargs):
|
||
if self.revision is None:
|
||
self.revision = self.page.revisions.all().count() + 1
|
||
super(PageRev, self).save(*args, **kwargs)
|
||
# Don't forget to unlock, otherwise, people will have to wait for the page's timeout
|
||
self.page.unset_lock()
|
||
|
||
class Notification(models.Model):
|
||
user = models.ForeignKey(User, related_name='notifications')
|
||
url = models.CharField(_("url"), max_length=255)
|
||
param = models.CharField(_("param"), max_length=128, default="")
|
||
type = models.CharField(_("type"), max_length=32, choices=settings.SITH_NOTIFICATIONS, default="GENERIC")
|
||
date = models.DateTimeField(_('date'), default=timezone.now)
|
||
viewed = models.BooleanField(_('viewed'), default=False)
|
||
|
||
def __str__(self):
|
||
if self.param:
|
||
return self.get_type_display() % self.param
|
||
return self.get_type_display()
|
||
|