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


class CustomUserAdmin(UserAdmin):
    model = CustomUser
    list_display = ('id', 'is_active', 'user_group',) + UserAdmin.list_display
    # Editable fields per instance
    fieldsets = UserAdmin.fieldsets + (
        (None, {'fields': ('avatar',)}),
    )


admin.site.register(CustomUser, CustomUserAdmin)