mirror of
https://github.com/TrentSPalmer/trentpalmerdotorg.git
synced 2024-11-24 02:11:30 -08:00
107 lines
3.6 KiB
Python
107 lines
3.6 KiB
Python
from django.contrib.auth.forms import ValidationError, UsernameField, UserCreationForm
|
|
from django.contrib.auth.models import User
|
|
from django import forms
|
|
from .models import Account, EmailWhiteList, twitter_handle_validator
|
|
|
|
|
|
class EnableTotpForm(forms.ModelForm):
|
|
|
|
totp_code = forms.CharField(max_length=6)
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.fields['totp_code'].widget.attrs.update({'autofocus': 'autofocus'})
|
|
|
|
class Meta:
|
|
model = Account
|
|
fields = ("totp_code", )
|
|
|
|
|
|
class OurUserCreationForm(UserCreationForm):
|
|
email = forms.EmailField(
|
|
required=True,
|
|
label='Email',
|
|
max_length=254,
|
|
widget=forms.EmailInput(attrs={'autocomplete': 'email'})
|
|
)
|
|
|
|
class Meta:
|
|
model = User
|
|
fields = ("username", "email", "password1", "password2")
|
|
|
|
def save(self, commit=True):
|
|
user = super(OurUserCreationForm, self).save(commit=False)
|
|
user.email = self.cleaned_data["email"]
|
|
if commit:
|
|
user.save()
|
|
return user
|
|
|
|
def clean(self):
|
|
email = self.cleaned_data.get('email')
|
|
if not EmailWhiteList.objects.filter(email=email).exists():
|
|
raise ValidationError("Email Not Authorized, try another.")
|
|
if User.objects.filter(email=email).exists():
|
|
raise ValidationError("An account already exists with this email address.")
|
|
username = self.cleaned_data.get('username')
|
|
if User.objects.filter(username=username).exists():
|
|
raise ValidationError("Try a different username, that one already exists.")
|
|
return self.cleaned_data
|
|
|
|
|
|
class EditProfileForm(forms.Form):
|
|
email = forms.EmailField(
|
|
required=True,
|
|
label='Email',
|
|
max_length=254,
|
|
widget=forms.EmailInput(attrs={'autocomplete': 'email'})
|
|
)
|
|
|
|
first_name = UsernameField(required=False)
|
|
last_name = UsernameField(required=False)
|
|
twitter_handle = forms.CharField(max_length=64, validators=[twitter_handle_validator])
|
|
|
|
password = forms.CharField(
|
|
label="confirm password",
|
|
strip=False,
|
|
widget=forms.PasswordInput(attrs={'autocomplete': 'current-password'}),
|
|
)
|
|
|
|
def __init__(self, user, *args, **kwargs):
|
|
self.user = user
|
|
self.account = user.account
|
|
super(EditProfileForm, self).__init__(*args, **kwargs)
|
|
|
|
def clean(self):
|
|
email = self.cleaned_data.get('email')
|
|
first_name = self.cleaned_data.get('first_name')
|
|
last_name = self.cleaned_data.get('last_name')
|
|
if 'password' in self.cleaned_data:
|
|
password = self.cleaned_data["password"]
|
|
else:
|
|
password = ''
|
|
if 'twitter_handle' in self.cleaned_data:
|
|
t_handle = self.cleaned_data['twitter_handle']
|
|
else:
|
|
t_handle = ''
|
|
if not self.user.check_password(password):
|
|
raise ValidationError("password is incorrect.")
|
|
if email != self.user.email:
|
|
if User.objects.filter(email=email).exists():
|
|
raise ValidationError("An account already exists with this email address.")
|
|
return {
|
|
'email': email,
|
|
'first_name': first_name,
|
|
'last_name': last_name,
|
|
'twitter_handle': t_handle,
|
|
}
|
|
|
|
def save(self, commit=True):
|
|
self.user.email = self.cleaned_data['email']
|
|
self.user.first_name = self.cleaned_data['first_name']
|
|
self.user.last_name = self.cleaned_data['last_name']
|
|
self.account.twitter_handle = self.cleaned_data['twitter_handle']
|
|
if commit:
|
|
self.user.save()
|
|
self.account.save()
|
|
return self.user
|