16 lines
428 B
Python
16 lines
428 B
Python
|
from django import forms
|
||
|
|
from django.contrib.auth.forms import UserCreationForm, UserChangeForm
|
||
|
|
|
||
|
|
from .models import CustomUser
|
||
|
|
|
||
|
|
class CustomUserCreationForm(UserCreationForm):
|
||
|
|
class Meta(UserCreationForm):
|
||
|
|
model = CustomUser
|
||
|
|
fields = UserCreationForm.Meta.fields + ('age',)
|
||
|
|
|
||
|
|
class CustomUserChangeForm(UserChangeForm):
|
||
|
|
|
||
|
|
class Meta:
|
||
|
|
model = CustomUser
|
||
|
|
fields = UserChangeForm.Meta.fields
|