from django.contrib import admin
from .models import Club, Tournament
from django.contrib.auth.admin import UserAdmin
from django.contrib.auth.forms import UserCreationForm, UserChangeForm

from .forms import CustomUserCreationForm, CustomUserChangeForm
from .models import CustomUser

class CustomUserAdmin(UserAdmin):
    form = CustomUserChangeForm
    add_form = CustomUserCreationForm
    model = CustomUser
    list_display = ["email", "username", "umpire_code"]
    fieldsets = [
            (None, {"fields": ["username", "email", "password", "club", "umpire_code"]}),
        ]
    add_fieldsets = [
        (
            None,
            {
                "classes": ["wide"],
                "fields": ["username", "email", "password1", "password2", "club", "umpire_code"],
            },
        ),
    ]

admin.site.register(CustomUser, CustomUserAdmin)
admin.site.register(Club)
admin.site.register(Tournament)