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

class CustomUserAdmin(UserAdmin):
    # 🚀 POPRAVLJENO: Zamijenjen 'is_serviser' s 'uloga' u stupcima tablice
    list_display = ['email', 'first_name', 'last_name', 'telefon', 'oib', 'uloga', 'is_staff']
    
    # Ako filtriraš korisnike na desnoj strani admina, ažuriraj i to:
    list_filter = ['uloga', 'is_staff', 'is_active']
    
    # 🚀 POPRAVLJENO: Dodavanje polja u sekciju za uređivanje postojećeg korisnika
    fieldsets = UserAdmin.fieldsets + (
        ('Dodatni podaci tvrtke', {'fields': ('telefon', 'oib', 'uloga')}),
    )

    # 🚀 POPRAVLJENO: Dodavanje polja u formu za kreiranje novog korisnika
    add_fieldsets = UserAdmin.add_fieldsets + (
        ('Dodatni podaci tvrtke', {'fields': ('telefon', 'oib', 'uloga')}),
    )

# Registracija modela i tvoje prilagođene admin klase
admin.site.register(CustomUser, CustomUserAdmin)