76 lines
2.5 KiB
Python
76 lines
2.5 KiB
Python
|
# accounts/forms.py
|
||
|
|
from django import forms
|
||
|
|
from django.contrib.auth import get_user_model
|
||
|
|
from django.contrib.auth import authenticate
|
||
|
|
|
||
|
|
User = get_user_model()
|
||
|
|
|
||
|
|
|
||
|
|
class RegisterForm(forms.ModelForm):
|
||
|
|
password = forms.CharField(widget=forms.PasswordInput)
|
||
|
|
|
||
|
|
class Meta:
|
||
|
|
model = User
|
||
|
|
fields = ['email', 'phone_number', 'password']
|
||
|
|
|
||
|
|
def clean_email(self):
|
||
|
|
email = self.cleaned_data.get('email')
|
||
|
|
if User.objects.filter(email=email).exists():
|
||
|
|
raise forms.ValidationError("Email is already registered.")
|
||
|
|
return email
|
||
|
|
|
||
|
|
def clean_phone_number(self):
|
||
|
|
phone_number = self.cleaned_data.get('phone_number')
|
||
|
|
if User.objects.filter(phone_number=phone_number).exists():
|
||
|
|
raise forms.ValidationError("Phone number is already registered.")
|
||
|
|
return phone_number
|
||
|
|
|
||
|
|
def save(self, commit=True):
|
||
|
|
user = super().save(commit=False)
|
||
|
|
user.set_password(self.cleaned_data['password'])
|
||
|
|
if commit:
|
||
|
|
user.save()
|
||
|
|
return user
|
||
|
|
|
||
|
|
|
||
|
|
class LoginForm(forms.Form):
|
||
|
|
username = forms.CharField()
|
||
|
|
password = forms.CharField(widget=forms.PasswordInput)
|
||
|
|
|
||
|
|
def clean(self):
|
||
|
|
cleaned_data = super().clean()
|
||
|
|
username = cleaned_data.get('username')
|
||
|
|
password = cleaned_data.get('password')
|
||
|
|
|
||
|
|
print('*' * 100)
|
||
|
|
print(username, password)
|
||
|
|
print('*' * 100)
|
||
|
|
|
||
|
|
if not username or not password:
|
||
|
|
raise forms.ValidationError("Username and password are required.")
|
||
|
|
|
||
|
|
# Check if username contains '@' (email) or is a regular username
|
||
|
|
try:
|
||
|
|
if '@' in username:
|
||
|
|
print('1 **********************')
|
||
|
|
# Try to find user by email
|
||
|
|
user = User.objects.get(email=username)
|
||
|
|
print(user)
|
||
|
|
print('2 **********************')
|
||
|
|
username = user.username
|
||
|
|
print('3 **********************')
|
||
|
|
else:
|
||
|
|
print('4 **********************')# Use username as-is
|
||
|
|
user = User.objects.get(username=username)
|
||
|
|
except User.DoesNotExist:
|
||
|
|
print('5 **********************')
|
||
|
|
raise forms.ValidationError("Invalid credentials.")
|
||
|
|
|
||
|
|
# Authenticate with the resolved username
|
||
|
|
user = authenticate(username=username, password=password)
|
||
|
|
if not user:
|
||
|
|
raise forms.ValidationError("Invalid credentials.")
|
||
|
|
|
||
|
|
cleaned_data['user'] = user
|
||
|
|
return cleaned_data
|