from django.contrib import admin
from django.contrib.auth.admin import UserAdmin as BaseUserAdmin

from notes.models import Note, User

# Register your models here.
#


@admin.register(User)
class UserAdmin(BaseUserAdmin):
    fieldsets = [
        *BaseUserAdmin.fieldsets,
        ("Notes", {"fields": ("allow_notes_from", "expiry_seconds")}),
    ]
    add_fieldsets = [
        *BaseUserAdmin.add_fieldsets,
        ("Notes", {"fields": ("allow_notes_from", "expiry_seconds")}),
    ]
    filter_horizontal = (
        *BaseUserAdmin.filter_horizontal,
        "allow_notes_from",
    )


@admin.register(Note)
class NoteAdmin(admin.ModelAdmin):
    pass