# -*- coding: utf-8 -*- """ Django Forms for Account Management This module contains all forms used in the accounts app for user management, profile updates, authentication, and role assignments. """ import pytz from django import forms from django.core.exceptions import ValidationError from django.utils.translation import gettext_lazy as _ from django.contrib.auth.forms import PasswordChangeForm from apps.accounts.models import User, Profile, Role, UserRole class ProfileUpdateForm(forms.ModelForm): """Form for updating user profile information.""" first_name = forms.CharField( max_length=30, widget=forms.TextInput(attrs={ 'class': 'form-control', 'placeholder': 'First Name' }) ) last_name = forms.CharField( max_length=30, widget=forms.TextInput(attrs={ 'class': 'form-control', 'placeholder': 'Last Name' }) ) phone_number = forms.CharField( max_length=17, required=False, widget=forms.TextInput(attrs={ 'class': 'form-control', 'placeholder': 'Phone Number' }) ) class Meta: model = Profile fields = [ 'job_title', 'bio', 'website', 'timezone', 'language', 'theme', 'avatar', 'notifications_enabled', 'email_notifications', 'sms_notifications' ] widgets = { 'job_title': forms.TextInput(attrs={'class': 'form-control', 'placeholder': 'Job Title'}), 'bio': forms.Textarea(attrs={'class': 'form-control', 'rows': 3, 'maxlength': 500}), 'website': forms.URLInput(attrs={'class': 'form-control', 'placeholder': 'Website URL'}), 'timezone': forms.Select(attrs={'class': 'form-select'}), 'language': forms.Select(attrs={'class': 'form-select'}), 'theme': forms.Select(attrs={'class': 'form-select'}), 'avatar': forms.FileInput(attrs={'class': 'form-control', 'accept': 'image/*'}), 'notifications_enabled': forms.CheckboxInput(attrs={'class': 'form-check-input'}), 'email_notifications': forms.CheckboxInput(attrs={'class': 'form-check-input'}), 'sms_notifications': forms.CheckboxInput(attrs={'class': 'form-check-input'}), } def __init__(self, *args, **kwargs): self.user = kwargs.pop('user', None) super().__init__(*args, **kwargs) # Add timezone choices self.fields['timezone'].choices = [(tz, tz) for tz in pytz.common_timezones] # Add language choices self.fields['language'].choices = [ ('en', 'English'), ('es', 'Spanish'), ('fr', 'French'), ('de', 'German'), ('it', 'Italian'), ('pt', 'Portuguese'), ('ru', 'Russian'), ('ja', 'Japanese'), ('ko', 'Korean'), ('zh', 'Chinese'), ] # Add theme choices self.fields['theme'].choices = [ ('light', 'Light'), ('dark', 'Dark'), ('auto', 'Auto'), ] # Add first_name, last_name and phone_number fields if self.user: self.fields['first_name'].initial = self.user.first_name self.fields['last_name'].initial = self.user.last_name self.fields['phone_number'].initial = self.user.phone_number def save(self, commit=True): instance = super().save(commit=False) # Update user fields if self.user: self.user.first_name = self.cleaned_data.get('first_name', '') self.user.last_name = self.cleaned_data.get('last_name', '') self.user.phone_number = self.cleaned_data.get('phone_number', '') if commit: self.user.save() if commit: instance.save() return instance class CustomPasswordChangeForm(PasswordChangeForm): """Custom password change form with Bootstrap styling.""" old_password = forms.CharField( widget=forms.PasswordInput(attrs={ 'class': 'form-control', 'placeholder': 'Current Password' }) ) new_password1 = forms.CharField( widget=forms.PasswordInput(attrs={ 'class': 'form-control', 'placeholder': 'New Password' }) ) new_password2 = forms.CharField( widget=forms.PasswordInput(attrs={ 'class': 'form-control', 'placeholder': 'Confirm New Password' }) ) class UserCreateForm(forms.ModelForm): """Form for creating new users.""" password1 = forms.CharField( label='Password', widget=forms.PasswordInput(attrs={'class': 'form-control'}) ) password2 = forms.CharField( label='Confirm Password', widget=forms.PasswordInput(attrs={'class': 'form-control'}) ) class Meta: model = User fields = ['email', 'username', 'first_name', 'last_name', 'is_active', 'is_staff'] widgets = { 'email': forms.EmailInput(attrs={'class': 'form-control'}), 'username': forms.TextInput(attrs={'class': 'form-control'}), 'first_name': forms.TextInput(attrs={'class': 'form-control'}), 'last_name': forms.TextInput(attrs={'class': 'form-control'}), 'is_active': forms.CheckboxInput(attrs={'class': 'form-check-input'}), 'is_staff': forms.CheckboxInput(attrs={'class': 'form-check-input'}), } def clean_password2(self): password1 = self.cleaned_data.get('password1') password2 = self.cleaned_data.get('password2') if password1 and password2 and password1 != password2: raise ValidationError("Passwords don't match") return password2 def save(self, commit=True): user = super().save(commit=False) user.set_password(self.cleaned_data['password1']) if commit: user.save() return user class UserUpdateForm(forms.ModelForm): """Form for updating user information.""" class Meta: model = User fields = ['email', 'username', 'first_name', 'last_name', 'is_active', 'is_staff'] widgets = { 'email': forms.EmailInput(attrs={'class': 'form-control'}), 'username': forms.TextInput(attrs={'class': 'form-control'}), 'first_name': forms.TextInput(attrs={'class': 'form-control'}), 'last_name': forms.TextInput(attrs={'class': 'form-control'}), 'is_active': forms.CheckboxInput(attrs={'class': 'form-check-input'}), 'is_staff': forms.CheckboxInput(attrs={'class': 'form-check-input'}), } class RoleAssignmentForm(forms.ModelForm): """Form for assigning roles to users.""" class Meta: model = UserRole fields = ['role', 'valid_from', 'valid_until', 'notes'] widgets = { 'role': forms.Select(attrs={'class': 'form-select'}), 'valid_from': forms.DateTimeInput(attrs={'class': 'form-control', 'type': 'datetime-local'}), 'valid_until': forms.DateTimeInput(attrs={'class': 'form-control', 'type': 'datetime-local'}), 'notes': forms.Textarea(attrs={'class': 'form-control', 'rows': 3}), } def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) self.fields['role'].queryset = Role.objects.filter(is_active=True) self.fields['valid_until'].required = False self.fields['notes'].required = False class RoleForm(forms.ModelForm): """Form for creating or updating roles.""" class Meta: model = Role fields = ['name', 'description', 'is_active'] widgets = { 'name': forms.TextInput(attrs={'class': 'form-control'}), 'description': forms.Textarea(attrs={'class': 'form-control', 'rows': 3}), 'is_active': forms.CheckboxInput(attrs={'class': 'form-check-input'}), }