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


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


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


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


admin.site.register(Profile, ProfileAdmin)
admin.site.register(Notification, NotificationAdmin)
admin.site.register(TrainerGymnast, TrainerGymnastAdmin)