kpsul/gestion/forms.py
2017-06-23 00:10:38 +01:00

66 lines
1.5 KiB
Python

from django import forms
from django.contrib.auth.models import User
from .models import Profile
# ---
# Profile edition
# ---
class UserForm(forms.ModelForm):
class Meta:
model = User
fields = ["first_name", "last_name"]
class ProfileForm(forms.ModelForm):
class Meta:
model = Profile
fields = ["phone", "departement"]
# ---
# Registration
# ---
class UserRegistrationForm(forms.ModelForm):
passwd1 = forms.CharField(
label="Mot de passe",
widget=forms.PasswordInput,
required=False,
)
passwd2 = forms.CharField(
label="Confirmation de mot de passe",
widget=forms.PasswordInput,
required=False,
)
def clean_passwd2(self):
passwd1 = self.cleaned_data["passwd1"]
passwd2 = self.cleaned_data["passwd2"]
if passwd1 and passwd2:
if passwd1 != passwd2:
raise forms.ValidationError("Mots de passes différents")
return passwd2
def save(self, commit=True, *args, **kwargs):
user = super().save(commit, *args, **kwargs)
if self.cleaned_data["passwd2"]:
user.set_password(self.cleaned_data["passwd2"])
if commit:
user.save()
return user
class Meta:
model = User
fields = ["username", "first_name", "last_name", "email"]
class ProfileRegistrationForm(forms.ModelForm):
class Meta:
model = Profile
fields = [
"login_clipper", "phone", "occupation", "departement", "comments"
]