mirror of
https://github.com/chibicitiberiu/ytsm.git
synced 2024-02-24 05:43:31 +00:00
46 lines
1.7 KiB
Python
46 lines
1.7 KiB
Python
|
from crispy_forms.helper import FormHelper
|
||
|
|
from crispy_forms.layout import Submit
|
||
|
|
from django import forms
|
||
|
|
from django.contrib.auth.forms import AuthenticationForm, UserCreationForm
|
||
|
|
from django.urls import reverse_lazy
|
||
|
|
|
||
|
|
|
||
|
|
class ExtendedAuthenticationForm(AuthenticationForm):
|
||
|
|
remember_me = forms.BooleanField(label='Remember me', required=False, initial=False)
|
||
|
|
|
||
|
|
def clean(self):
|
||
|
|
remember_me = self.cleaned_data.get('remember_me')
|
||
|
|
if remember_me:
|
||
|
|
expiry = 3600 * 24 * 30
|
||
|
|
else:
|
||
|
|
expiry = 0
|
||
|
|
self.request.session.set_expiry(expiry)
|
||
|
|
|
||
|
|
return super().clean()
|
||
|
|
|
||
|
|
|
||
|
|
class ExtendedUserCreationForm(UserCreationForm):
|
||
|
|
email = forms.EmailField(required=False,
|
||
|
|
label='E-mail address',
|
||
|
|
help_text='The e-mail address is optional, but it is the only way to recover a lost '
|
||
|
|
'password.')
|
||
|
|
first_name = forms.CharField(max_length=30, required=False,
|
||
|
|
label='First name')
|
||
|
|
last_name = forms.CharField(max_length=150, required=False,
|
||
|
|
label='Last name')
|
||
|
|
|
||
|
|
form_action = reverse_lazy('register')
|
||
|
|
|
||
|
|
def __init__(self, *args, **kwargs):
|
||
|
|
super().__init__(*args, **kwargs)
|
||
|
|
self.helper = FormHelper()
|
||
|
|
self.helper.label_class = 'col-3'
|
||
|
|
self.helper.field_class = 'col-9'
|
||
|
|
self.helper.form_class = 'form-horizontal'
|
||
|
|
self.helper.form_method = 'post'
|
||
|
|
self.helper.form_action = self.form_action
|
||
|
|
self.helper.add_input(Submit('submit', 'register'))
|
||
|
|
|
||
|
|
class Meta(UserCreationForm.Meta):
|
||
|
|
fields = ['username', 'email', 'first_name', 'last_name']
|