17 lines
552 B
Python
17 lines
552 B
Python
from django import forms
|
|
from django.contrib.auth.forms import UserCreationForm
|
|
from .models import CustomUser # adjust import to your app
|
|
|
|
#using: admin.py
|
|
class CustomUserCreationForm(UserCreationForm):
|
|
class Meta:
|
|
model = CustomUser
|
|
fields = ("username", "email", "role", "account_type", "password1", "password2")
|
|
|
|
def save(self, commit=True):
|
|
user = super().save(commit=False)
|
|
# Optional logic: assign role-based permissions here if needed
|
|
if commit:
|
|
user.save()
|
|
return user
|