from django.contrib import admin
from django_admin_listfilter_dropdown.filters import RelatedDropdownFilter
from .models import Profile, Notification, TrainerGymnast


@admin.register(Profile)
class ProfileAdmin(admin.ModelAdmin):
    model = Profile
    list_display = ("user", "template_color", "sidebar_color")
    autocomplete_fields = ("user",)


@admin.register(Notification)
class NotificationAdmin(admin.ModelAdmin):
    model = Notification
    list_display = ("user", "gymnast", "functionality")
    autocomplete_fields = ("user", "gymnast")
    list_filter = (
        ("user", RelatedDropdownFilter),
        ("gymnast", RelatedDropdownFilter),
        ("functionality", RelatedDropdownFilter),
    )


@admin.register(TrainerGymnast)
class TrainerGymnastAdmin(admin.ModelAdmin):
    model = TrainerGymnast
    list_display = ("user", "gymnast")
    autocomplete_fields = ("user", "gymnast")
    list_filter = (
        ("user", RelatedDropdownFilter),
        ("gymnast", RelatedDropdownFilter),
    )