from django.contrib import admin
from django.contrib.auth.admin import UserAdmin
from .models import CustomUser, AbsencePeriod

class CustomUserAdmin(UserAdmin):
    model = CustomUser
    list_display = ['email', 'username', 'team', 'display_parents', 'is_staff','birth_date','player_number']
    fieldsets = UserAdmin.fieldsets + (
        (None, {'fields': ('team', 'parents','birth_date','player_number')}),
    )
    filter_horizontal = ('parents',)

    def display_parents(self, obj):
        return ", ".join([parent.username for parent in obj.parents.all()])
    display_parents.short_description = 'Parents'

admin.site.register(CustomUser, CustomUserAdmin)
admin.site.register(AbsencePeriod)