Initial commit

dashboard_project/accounts/forms.py

@@ -0,0 +1,45 @@
+# accounts/forms.py
+
+from django import forms
+from django.contrib.auth.forms import UserCreationForm
+
+from .models import Company, CustomUser
+
+
+class CustomUserCreationForm(UserCreationForm):
+    """Form for creating new users"""
+
+    class Meta:
+        model = CustomUser
+        fields = ("username", "email", "password1", "password2")
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        # Add help text for fields
+        self.fields["email"].required = True
+        self.fields["email"].help_text = "Required. Enter a valid email address."
+
+
+class CustomUserChangeForm(forms.ModelForm):
+    """Form for updating users"""
+
+    class Meta:
+        model = CustomUser
+        fields = ("username", "email", "company", "is_company_admin")
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        # Only staff members can change company and admin status
+        if not kwargs.get("instance") or not kwargs.get("instance").is_staff:
+            if "company" in self.fields:
+                self.fields["company"].disabled = True
+            if "is_company_admin" in self.fields:
+                self.fields["is_company_admin"].disabled = True
+
+
+class CompanyForm(forms.ModelForm):
+    """Form for creating and updating companies"""
+
+    class Meta:
+        model = Company
+        fields = ("name", "description")