admin.py

rom django.contrib import admin
from django.contrib.auth.admin import UserAdmin
from django import forms
from django.contrib.auth.forms import (UserChangeForm, UserCreationForm, )
from account.models import User, ROLES

class UserForm(UserChangeForm):
    role = forms.MultipleChoiceField(choices=ROLES, widget=forms.SelectMultiple)


class MyUserAdmin(UserAdmin):
    search_fields = ('email',)
    list_display = ('id', 'email', 'name', 'is_staff', 'role')
    form = UserForm
    add_form = UserCreationForm

    fieldsets = (
        (None, {'fields': ('username', 'email', 'password', 'role', 'image')}),
        ("Personal information", {'fields': ('name',)}),
        ("Permissions", {'fields': ('is_active', 'is_staff', 'is_superuser',)}),
        ("Activity", {'fields': ('last_login', 'date_joined')}),
    )


admin.site.register(Client, ClientAdmin)
admin.site.register(Notification, NotificationAdmin)
admin.site.register(User, MyUserAdmin)
admin.site.register(ClientProfile, ClientProfileAdmin)