mirror of
https://gitlab2.federez.net/re2o/re2o
synced 2024-12-25 00:13:45 +00:00
178 lines
5.6 KiB
Python
178 lines
5.6 KiB
Python
from django.db import models
|
|
from django.forms import ModelForm, Form
|
|
from django import forms
|
|
|
|
from django.utils import timezone
|
|
|
|
from topologie.models import Room
|
|
|
|
def remove_user_room(room):
|
|
""" Déménage de force l'ancien locataire de la chambre """
|
|
try:
|
|
user = User.objects.get(room=room)
|
|
except User.DoesNotExist:
|
|
return
|
|
user.room = None
|
|
user.save()
|
|
|
|
class User(models.Model):
|
|
STATE_ACTIVE = 0
|
|
STATE_DEACTIVATED = 1
|
|
STATE_ARCHIVED = 2
|
|
STATES = (
|
|
(0, 'STATE_ACTIVE'),
|
|
(1, 'STATE_DEACTIVATED'),
|
|
(2, 'STATE_ARCHIVED'),
|
|
)
|
|
|
|
name = models.CharField(max_length=255)
|
|
surname = models.CharField(max_length=255)
|
|
pseudo = models.CharField(max_length=255, unique=True)
|
|
email = models.EmailField()
|
|
school = models.ForeignKey('School', on_delete=models.PROTECT)
|
|
comment = models.CharField(help_text="Commentaire, promo", max_length=255, blank=True)
|
|
room = models.OneToOneField('topologie.Room', on_delete=models.PROTECT, blank=True, null=True)
|
|
pwd_ssha = models.CharField(max_length=255)
|
|
pwd_ntlm = models.CharField(max_length=255)
|
|
state = models.IntegerField(choices=STATES, default=STATE_ACTIVE)
|
|
registered = models.DateTimeField(auto_now_add=True)
|
|
|
|
def __str__(self):
|
|
return self.pseudo
|
|
|
|
class Right(models.Model):
|
|
user = models.ForeignKey('User', on_delete=models.PROTECT)
|
|
right = models.ForeignKey('ListRight', on_delete=models.PROTECT)
|
|
|
|
class Meta:
|
|
unique_together = ("user", "right")
|
|
|
|
def __str__(self):
|
|
return str(self.user) + " - " + str(self.right)
|
|
|
|
class School(models.Model):
|
|
name = models.CharField(max_length=255)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
class ListRight(models.Model):
|
|
listright = models.CharField(max_length=255)
|
|
|
|
def __str__(self):
|
|
return self.listright
|
|
|
|
class Ban(models.Model):
|
|
user = models.ForeignKey('User', on_delete=models.PROTECT)
|
|
raison = models.CharField(max_length=255)
|
|
date_start = models.DateTimeField(auto_now_add=True)
|
|
date_end = models.DateTimeField(help_text='%m/%d/%y %H:%M:%S')
|
|
|
|
def __str__(self):
|
|
return str(self.user) + ' ' + str(self.raison)
|
|
|
|
class Whitelist(models.Model):
|
|
user = models.ForeignKey('User', on_delete=models.PROTECT)
|
|
raison = models.CharField(max_length=255)
|
|
date_start = models.DateTimeField(auto_now_add=True)
|
|
date_end = models.DateTimeField(help_text='%m/%d/%y %H:%M:%S')
|
|
|
|
def __str__(self):
|
|
return str(self.user) + ' ' + str(self.raison)
|
|
|
|
class UserForm(ModelForm):
|
|
def __init__(self, *args, **kwargs):
|
|
super(InfoForm, self).__init__(*args, **kwargs)
|
|
self.fields['name'].label = 'Nom'
|
|
self.fields['surname'].label = 'Prénom'
|
|
self.fields['school'].label = 'Établissement'
|
|
self.fields['comment'].label = 'Commentaire'
|
|
self.fields['room'].label = 'Chambre'
|
|
|
|
class Meta:
|
|
model = User
|
|
fields = '__all__'
|
|
|
|
class InfoForm(ModelForm):
|
|
force = forms.BooleanField(label="Forcer le déménagement ?", initial=False, required=False)
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(InfoForm, self).__init__(*args, **kwargs)
|
|
self.fields['name'].label = 'Nom'
|
|
self.fields['surname'].label = 'Prénom'
|
|
self.fields['school'].label = 'Établissement'
|
|
self.fields['comment'].label = 'Commentaire'
|
|
self.fields['room'].label = 'Chambre'
|
|
|
|
def clean_force(self):
|
|
if self.cleaned_data.get('force', False):
|
|
remove_user_room(self.cleaned_data.get('room'))
|
|
return
|
|
|
|
class Meta:
|
|
model = User
|
|
fields = ['name','surname','pseudo','email', 'school', 'comment', 'room']
|
|
|
|
class PasswordForm(ModelForm):
|
|
class Meta:
|
|
model = User
|
|
fields = ['pwd_ssha','pwd_ntlm']
|
|
|
|
class StateForm(ModelForm):
|
|
class Meta:
|
|
model = User
|
|
fields = ['state']
|
|
|
|
class SchoolForm(ModelForm):
|
|
class Meta:
|
|
model = School
|
|
fields = ['name']
|
|
|
|
class RightForm(ModelForm):
|
|
def __init__(self, *args, **kwargs):
|
|
super(RightForm, self).__init__(*args, **kwargs)
|
|
self.fields['right'].label = 'Droit'
|
|
|
|
class Meta:
|
|
model = Right
|
|
fields = ['right']
|
|
|
|
class DelRightForm(ModelForm):
|
|
rights = forms.ModelMultipleChoiceField(queryset=Right.objects.all(), label="Droits actuels", widget=forms.CheckboxSelectMultiple)
|
|
|
|
class Meta:
|
|
model = Right
|
|
exclude = ['user', 'right']
|
|
|
|
class BanForm(ModelForm):
|
|
def __init__(self, *args, **kwargs):
|
|
super(BanForm, self).__init__(*args, **kwargs)
|
|
self.fields['date_end'].label = 'Date de fin'
|
|
|
|
class Meta:
|
|
model = Ban
|
|
exclude = ['user']
|
|
|
|
def clean_date_end(self):
|
|
date_end = self.cleaned_data['date_end']
|
|
if date_end < timezone.now():
|
|
raise forms.ValidationError("Triple buse, la date de fin ne peut pas être avant maintenant... Re2o ne voyage pas dans le temps")
|
|
return date_end
|
|
|
|
class WhitelistForm(ModelForm):
|
|
def __init__(self, *args, **kwargs):
|
|
super(WhitelistForm, self).__init__(*args, **kwargs)
|
|
self.fields['date_end'].label = 'Date de fin'
|
|
|
|
class Meta:
|
|
model = Whitelist
|
|
exclude = ['user']
|
|
|
|
def clean_date_end(self):
|
|
date_end = self.cleaned_data['date_end']
|
|
if date_end < timezone.now():
|
|
raise forms.ValidationError("Triple buse, la date de fin ne peut pas être avant maintenant... Re2o ne voyage pas dans le temps")
|
|
return date_end
|
|
|
|
class ProfilForm(Form):
|
|
user =forms.CharField(label ='Ok', max_length=100)
|