28 lines
768 B
Python
28 lines
768 B
Python
from django import forms
|
|
from django.contrib.auth import forms as base_auth_forms
|
|
from django.contrib.auth.forms import AdminUserCreationForm
|
|
|
|
from user.models import User
|
|
|
|
|
|
class RegisterForm(base_auth_forms.UserCreationForm):
|
|
email = forms.EmailField(required=True)
|
|
|
|
class Meta(base_auth_forms.UserCreationForm.Meta):
|
|
model = User
|
|
|
|
|
|
class UserCreationForm(AdminUserCreationForm):
|
|
max_size = forms.IntegerField(required=True)
|
|
email = forms.EmailField(required=True)
|
|
class Meta(base_auth_forms.UserCreationForm):
|
|
model = User
|
|
fields = base_auth_forms.BaseUserCreationForm.Meta.fields + ("max_size",)
|
|
|
|
|
|
class UserChangeForm(base_auth_forms.UserChangeForm):
|
|
class Meta:
|
|
model = User
|
|
fields = ["max_size"]
|
|
|