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


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


admin.site.register(Profile, ProfileAdmin)


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


admin.site.register(Notification, NotificationAdmin)