from django.contrib import admin
from django.contrib.auth.admin import UserAdmin as BaseUserAdmin
from .models import Book, User, Order, OrderItem, Comment

class UserAdmin(BaseUserAdmin):
    fieldsets = BaseUserAdmin.fieldsets + (
        (None, {'fields': ('avatar', 'phone', 'address', 'name', 'roles')}),
    )
    add_fieldsets = BaseUserAdmin.add_fieldsets + (
        (None, {'fields': ('email', 'phone', 'address', 'name')}),
    )
    list_display = BaseUserAdmin.list_display + ('phone', 'is_staff')

admin.site.register(Book)
admin.site.register(User, UserAdmin)
admin.site.register(Order)
admin.site.register(OrderItem)
admin.site.register(Comment)