from django.contrib import admin
from .models import Gymnast


class GymnastAdmin(admin.ModelAdmin):
    model = Gymnast

    def last_name(self, obj):
        return obj.user.last_name

    def first_name(self, obj):
        return obj.user.first_name

    def email(self, obj):
        return obj.user.email

    def is_active(self, obj):
        return obj.user.is_active

    fields = (
        "last_name",
        "first_name",
        "user",
        "birthdate",
        "gender",
        # "email",
        "club",
        "email_trainer",
        "trainings_by_week",
        "hours_by_week",
        "informations",
        "is_active",
    )

    list_display = ("last_name", "first_name", "age", "is_active")  # , 'club'
    list_filter = ("gender", "user__is_active")  # , 'club'
    search_fields = ("last_name", "first_name")
    # autocomplete_fields = ('club',)


admin.site.register(Gymnast, GymnastAdmin)