from django import forms from django.contrib.auth.models import User from django.contrib.auth.forms import AuthenticationForm from django.core.exceptions import ValidationError import re from .models import Cart, ShippingAddress, Order, OrderItem, ShippingMethod from inventory.models import Product, Category class CustomerRegistrationForm(forms.ModelForm): password = forms.CharField( widget=forms.PasswordInput(attrs={ 'class': 'form-control', 'placeholder': 'Enter password' }), min_length=6, help_text='Password must be at least 6 characters long' ) confirm_password = forms.CharField( widget=forms.PasswordInput(attrs={ 'class': 'form-control', 'placeholder': 'Confirm password' }) ) phone = forms.CharField( max_length=15, required=False, widget=forms.TextInput(attrs={ 'class': 'form-control', 'placeholder': 'Phone number (optional)' }) ) address = forms.CharField( max_length=255, required=False, widget=forms.Textarea(attrs={ 'class': 'form-control', 'placeholder': 'Address (optional)', 'rows': 3 }) ) class Meta: model = User fields = ['username', 'email', 'first_name', 'last_name'] widgets = { 'username': forms.TextInput(attrs={ 'class': 'form-control', 'placeholder': 'Choose a username' }), 'email': forms.EmailInput(attrs={ 'class': 'form-control', 'placeholder': 'Email address' }), 'first_name': forms.TextInput(attrs={ 'class': 'form-control', 'placeholder': 'First name' }), 'last_name': forms.TextInput(attrs={ 'class': 'form-control', 'placeholder': 'Last name' }), } def clean_username(self): username = self.cleaned_data.get('username') if User.objects.filter(username=username).exists(): raise ValidationError('Username already exists. Please choose a different one.') return username def clean_email(self): email = self.cleaned_data.get('email') if User.objects.filter(email=email).exists(): raise ValidationError('Email already registered. Please use a different email.') return email def clean_phone(self): phone = self.cleaned_data.get('phone') if phone and not re.match(r'^\+?[\d\s\-\(\)]+$', phone): raise ValidationError('Please enter a valid phone number.') return phone def clean_confirm_password(self): password = self.cleaned_data.get('password') confirm_password = self.cleaned_data.get('confirm_password') if password and confirm_password and password != confirm_password: raise ValidationError('Passwords do not match.') return confirm_password class CustomerLoginForm(AuthenticationForm): username = forms.CharField( widget=forms.TextInput(attrs={ 'class': 'form-control', 'placeholder': 'Username or Email' }) ) password = forms.CharField( widget=forms.PasswordInput(attrs={ 'class': 'form-control', 'placeholder': 'Password' }) ) def clean_username(self): username = self.cleaned_data.get('username') # Allow login with email if '@' in username: try: user = User.objects.get(email=username) return user.username except User.DoesNotExist: pass return username class CustomerProfileForm(forms.ModelForm): phone = forms.CharField( max_length=15, required=False, widget=forms.TextInput(attrs={ 'class': 'form-control', 'placeholder': 'Phone number' }) ) address = forms.CharField( max_length=255, required=False, widget=forms.Textarea(attrs={ 'class': 'form-control', 'placeholder': 'Address', 'rows': 3 }) ) class Meta: model = User fields = ['username', 'email', 'first_name', 'last_name'] widgets = { 'username': forms.TextInput(attrs={ 'class': 'form-control', 'readonly': True }), 'email': forms.EmailInput(attrs={ 'class': 'form-control', 'placeholder': 'Email address' }), 'first_name': forms.TextInput(attrs={ 'class': 'form-control', 'placeholder': 'First name' }), 'last_name': forms.TextInput(attrs={ 'class': 'form-control', 'placeholder': 'Last name' }), } def clean_email(self): email = self.cleaned_data.get('email') # Check if email is already taken by another user if User.objects.filter(email=email).exclude(id=self.instance.id).exists(): raise ValidationError('Email already registered by another user.') return email class CustomerAddressForm(forms.ModelForm): class Meta: model = ShippingAddress fields = ['address_line_1', 'address_line_2', 'city', 'postal_code', 'country'] widgets = { 'address_line_1': forms.TextInput(attrs={ 'class': 'form-control', 'placeholder': 'Address Line 1' }), 'address_line_2': forms.TextInput(attrs={ 'class': 'form-control', 'placeholder': 'Address Line 2 (optional)' }), 'city': forms.TextInput(attrs={ 'class': 'form-control', 'placeholder': 'City' }), 'postal_code': forms.TextInput(attrs={ 'class': 'form-control', 'placeholder': 'Postal Code' }), 'country': forms.TextInput(attrs={ 'class': 'form-control', 'placeholder': 'Country' }), } class CustomerPasswordChangeForm(forms.Form): current_password = forms.CharField( widget=forms.PasswordInput(attrs={ 'class': 'form-control', 'placeholder': 'Current Password' }), label='Current Password' ) new_password1 = forms.CharField( widget=forms.PasswordInput(attrs={ 'class': 'form-control', 'placeholder': 'New Password' }), label='New Password', min_length=6, help_text='Password must be at least 6 characters long' ) new_password2 = forms.CharField( widget=forms.PasswordInput(attrs={ 'class': 'form-control', 'placeholder': 'Confirm New Password' }), label='Confirm New Password' ) def __init__(self, user, *args, **kwargs): self.user = user super().__init__(*args, **kwargs) def clean_current_password(self): current_password = self.cleaned_data.get('current_password') if not self.user.check_password(current_password): raise ValidationError('Current password is incorrect.') return current_password def clean_new_password2(self): new_password1 = self.cleaned_data.get('new_password1') new_password2 = self.cleaned_data.get('new_password2') if new_password1 and new_password2 and new_password1 != new_password2: raise ValidationError('New passwords do not match.') return new_password2 def save(self): self.user.set_password(self.cleaned_data['new_password1']) self.user.save() return self.user # ----------------- CART FORMS ----------------- class AddToCartForm(forms.Form): """ Form for adding products to the cart with validation for stock availability. """ product_id = forms.IntegerField(widget=forms.HiddenInput) # Hidden field for the product ID quantity = forms.IntegerField( min_value=1, label="Quantity", widget=forms.NumberInput(attrs={'class': 'form-control', 'placeholder': 'Enter quantity'}), ) def clean(self): cleaned_data = super().clean() product_id = cleaned_data.get('product_id') quantity = cleaned_data.get('quantity') # Validate product availability product = Product.objects.filter(id=product_id).first() if not product: raise forms.ValidationError("The selected product does not exist.") if quantity > product.stock_quantity: raise forms.ValidationError(f"Only {product.stock_quantity} units available.") return cleaned_data # ----------------- CHECKOUT FORMS ----------------- class GuestCheckoutForm(forms.Form): """ Form for guest users to provide an email during checkout, with validation for existing users. """ email = forms.EmailField( label="Email Address", widget=forms.EmailInput(attrs={'class': 'form-control', 'placeholder': 'Enter your email'}), required=True, ) def clean_email(self): email = self.cleaned_data['email'] if User.objects.filter(email=email).exists(): raise forms.ValidationError("This email is associated with an existing account. Please log in.") return email class ShippingAddressForm(forms.ModelForm): """ Form for capturing shipping details during checkout. """ class Meta: model = ShippingAddress fields = ['address_line_1', 'address_line_2', 'city', 'postal_code', 'country'] widgets = { 'address_line_1': forms.TextInput(attrs={'class': 'form-control', 'placeholder': 'Address Line 1'}), 'address_line_2': forms.TextInput(attrs={'class': 'form-control', 'placeholder': 'Address Line 2 (optional)'}), 'city': forms.TextInput(attrs={'class': 'form-control', 'placeholder': 'City'}), 'postal_code': forms.TextInput(attrs={'class': 'form-control', 'placeholder': 'Postal Code'}), 'country': forms.Select(attrs={'class': 'form-control'}), } class ShippingMethodForm(forms.Form): """ Form for selecting a shipping method during checkout. """ shipping_method = forms.ModelChoiceField( queryset=ShippingMethod.objects.filter(is_active=True), widget=forms.RadioSelect, label="Shipping Method", empty_label=None, ) # ----------------- ORDER FORMS ----------------- class OrderItemForm(forms.ModelForm): """ Form for managing items in an order, with validation for quantity and price. """ class Meta: model = OrderItem fields = ['product', 'quantity', 'price'] widgets = { 'quantity': forms.NumberInput(attrs={'class': 'form-control', 'placeholder': 'Quantity'}), 'price': forms.NumberInput(attrs={'class': 'form-control', 'placeholder': 'Price'}), } def clean_quantity(self): quantity = self.cleaned_data.get('quantity') if quantity <= 0: raise forms.ValidationError("Quantity must be at least 1.") return quantity def clean_price(self): price = self.cleaned_data.get('price') product = self.cleaned_data.get('product') if product and price != product.selling_price: raise forms.ValidationError("Invalid price for the selected product.") return price class OrderForm(forms.ModelForm): """ Form for creating or editing orders. """ class Meta: model = Order fields = ['shipping_address', 'shipping_method', 'status'] widgets = { 'shipping_address': forms.Select(attrs={'class': 'form-control'}), 'shipping_method': forms.Select(attrs={'class': 'form-control'}), 'status': forms.Select(attrs={'class': 'form-control'}), } # ----------------- SEARCH FORMS ----------------- class SearchForm(forms.Form): """ Form for searching products with optional filters. """ query = forms.CharField( label="Search", required=False, widget=forms.TextInput(attrs={'class': 'form-control', 'placeholder': 'Search for products...'}), ) category = forms.ChoiceField( label="Category", required=False, choices=[], # To be dynamically populated in the view widget=forms.Select(attrs={'class': 'form-control'}), ) min_price = forms.DecimalField( label="Min Price", required=False, min_value=0, widget=forms.NumberInput(attrs={'class': 'form-control', 'placeholder': 'Minimum price'}), ) max_price = forms.DecimalField( label="Max Price", required=False, min_value=0, widget=forms.NumberInput(attrs={'class': 'form-control', 'placeholder': 'Maximum price'}), ) def __init__(self, *args, **kwargs): """ Dynamically populate category choices. """ super().__init__(*args, **kwargs) self.fields['category'].choices = [('', 'All Categories')] + [ (category.id, category.name) for category in Category.objects.all() ] def clean(self): cleaned_data = super().clean() min_price = cleaned_data.get('min_price') max_price = cleaned_data.get('max_price') if min_price is not None and max_price is not None and min_price > max_price: raise forms.ValidationError("Minimum price cannot be greater than maximum price.") return cleaned_data class ShippingMethodForm(forms.ModelForm): """ Form for creating and updating shipping methods. """ class Meta: model = ShippingMethod fields = ['name', 'description', 'cost', 'is_active']