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

@admin.register(CustomUser)
class CustomUserAdmin(UserAdmin):    
  add_form = CustomUserCreationForm
  
  fieldsets = (
    *UserAdmin.fieldsets,
    (
      'Individual data',
      {
        'fields': (
          'favorite_videos',
          'watched_videos',
          'verify_email_token',
          )
       }
     )
   )