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

class CustomUserAdmin(UserAdmin):
    model = User
    fieldsets = UserAdmin.fieldsets + (
        ('Tenant Info', {'fields': ('tenant', 'role')}),
    )
    list_display = ['username', 'email', 'first_name', 'last_name', 'role', 'tenant', 'is_staff']

admin.site.register(User, CustomUserAdmin)