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

from .models import User


@admin.register(User)
class CustomUserAdmin(UserAdmin):
    list_display = ("username", "email", "role", "is_staff", "is_superuser")
    list_filter = UserAdmin.list_filter + ("role",)
    fieldsets = UserAdmin.fieldsets + (
        (
            "J621",
            {
                "fields": (
                    "role",
                    "e621_username",
                    "e621_api_key",
                    "e621_base_url",
                )
            },
        ),
    )