mirror of
https://github.com/ae-utbm/sith3.git
synced 2024-06-29 09:58:04 +00:00
65 lines
2.1 KiB
Python
65 lines
2.1 KiB
Python
from django.contrib.auth.forms import UserCreationForm, AuthenticationForm
|
|
from django import forms
|
|
from django.contrib.auth import logout, login, authenticate
|
|
import logging
|
|
|
|
from .models import User, Page
|
|
|
|
class RegisteringForm(UserCreationForm):
|
|
error_css_class = 'error'
|
|
required_css_class = 'required'
|
|
class Meta:
|
|
model = User
|
|
fields = ('first_name', 'last_name', 'email', 'date_of_birth')
|
|
|
|
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
|
|
|
|
class LoginForm(AuthenticationForm):
|
|
def login(self):
|
|
u = authenticate(username=self.request.POST['username'],
|
|
password=self.request.POST['password'])
|
|
if u is not None:
|
|
if u.is_active:
|
|
login(self.request, u)
|
|
logging.debug("Logging in "+str(u))
|
|
else:
|
|
raise forms.ValidationError(
|
|
self.error_messages['invalid_login'],
|
|
code='inactive',
|
|
params={'username': self.username_field.verbose_name},
|
|
)
|
|
else:
|
|
logging.debug("Login failed")
|
|
raise forms.ValidationError(
|
|
self.error_messages['invalid_login'],
|
|
code='invalid_login',
|
|
params={'username': self.username_field.verbose_name},
|
|
)
|
|
|
|
class PageForm(forms.ModelForm):
|
|
error_css_class = 'error'
|
|
required_css_class = 'required'
|
|
parent = forms.ModelChoiceField(queryset=Page.objects.all())
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(PageForm, self).__init__(*args, **kwargs)
|
|
self.fields['parent'].required = False
|
|
|
|
class Meta:
|
|
model = Page
|
|
fields = ['parent', 'name', 'title', 'content', ]
|
|
|
|
def save(self, commit=True):
|
|
page = super(PageForm, self).save(commit=False)
|
|
if commit:
|
|
page.save()
|
|
return page
|
|
|
|
|