61 lines
2.4 KiB
Python
61 lines
2.4 KiB
Python
from django import forms
|
|
from django.contrib.auth.forms import ReadOnlyPasswordHashField
|
|
from django.contrib.auth.forms import ReadOnlyPasswordHashField
|
|
from django.core.exceptions import ValidationError
|
|
|
|
from account.models import MyUser
|
|
|
|
|
|
class UserCreationForm(forms.ModelForm):
|
|
"""A form for creating new users. Includes all the required
|
|
fields, plus a repeated password."""
|
|
password1 = forms.CharField(label='Пароль', widget=forms.PasswordInput)
|
|
password2 = forms.CharField(label='Пароль ещё раз', widget=forms.PasswordInput)
|
|
|
|
class Meta:
|
|
model = MyUser
|
|
fields = ('email',)
|
|
|
|
def clean_password2(self):
|
|
# Check that the two password entries match
|
|
password1 = self.cleaned_data.get("password1")
|
|
password2 = self.cleaned_data.get("password2")
|
|
if password1 and password2 and password1 != password2:
|
|
raise ValidationError("Пароли не совпадают")
|
|
return password2
|
|
|
|
def save(self, commit=True):
|
|
# Save the provided password in hashed format
|
|
user = super().save(commit=False)
|
|
user.set_password(self.cleaned_data["password1"])
|
|
if commit:
|
|
user.save()
|
|
return user
|
|
|
|
|
|
class UserChangeForm(forms.ModelForm):
|
|
"""A form for updating users. Includes all the fields on
|
|
the user, but replaces the password field with admin's
|
|
disabled password hash display field.
|
|
"""
|
|
password = ReadOnlyPasswordHashField(
|
|
label='Пароль',
|
|
help_text='Пароли не хранятся в открытом виде, поэтому нет возможности увидеть пароль этого пользователя, '
|
|
'но вы можете изменить пароль, воспользовавшись '
|
|
'<a href="{}">этой формой</a>.',
|
|
)
|
|
|
|
class Meta:
|
|
model = MyUser
|
|
fields = ('email', 'password', 'is_active', 'is_staff', 'is_superuser')
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
password = self.fields.get('password')
|
|
if password:
|
|
password.help_text = password.help_text.format('../password/')
|
|
user_permissions = self.fields.get('user_permissions')
|
|
if user_permissions:
|
|
user_permissions.queryset = user_permissions.queryset.select_related('content_type')
|
|
|