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

from .models import Follow, User


@admin.register(User)
class UserAdmin(DjangoUserAdmin):
    fieldsets = DjangoUserAdmin.fieldsets + (
        ("Aurora Moments profile", {"fields": ("display_name", "bio", "avatar", "is_private")}),
    )
    list_display = ("username", "email", "display_name", "is_private", "is_staff")


@admin.register(Follow)
class FollowAdmin(admin.ModelAdmin):
    list_display = ("follower", "following", "status", "created_at")
    list_filter = ("status",)