mirror of
https://github.com/ae-utbm/sith.git
synced 2024-11-18 04:03:22 +00:00
217 lines
8.6 KiB
Python
217 lines
8.6 KiB
Python
from django.contrib.auth.forms import UserCreationForm, AuthenticationForm, UserChangeForm
|
|
from django import forms
|
|
from django.db import transaction
|
|
from django.core.exceptions import ValidationError
|
|
from django.contrib.auth import logout, login, authenticate
|
|
from django.forms import CheckboxSelectMultiple, Select, DateInput, TextInput, DateTimeInput
|
|
from django.utils.translation import ugettext as _
|
|
from phonenumber_field.widgets import PhoneNumberInternationalFallbackWidget
|
|
|
|
import logging
|
|
import re
|
|
|
|
from core.models import User, Page, RealGroup, SithFile
|
|
|
|
|
|
# Widgets
|
|
|
|
class SelectSingle(Select):
|
|
def render(self, name, value, attrs=None):
|
|
if attrs:
|
|
attrs['class'] = "select_single"
|
|
else:
|
|
attrs = {'class': "select_single"}
|
|
return super(SelectSingle, self).render(name, value, attrs)
|
|
|
|
class SelectMultiple(Select):
|
|
def render(self, name, value, attrs=None):
|
|
if attrs:
|
|
attrs['class'] = "select_multiple"
|
|
else:
|
|
attrs = {'class': "select_multiple"}
|
|
return super(SelectMultiple, self).render(name, value, attrs)
|
|
|
|
class SelectDateTime(DateTimeInput):
|
|
def render(self, name, value, attrs=None):
|
|
if attrs:
|
|
attrs['class'] = "select_datetime"
|
|
else:
|
|
attrs = {'class': "select_datetime"}
|
|
return super(SelectDateTime, self).render(name, value, attrs)
|
|
|
|
class SelectDate(DateInput):
|
|
def render(self, name, value, attrs=None):
|
|
if attrs:
|
|
attrs['class'] = "select_date"
|
|
else:
|
|
attrs = {'class': "select_date"}
|
|
return super(SelectDate, self).render(name, value, attrs)
|
|
|
|
class SelectFile(TextInput):
|
|
def render(self, name, value, attrs=None):
|
|
if attrs:
|
|
attrs['class'] = "select_file"
|
|
else:
|
|
attrs = {'class': "select_file"}
|
|
output = '%(content)s<div name="%(name)s" class="choose_file_widget" title="%(title)s"></div>' % {
|
|
'content': super(SelectFile, self).render(name, value, attrs),
|
|
'title': _("Choose file"),
|
|
'name': name,
|
|
}
|
|
output += '<span name="' + name + '" class="choose_file_button">' + _("Choose file") + '</span>'
|
|
return output
|
|
|
|
class SelectUser(TextInput):
|
|
def render(self, name, value, attrs=None):
|
|
if attrs:
|
|
attrs['class'] = "select_user"
|
|
else:
|
|
attrs = {'class': "select_user"}
|
|
output = '%(content)s<div name="%(name)s" class="choose_user_widget" title="%(title)s"></div>' % {
|
|
'content': super(SelectUser, self).render(name, value, attrs),
|
|
'title': _("Choose user"),
|
|
'name': name,
|
|
}
|
|
output += '<span name="' + name + '" class="choose_user_button">' + _("Choose user") + '</span>'
|
|
return output
|
|
|
|
# Forms
|
|
|
|
class LoginForm(AuthenticationForm):
|
|
def __init__(self, *arg, **kwargs):
|
|
if 'data' in kwargs.keys():
|
|
from counter.models import Customer
|
|
data = kwargs['data'].copy()
|
|
account_code = re.compile(r"^[0-9]+[A-Za-z]$")
|
|
try:
|
|
if account_code.match(data['username']):
|
|
user = Customer.objects.filter(account_id__iexact=data['username']).first().user
|
|
elif '@' in data['username']:
|
|
user = User.objects.filter(email__iexact=data['username']).first()
|
|
else:
|
|
user = User.objects.filter(username=data['username']).first()
|
|
data['username'] = user.username
|
|
except: pass
|
|
kwargs['data'] = data
|
|
super(LoginForm, self).__init__(*arg, **kwargs)
|
|
self.fields['username'].label = _("Username, email, or account number")
|
|
|
|
class RegisteringForm(UserCreationForm):
|
|
error_css_class = 'error'
|
|
required_css_class = 'required'
|
|
class Meta:
|
|
model = User
|
|
fields = ('first_name', 'last_name', 'email')
|
|
|
|
def save(self, commit=True):
|
|
user = super(RegisteringForm, self).save(commit=False)
|
|
user.set_password(self.cleaned_data["password1"])
|
|
user.generate_username()
|
|
if commit:
|
|
user.save()
|
|
return user
|
|
|
|
from core.utils import resize_image
|
|
from io import BytesIO
|
|
from PIL import Image
|
|
|
|
class UserProfileForm(forms.ModelForm):
|
|
"""
|
|
Form handling the user profile, managing the files
|
|
This form is actually pretty bad and was made in the rush before the migration. It should be refactored.
|
|
TODO: refactor this form
|
|
"""
|
|
class Meta:
|
|
model = User
|
|
fields = ['first_name', 'last_name', 'nick_name', 'email', 'date_of_birth', 'profile_pict', 'avatar_pict',
|
|
'scrub_pict', 'sex', 'second_email', 'address', 'parent_address', 'phone', 'parent_phone',
|
|
'tshirt_size', 'role', 'department', 'dpt_option', 'semester', 'quote', 'school', 'promo',
|
|
'forum_signature', 'is_subscriber_viewable']
|
|
widgets = {
|
|
'date_of_birth': SelectDate,
|
|
'profile_pict': forms.ClearableFileInput,
|
|
'avatar_pict': forms.ClearableFileInput,
|
|
'scrub_pict': forms.ClearableFileInput,
|
|
'phone': PhoneNumberInternationalFallbackWidget,
|
|
'parent_phone': PhoneNumberInternationalFallbackWidget,
|
|
}
|
|
labels = {
|
|
'profile_pict': _("Profile: you need to be visible on the picture, in order to be recognized (e.g. by the barmen)"),
|
|
'avatar_pict': _("Avatar: used on the forum"),
|
|
'scrub_pict': _("Scrub: let other know how your scrub looks like!"),
|
|
}
|
|
|
|
def __init__(self, *arg, **kwargs):
|
|
super(UserProfileForm, self).__init__(*arg, **kwargs)
|
|
|
|
def full_clean(self):
|
|
super(UserProfileForm, self).full_clean()
|
|
|
|
def generate_name(self, field_name, f):
|
|
field_name = field_name[:-4]
|
|
return field_name + str(self.instance.id) + "." + f.content_type.split('/')[-1]
|
|
|
|
def process(self, files):
|
|
avatar = self.instance.avatar_pict
|
|
profile = self.instance.profile_pict
|
|
scrub = self.instance.scrub_pict
|
|
self.full_clean()
|
|
self.cleaned_data['avatar_pict'] = avatar
|
|
self.cleaned_data['profile_pict'] = profile
|
|
self.cleaned_data['scrub_pict'] = scrub
|
|
parent = SithFile.objects.filter(parent=None, name="profiles").first()
|
|
for field,f in files:
|
|
with transaction.atomic():
|
|
try:
|
|
im = Image.open(BytesIO(f.read()))
|
|
new_file = SithFile(parent=parent, name=self.generate_name(field, f),
|
|
file=resize_image(im, 400, f.content_type.split('/')[-1]),
|
|
owner=self.instance, is_folder=False, mime_type=f.content_type, size=f._size)
|
|
new_file.file.name = new_file.name
|
|
old = SithFile.objects.filter(parent=parent, name=new_file.name).first()
|
|
if old:
|
|
old.delete()
|
|
new_file.clean()
|
|
new_file.save()
|
|
self.cleaned_data[field] = new_file
|
|
self._errors.pop(field, None)
|
|
except ValidationError as e:
|
|
self._errors.pop(field, None)
|
|
self.add_error(field, _("Error uploading file %(file_name)s: %(msg)s") %
|
|
{'file_name': f, 'msg': str(e.message)})
|
|
except IOError:
|
|
self._errors.pop(field, None)
|
|
self.add_error(field, _("Error uploading file %(file_name)s: %(msg)s") %
|
|
{'file_name': f, 'msg': _("Bad image format, only jpeg, png, and gif are accepted")})
|
|
self._post_clean()
|
|
|
|
class UserPropForm(forms.ModelForm):
|
|
error_css_class = 'error'
|
|
required_css_class = 'required'
|
|
class Meta:
|
|
model = User
|
|
fields = ['groups']
|
|
help_texts = {
|
|
'groups': "Which groups this user belongs to",
|
|
}
|
|
widgets = {
|
|
'groups': CheckboxSelectMultiple,
|
|
}
|
|
|
|
class PagePropForm(forms.ModelForm):
|
|
error_css_class = 'error'
|
|
required_css_class = 'required'
|
|
class Meta:
|
|
model = Page
|
|
fields = ['parent', 'name', 'owner_group', 'edit_groups', 'view_groups', ]
|
|
widgets = {
|
|
'edit_groups': CheckboxSelectMultiple,
|
|
'view_groups': CheckboxSelectMultiple,
|
|
}
|
|
|
|
def __init__(self, *arg, **kwargs):
|
|
super(PagePropForm, self).__init__(*arg, **kwargs)
|
|
self.fields['edit_groups'].required = False
|
|
self.fields['view_groups'].required = False
|
|
|