from django.contrib import admin
from .models import UserRelation
from django.contrib import admin
from django.contrib.auth.admin import UserAdmin
from django.contrib.auth.models import User
from .models import Messages


class MessagesAdmin(admin.ModelAdmin):
    list_display = ("sender_name", "receiver_name", "id" , "seen", "description")
    list_filter = ("sender_name", "receiver_name", "seen")
    search_fields = ("sender_name__username", "receiver_name__username", "description")


admin.site.register(Messages, MessagesAdmin)


# Define a custom admin class for the User model
class CustomUserAdmin(UserAdmin):
    list_display = (
        "id",
        "username",
        "email",
        "first_name",
        "last_name",
        "is_staff",
        "date_joined",
    )


# Unregister the default UserAdmin
admin.site.unregister(User)

# Register the User model with the custom admin class
admin.site.register(User, CustomUserAdmin)


class UserRelationAdmin(admin.ModelAdmin):
    list_display = ("id", "user", "friend", "accepted", "relation_key")
    list_filter = ("user", "accepted")
    search_fields = ("user__username", "friend")


admin.site.register(UserRelation, UserRelationAdmin)