from django.contrib import admin
from django.contrib.auth.admin import UserAdmin
from .models import User, Item, Favorite

class CustomUserAdmin(UserAdmin):
    fieldsets = UserAdmin.fieldsets + (
        ('Custom Role', {'fields': ('is_auctioneer',)}),
    )
    add_fieldsets = UserAdmin.add_fieldsets + (
        ('Custom Role', {'fields': ('is_auctioneer',)}),
    )
    list_display = ('username', 'email', 'is_staff', 'is_auctioneer')

admin.site.register(User, CustomUserAdmin)
admin.site.register(Item)
admin.site.register(Favorite)