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"]